Change final answer and prompt

app.py

@@ -18,28 +18,34 @@ class BasicAgent:
         self.graph = build_graph()
         print("BasicAgent initialized.")
     def __call__(self, question: str) -> str:
+        # print(f"Agent received question (first 50 chars): {question[:50]}...")
+        # messages = [HumanMessage(content=question)]
+        # try:
+        #     result = self.graph.invoke({"messages": messages})
+
+        #     if not result.get("messages") or len(result["messages"]) == 0:
+        #         raise ValueError("No messages returned from the graph")
+
+        #     last_message = result["messages"][-1].content
+
+        #     match = re.search(r"(?i)final\s*answer\s*[:\-]?\s*(.+)", last_message, re.DOTALL)
+        #     if match:
+        #         final_answer = match.group(1).strip()
+        #     else:
+        #        # Try to find a name or string answer
+        #         candidates = [line.strip() for line in last_message.split("\n") if line.strip()]
+        #         final_answer = next((line for line in reversed(candidates) if len(line.split()) <= 5), last_message.strip())
+
+        #     return final_answer
+        # except Exception as e: 
+        #     print("Agent processing failed")
+        #     return
         print(f"Agent received question (first 50 chars): {question[:50]}...")
+        # Wrap the question in a HumanMessage from langchain_core
         messages = [HumanMessage(content=question)]
-        try:
-            result = self.graph.invoke({"messages": messages})
-
-            if not result.get("messages") or len(result["messages"]) == 0:
-                raise ValueError("No messages returned from the graph")
-
-            last_message = result["messages"][-1].content
-
-            match = re.search(r"(?i)final\s*answer\s*[:\-]?\s*(.+)", last_message, re.DOTALL)
-            if match:
-                final_answer = match.group(1).strip()
-            else:
-               # Try to find a name or string answer
-                candidates = [line.strip() for line in last_message.split("\n") if line.strip()]
-                final_answer = next((line for line in reversed(candidates) if len(line.split()) <= 5), last_message.strip())
-                
-            return final_answer
-        except Exception as e: 
-            print("Agent processing failed")
-            return
+        messages = self.graph.invoke({"messages": messages})
+        answer = messages['messages'][-1].content
+        return answer[14:]
             
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):

system_prompt.txt

@@ -1,13 +1,5 @@
-You are a helpful and precise AI assistant. I will ask you a question.
-First, think through the answer step by step. Then, conclude your response using the following format:
-FINAL ANSWER: [your final answer]
-Always provide a FINAL ANSWER, even if you're uncertain — give your best guess. If no valid answer can be found, respond with:
-FINAL ANSWER: I cannot answer that
-Formatting rules for the final answer:
-If the answer is a number, write it without commas or units (e.g. 1000000 instead of 1,000,000 or $1M), unless explicitly told otherwise.
-If the answer is a string, write it as concisely as possible:
-Do not use articles ("a", "an", "the")
-Do not use abbreviations (e.g. write “New York” instead of “NY”)
-Write digits in plain text (e.g. "twenty three") unless numeric form is explicitly required
-If the answer is a comma-separated list, apply the above rules to each item in the list.
-Do not include any text after the FINAL ANSWER line.
+You are a helpful assistant tasked with answering questions using a set of tools. 
+Now, I will ask you a question. Report your thoughts, and finish your answer with the following template: 
+FINAL ANSWER: [YOUR FINAL ANSWER]. 
+YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.
+Your answer should only start with "FINAL ANSWER: ", then follows with the answer.