Update chatbot_rag.py

chatbot_rag.py

@@ -64,18 +64,19 @@ def build_qa():
 
     # 5. Prompt
     prompt = PromptTemplate(
-    input_variables=["context", "question"],
-    template = """
-    Use the following context to answer the question at the end. 
-    If you don't know the answer, just say "I don't know" — do not make up an answer.
-    
-    
-    Question: {question}
-    Answer (one short sentence):
-    """
+        input_variables=["context", "question"],
+        template="""
+        Use the following context to answer the question at the end. 
+        If you don't know the answer, just say "I don't know" — do not make up an answer.
+        
+        Context:
+        {context}
+        
+        Question: {question}
+        Answer (one short sentence):
+        """,
     )
 
-
     # 6. Helper functions
     def format_docs(docs):
         return "\n".join(doc.page_content for doc in docs)