Update tools

app.py

@@ -19,12 +19,25 @@ def my_search_tool(search_query: str) -> str:
     try:
         if not search_query or not search_query.strip():
             search_query = "latest popular arxiv papers in reinforcement learning"
-        result = _search_tool.run(search_query=search_query)
+        result = _search_tool(search_query)
     except Exception as e:
         result = f"An error occurred while performing the search, error: {e}."
         print(f"Error occurred: {e}")
     return result
 
+@tool
+def get_weather(city: str) -> str:
+    """Get the current weather for a city.
+    Args:
+        city: the name of the city (e.g., 'London', 'Shanghai')
+    """
+    try:
+        response = requests.get(f"https://wttr.in/{city}?format=3", timeout=5)
+        response.raise_for_status()
+        return response.text.strip()
+    except Exception as e:
+        return f"Error fetching weather for '{city}': {e}"
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -62,7 +75,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, my_search_tool, get_current_time_in_timezone], ## add your tools here (don't remove final answer)
+    tools=[final_answer, my_search_tool, get_weather, get_current_time_in_timezone], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,