Update agent.py

agent.py

@@ -32,25 +32,25 @@ def build_agent():
         }
 
     def enhancer(state: AgentState):
-        sys_msg = """
-                            You are a helpful assistant tasked with answering questions using a set of tools.  You can reason through problems and take actions using available tools to solve complex tasks.
-                            Follow this format strictly:
-                            Message: {user question}
-                            Thought: Describe your reasoning about the question.
-                            Action: Call one of the available tools with a specific input.
-                            Observation: Note what was returned from the tool.
-                            Repeat the Thought → Action → Observation steps as many times as necessary until you have enough information to answer the original question.
-                            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.
-                        """
+        sys_msg = """You are a reasoning assistant that can answer complex questions using external tools. You must follow a structured reasoning format and always output a final answer.
+                            Always follow this step-by-step structure:
+                            
+                            - Thought: Describe your reasoning.
+                            - Action: Call a tool, using this format → Action: tool_name(input)
+                            - Observation: Describe what the tool returned.
+                            - Repeat Thought → Action → Observation as needed.
+                            
+                            When you are ready to answer, say:
+                            
+                            FINAL ANSWER: [your final answer here]
+                            
+                            Formatting rules for FINAL ANSWER:
+                            - If a number: no commas, no units unless explicitly asked.
+                            - If a string: no articles, no abbreviations, digits in plain text.
+                            - If a comma-separated list: apply the above rules to each element."""
         return {
                 "messages": [{"role": "system", "content": sys_msg}] + state["messages"],
-                "wiki_table": state.get("wiki_table")
+                "wiki_table": None
             }
 
     ## The graph