Update app.py

app.py

@@ -25,20 +25,22 @@ class BasicAgent:
         weather_info_tool = WeatherInfoTool()
 
         # Create agent
-        tools = [search_tool, weather_info_tool]
-        self.agent= model.bind_tools(tools)
-        #= CodeAgent(
-        #    tools=[weather_info_tool, search_tool], 
-        #    model=model,
-        #    add_base_tools=True,  # Add any additional base tools
-        #    planning_interval=3   # Enable planning every 3 steps
-        #)
+        self.agent= CodeAgent(
+            tools=[weather_info_tool, search_tool], 
+            model=model,
+            add_base_tools=True,  # Add any additional base tools
+            planning_interval=3   # Enable planning every 3 steps
+        )
 
     def __call__(self, question: str) -> str:
+        prompt = """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:
+        """
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        answer = self.agent.invoke(question)
+        answer = self.agent.run(prompt + question)
         print(f"Agent returning fixed answer: {fixed_answer}")
-        return answer
+        return answer.replace("FINAL ANSWER:", "").strip()
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """