Update chatbot_rag.py

chatbot_rag.py

@@ -79,23 +79,22 @@ def build_qa():
     retriever = vectorstore.as_retriever(search_kwargs={"k": 3})
 
     prompt = PromptTemplate(
-    input_variables=["context", "question"],
-    template="""
-    Use the following context to answer the question.
-    - Answer from the docs
-    - Answer in plain natural language.
-    - Do not include code, imports, functions, or explanations of how to implement code.
-    - If you don't know, just say "I don't know."
-
-    Context:
-    {context}
-
-    Question: {question}
-    Answer (one short sentence):
-    """,
+        input_variables=["context", "question"],
+        template="""
+        You are a helpful assistant.
+        Use the following context to answer the question.
+        - If the answer is in the context, provide it in ONE short, plain sentence.
+        - If not in the context, say: "I don't know."
+        
+        Context:
+        {context}
+        Question: {question}
+        Answer:
+        """,
     )
 
 
+
     # 6. Helper functions
     def format_docs(docs):
         """Join document contents into a single string, skipping empty ones."""