we4

app.py

@@ -118,9 +118,20 @@ load_or_create_vectorstore()
 2. Create a Prompt Template from the String Template
 """
 ### 1. DEFINE STRING TEMPLATE
-RAG_PROMPT_TEMPLATE =  """"You are a helpful assistant. 
+RAG_PROMPT_TEMPLATE =  """"\
+<|start_header_id|>system<|end_header_id|>
+
+You are a helpful assistant. 
 You answer user questions based on provided context. 
-If you can't answer the question with the provided context, say you don't know.
+If you can't answer the question with the provided context, say you don't know.<|eot_id|>
+<|start_header_id|>user<|end_header_id|>
+User Query:
+{query}
+
+Context:
+{context}<|eot_id|>
+
+<|start_header_id|>assistant<|end_header_id|>
 """
 
 ### 2. CREATE PROMPT TEMPLATE