Update chatbot_rag.py

chatbot_rag.py

@@ -52,8 +52,8 @@ def build_qa():
         model=model,
         tokenizer=tokenizer,
         max_new_tokens=256,
-        temperature=0.7,
-        do_sample=True,
+        temperature=0.2,
+        do_sample=False,
         return_full_text=False   # 🚀 only return new text, avoids messy context echoes
     )
     
@@ -64,19 +64,20 @@ 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.
-        
-        Context:
-        {context}
-        
-        Question: {question}
-        Answer (one short sentence):
-        """,
+    input_variables=["context", "question"],
+    template="""
+    You are a strict assistant. Only answer using the CONTEXT provided.  
+    If the CONTEXT does not have the answer, reply exactly with: "I don't know".  
+    
+    CONTEXT:
+    {context}
+    
+    QUESTION: {question}  
+    ANSWER:
+    """,
     )
 
+
     # 6. Helper functions
     def format_docs(docs):
         return "\n".join(doc.page_content for doc in docs)