fix: missing shared ask personas

docqai · Feb 2, 2024 · 5963af5 · 5963af5
1 parent f2607b5
commit 5963af5
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 8 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "docq"
-version = "0.8.1"
+version = "0.8.2"
 description = "Docq.AI - private and secure knowledge insight on your data."
 authors = ["Docq.AI Team <[email protected]>"]
 maintainers = ["Docq.AI Team <[email protected]>"]

diff --git a/source/docq/manage_personas.py b/source/docq/manage_personas.py
@@ -58,6 +58,17 @@
             Query: {query_str}\n
             Answer: """,
     },
+}
+
+
+AGENT_PERSONAS = {}
+
+ASK_PERSONAS = {
+    "default": {
+        "name": "General Q&A Assistant",
+        "system_prompt_content": DEFAULT_QA_SYSTEM_PROMPT,
+        "user_prompt_template_content": DEFAULT_QA_USER_PROMPT_TEMPLATE,
+    },
     "meeting-assistant": {
         "name": "Meeting Assistant",
         "system_prompt_content": """You are a extremely helpful meeting assistant.
@@ -82,13 +93,6 @@
 }
 
 
-AGENT_PERSONAS = {
-}
-
-ASK_PERSONAS = {
-}
-
-
 def llama_index_chat_prompt_template_from_persona(persona: Persona) -> ChatPromptTemplate:
     """Get the prompt template for the llama index."""
     _system_prompt_message = ChatMessage(
@@ -121,6 +125,7 @@ def get_personas(persona_type: Optional[PersonaType] = None) -> dict[str, Person
         }
     return result
 
+
 def get_persona(key: str) -> Persona:
     """Get the persona."""
     if key not in SIMPLE_CHAT_PERSONAS: