from smolagents import Tool

from rest_clients.open_weather_client import OpenWeatherClient


class WeatherInfoTool(Tool):
    name = "weather_info"
    description = "Fetches real weather information for a given location using OpenWeatherMap."
    inputs = {
        "location": {
            "type": "string",
            "description": "The location to get weather information for."
        }
    }
    output_type = "string"

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.weather_client = OpenWeatherClient()

    def forward(self, location: str):
        return self.weather_client.get_weather(location)