tools/weather_tool.py

from smolagents import Tool
import random


class WeatherInfoTool(Tool):

    name = "weather_info"

    description = (
        "Fetches weather information for a given location. "
        "Use this tool when you need to know weather conditions "
        "before planning outdoor events like fireworks."
    )

    inputs = {
        "location": {
            "type": "string",
            "description": "The location to get weather information for."
        }
    }

    output_type = "string"


    def forward(self, location: str):

        weather_conditions = [
            {"condition": "Rainy", "temp_c": 15},
            {"condition": "Clear", "temp_c": 25},
            {"condition": "Windy", "temp_c": 20}
        ]

        data = random.choice(weather_conditions)

        return (
            f"Weather in {location}: "
            f"{data['condition']}, "
            f"{data['temp_c']}°C"
        )


weather_info_tool = WeatherInfoTool()