changes

app.py

@@ -28,11 +28,16 @@ class BasicAgent:
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
         messages = [HumanMessage(content=question)]
-        messages = self.graph.invoke({"messages": messages})
-        raw_answer = messages['messages'][-1].content
-        if raw_answer.startswith("FINAL ANSWER: "):
-            return raw_answer[len("FINAL ANSWER: "):].strip()
-        return f"Agent response did not follow FINAL ANSWER format: {raw_answer}"
+        result = self.graph.invoke({"messages": messages})
+        full_output = result["messages"][-1].content.strip()
+
+        # Enforce strict FINAL ANSWER format parsing
+        match = re.search(r"FINAL ANSWER:\s*(.+)", full_output, re.IGNORECASE)
+        if match:
+            return match.group(0).strip()  # Returns the entire "FINAL ANSWER: xxx"
+        else:
+            print(" FINAL ANSWER not found in output, returning fallback.")
+            return "FINAL ANSWER: unknown"
 
 def run_agent_only(profile: gr.OAuthProfile | None):
     global cached_answers

system_prompt.txt

@@ -1,18 +1,11 @@
-You are a helpful assistant answering questions using a set of tools.
+You are a helpful assistant tasked with answering questions using a set of tools.
 
-You must strictly follow this output format:
+You MUST strictly follow this rule: YOUR FINAL ANSWER MUST be the last line of your response and must follow this exact format:
 
-FINAL ANSWER: [YOUR FINAL ANSWER]
+FINAL ANSWER: [your answer here]
 
-Where [YOUR FINAL ANSWER] is:
-- A number (e.g., 42) → Do NOT use commas, units ($, %, etc.), or extra words.
-- A string (e.g., Paris) → Do NOT use articles (e.g., "the", "an"), abbreviations, or numeric digits unless required.
-- A comma-separated list (e.g., apple, banana, cherry) → Apply the above rules to each item.
+- Your final answer should be a **number** OR as few words as possible OR a comma-separated list.
+- Do not include explanations, markdown, or any additional text after FINAL ANSWER.
+- If the answer is a string, do not include articles (e.g., "the", "a") or abbreviations. Write digits in full words if requested.
 
-Important:
-- Always begin your final output with **exactly** "FINAL ANSWER: ".
-- Do NOT include any reasoning or explanation after your final answer.
-- Do NOT add anything after the period.
-- Think step-by-step internally, but return **only** the FINAL ANSWER line in your output.
-
-I will now ask you a question.
+Any answer that does not follow the `FINAL ANSWER: ...` format exactly will be considered incorrect.