Update llm.py

llm.py

@@ -23,12 +23,13 @@ def similarity(filename, repo_id, model_kwargs, query):
         ),
         chain_type="stuff"
     )
-    question = f"""
-    Answer the question based on the context, if you don't know then output "Out of Context".
-    Context: \n {chunks[0].page_content} \n
-    Question: \n {query} \n
-    Answer:
-    """
+    question = f"<|system|>\nYou are a intelligent chatbot and expertise in {chunks[0].page_content}.</s>\n<|user|>\n{query}.\n<|assistant|>"
+    # question = f"""
+    # Answer the question based on the context, if you don't know then output "Out of Context".
+    # Context: \n {chunks[0].page_content} \n
+    # Question: \n {query} \n
+    # Answer:
+    # """
     result = chain.run(
         input_documents=docs,
         question=question