Update agents/solver_agent.py

agents/solver_agent.py

@@ -10,35 +10,32 @@ def is_expression(text: str) -> bool:
     return "=" not in text
 
 
-def solve_problem(text: str) -> dict:
-    text = text.strip()
-
-    # 1️⃣ FAST PATH: simple arithmetic
-    if is_expression(text):
-        try:
-            result = eval(text, {"__builtins__": {}})
-            return {
-                "result": result,
-                "method": "arithmetic"
-            }
-        except Exception:
-            pass  # fallback to symbolic
-
-    # 2️⃣ SYMBOLIC SOLVER (equations)
+def solve_problem(text: str):
+    # 1️⃣ Try symbolic solve FIRST
     symbolic = try_sympy_solve(text)
+
     if symbolic is not None:
         return {
-            "result": symbolic,
+            "result": str(symbolic),
             "method": "symbolic"
         }
 
-    # 3️⃣ RAG + LLM fallback
-    context = retrieve_context(text)
+    # 2️⃣ Fallback to LLM ONLY if symbolic fails
     llm_answer = generate(
-        f"Answer the math problem step by step.\n\nContext:\n{context}\n\nProblem:\n{text}"
+        f"""
+Solve the following math problem.
+Return ONLY the final answer.
+
+Problem:
+{text}
+
+Final Answer:
+""",
+        max_new_tokens=50,
+        temperature=0.0
     )
 
     return {
-        "result": llm_answer,
+        "result": llm_answer.strip(),
         "method": "llm"
-    }
+    }