Update chatbot_rag.py

chatbot_rag.py

@@ -78,19 +78,22 @@ def build_qa():
     # 4. Retriever
     retriever = vectorstore.as_retriever(search_kwargs={"k": 3})
 
-    prompt = PromptTemplate(
-        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."
+    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: 
         {context}
-        Question: {question}
-        Answer:
-        """,
+    
+    
+        Question: 
+        {question} 
+        
+        Answer (one short sentence): """,
     )