Update app.py

app.py

@@ -195,12 +195,11 @@ def route_llm(context: str, query: str):
 
 @outlines.prompt
 def answer_with_context(context: str, query: str):
-    """Context information is below.
-    ---------------------
-    {context}
-    ---------------------
-    Given the context information and not prior knowledge, answer the query.
-    Query: {query}
+    """You are a world class AI model who answer `Query` with informations extracted from the `Context`.
+
+    `Context`: {context}
+    
+    `Query`: {query}
     Answer:
     """
 
@@ -223,7 +222,7 @@ def main(query: str, client: QdrantClient, collection_name: str, llm, dense_mode
     contents, metadatas = [list(t) for t in zip(*docs)]
         
     context = "\n".join(contents)
-    print(f'Context : {context}')
+    print(f'Context : \n + {context}')
     
     gen_text = outlines.generate.json(llm, Question, whitespace_pattern=r"[\n ]?")