feat: get_weather tool

.gitignore

@@ -1,2 +1,3 @@
 /.idea
-.venv/
+.venv/
+*/__pycache__/

app.py

@@ -1,8 +1,9 @@
-from dataclasses import asdict
-from smolagents import CodeAgent, TransformersModel, InferenceClientModel, stream_to_gradio
 import torch
 import gradio as gr
+from dataclasses import asdict
+from smolagents import CodeAgent, TransformersModel, InferenceClientModel, stream_to_gradio
 from transformers import BitsAndBytesConfig
+from tools import get_weather
 
 
 model_path = "Qwen/Qwen3-4B-Instruct-2507"
@@ -24,9 +25,9 @@ else:
     )
 
 agent = CodeAgent(
-    tools=[],
+    tools=[get_weather],
     model=model,
-    max_steps=6,
+    max_steps=8,
     verbosity_level=2,
     add_base_tools=True
 )

requirements.txt

@@ -13,6 +13,7 @@ fastapi==0.118.0
 ffmpy==0.6.1
 filelock==3.19.1
 fsspec==2025.9.0
+geojson==3.2.0
 gradio==5.49.0
 gradio_client==1.13.3
 groovy==0.1.2
@@ -57,6 +58,8 @@ pydantic==2.11.10
 pydantic_core==2.33.2
 pydub==0.25.1
 Pygments==2.19.2
+pyowm==3.5.0
+PySocks==1.7.1
 python-dateutil==2.9.0.post0
 python-dotenv==1.1.1
 python-multipart==0.0.20

tools/__init__.py

@@ -0,0 +1 @@
+from .weather import get_weather

tools/weather.py

@@ -0,0 +1,32 @@
+from pyowm import OWM
+from smolagents import tool
+import os
+
+
+@tool
+def get_weather(city: str, nation: str) -> str:
+    """
+    This tool returns the current weather for one specific city.
+
+    Args:
+        city: The city to search the weather.
+        nation: The nation where the city is located.
+    """
+    API_KEY = os.getenv("OPENWEATHER_API_KEY")
+    owm = OWM(API_KEY)
+    mgr = owm.weather_manager()
+
+    location = f'{city}, {nation}'
+
+    observation = mgr.weather_at_place(location)
+    w = observation.weather
+
+    temperature = w.temperature('celsius')['temp']
+    status = w.detailed_status
+    humidity = w.humidity
+
+    return f"Location: {location}\nStatus: {status}\nTemperature: {temperature}\nHumidity: {humidity}%"
+
+
+if __name__ == "__main__":
+    print(get_weather('Sao Paulo', 'BR'))