json/dict output type

tools/weather_api.py

@@ -1,4 +1,5 @@
 import os
+from typing import Dict
 
 from smolagents.tools import Tool
 
@@ -6,13 +7,13 @@ class WeatherApiTool(Tool):
     name = "weather_api"
     description = "Fetches the weather for a given location and date."
     inputs = {'location': {'type': 'string', 'description': 'The location for which to fetch the weather.'}, 'date': {'type': 'string', 'description': 'The date for which to fetch the weather.'}}
-    output_type = "string"
+    output_type = "json"
     
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
         self.api_key = os.environ.get("OPEN_WEATHER_MAP_API_KEY")
 
-    def forward(self, location: str, date: str) -> str:
+    def forward(self, location: str, date: str) -> Dict:
         import requests
         import json
         
@@ -49,5 +50,5 @@ class WeatherApiTool(Tool):
         #         temperature = day["temp"]["day"]
         #         return f"The weather in {location} on {date} is {weather} with a temperature of {temperature}°C."
         
-        return json.dumps(weather_data, indent=4)
+        return weather_data