Update agent.py

agent.py

@@ -54,14 +54,28 @@ class BasicAgent:
             context = "No context found. Answer based on your knowledge."
 
         # Use truncated_question for LLM too
-        messages = [
-            {"role": "system", "content": "You are a helpful assistant. Do not call tools. Use context faithfully."},
-            {"role": "user", "content": f"Question: {truncated_question}\nContext:\n{context}\nAnswer concisely."}
+       messages = [
+            {
+                "role": "system", 
+                "content": (
+                    "You are a precise data extraction engine. "
+                    "Your task is to provide ONLY the exact answer to the user's question based on the context. "
+                    "Do not provide explanations, introductory text, or conversational filler. "
+                    "Do not say 'The answer is' or 'Based on the context'. "
+                    "If the answer is a name, number, or date, return JUST that specific value."
+                )
+            },
+            {
+                "role": "user", 
+                "content": f"Context:\n{context}\n\nQuestion: {truncated_question}\n\nExact Answer:"
+            }
         ]
 
         comp = self.llm_client.chat.completions.create(
             model=self.model,
             messages=messages,
+            temperature=0.0,  # 0.0 makes the model very strict and factual
+            max_tokens=100
         )
 
         return comp.choices[0].message.content.strip()