Update app.py

app.py

@@ -262,7 +262,8 @@ class BasicAgent:
         
         return [
             # web_search, 
-            wikipedia_search, python_calculator, simple_math
+            wikipedia_search, python_calculator
+            # , simple_math
             # , file_analyzer
         ]
     
@@ -312,35 +313,21 @@ class BasicAgent:
                 "action_type": "final"
             }
         
-        planning_prompt = f"""You are a GAIA benchmark specialist. Your task is to solve this question with MAXIMUM ACCURACY.
-
-QUESTION: {messages[0]['content'] if messages else 'No question provided'}
-
-EXECUTION HISTORY: {plan_history}
-
-CRITICAL SUCCESS FACTORS:
-1. PRECISION: GAIA answers must be EXACT - no approximations, no explanations
-2. STEP EFFICIENCY: Use minimal steps (typically 1-3 for Level 1)
-3. TOOL SELECTION: Choose the RIGHT tool for each specific task
-
-AVAILABLE TOOLS:
-- web_search: Current/recent information, news, live data
-- wikipedia_search: Established facts, biographical data, historical info  
-- python_calculator: Complex calculations, data processing, file operations
-- simple_math: Quick arithmetic, basic math functions
-- file_analyzer: Examine uploaded files
-
-PLANNING STRATEGY:
-1. Identify the EXACT answer format needed (number, name, date, etc.)
-2. Determine the specific information required
-3. Choose the BEST tool for that information type
-4. Plan for verification if needed
-
-RESPONSE FORMAT:
-If you need to use a tool: "EXECUTE: [tool_name] | INPUT: [specific_input] | GOAL: [what_you_expect]"
-If you have the final answer: "FINAL: [exact_answer_only]"
-
-Be extremely specific in your tool inputs. Avoid vague searches."""
+        planning_prompt = f"""
+                                You are a general AI assistant. 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.
+
+                                QUESTION: {messages[0]['content'] if messages else 'No question provided'}
+                                
+                                EXECUTION HISTORY: {plan_history}
+                            """
 
         response = self.model.invoke([{"role": "system", "content": planning_prompt}])
         content = response.content.strip()
@@ -399,7 +386,7 @@ Be extremely specific in your tool inputs. Avoid vague searches."""
             # "web_search": ["web", "search", "google", "internet"],
             "wikipedia_search": ["wiki", "wikipedia"],  
             "python_calculator": ["python", "code", "calc", "calculate"],
-            "simple_math": ["math", "arithmetic"],
+            # "simple_math": ["math", "arithmetic"],
             # "file_analyzer": ["file", "analyze"]
         }