fix tz

tools/weather_api.py

@@ -1,6 +1,7 @@
 import os
 from typing import Dict
 from datetime import datetime
+import pytz
 
 from smolagents.tools import Tool
 
@@ -13,6 +14,9 @@ class WeatherApiTool(Tool):
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
         self.api_key = os.environ.get("OPEN_WEATHER_MAP_API_KEY")
+        
+        # TODO: get the timezone from the location
+        self.local_tz = pytz.timezone("Europe/Brussels")
 
     def forward(self, location: str, date: str) -> Dict:
         import requests
@@ -46,7 +50,7 @@ class WeatherApiTool(Tool):
         
         # Extract the weather data for the specified date
         date_obj = datetime.strptime(date, "%Y-%m-%d")
-        date_obj = date_obj.replace(tzinfo="Europe/Brussels")
+        date_obj = self.local_tz.localize(date_obj)
         epoch = date_obj.timestamp()
         print(f"Epoch: {epoch}")
         # for day in weather_data["daily"]: