Update app.py

app.py

@@ -1,12 +1,11 @@
 from smolagents import CodeAgent, DuckDuckGoSearchTool, load_tool, tool
-from smolagents.models import HuggingFaceHubModel
+from smolagents.models import HuggingFaceLLM  # ✅ unified model class in latest versions
 import datetime
 import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
 
-# Weather-specific imports
 import openmeteo_requests
 import pandas as pd
 import requests_cache
@@ -25,12 +24,10 @@ def hourly_temperature(latitude: float, longitude: float) -> str:
         longitude: Longitude of the location.
     """
     try:
-        # Setup API client with cache and retries
         cache_session = requests_cache.CachedSession('.cache', expire_after=3600)
         retry_session = retry(cache_session, retries=5, backoff_factor=0.2)
         openmeteo = openmeteo_requests.Client(session=retry_session)
 
-        # Request weather data
         url = "https://api.open-meteo.com/v1/forecast"
         params = {
             "latitude": latitude,
@@ -40,12 +37,10 @@ def hourly_temperature(latitude: float, longitude: float) -> str:
         responses = openmeteo.weather_api(url, params=params)
         response = responses[0]
 
-        # Process hourly data
         hourly = response.Hourly()
         hourly_temperature_2m = hourly.Variables(0).ValuesAsNumpy()
         timestamps = pd.to_datetime(hourly.Time(), unit="s")
 
-        # Prepare summary
         data = pd.DataFrame({
             "time": timestamps,
             "temperature_2m": hourly_temperature_2m
@@ -60,10 +55,7 @@ def hourly_temperature(latitude: float, longitude: float) -> str:
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
-    """Fetch the current local time in a specified timezone.
-    Args:
-        timezone: A string representing a valid timezone (e.g., 'Africa/Kigali').
-    """
+    """Fetch the current local time in a specified timezone."""
     try:
         tz = pytz.timezone(timezone)
         local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
@@ -76,16 +68,14 @@ def get_current_time_in_timezone(timezone: str) -> str:
 
 final_answer = FinalAnswerTool()
 
-model = HuggingFaceHubModel(
+model = HuggingFaceLLM(
     model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
     temperature=0.5,
-    max_tokens=2048
+    max_new_tokens=2048
 )
 
-# Import remote tool example (text-to-image)
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
-# Load prompt templates
 with open("prompts.yaml", "r") as stream:
     prompt_templates = yaml.safe_load(stream)
 
@@ -98,10 +88,8 @@ agent = CodeAgent(
 )
 
 
-# ===== Test Tools =====
+# ===== Run the Agent =====
 if __name__ == "__main__":
-    print(hourly_temperature(-1.95, 30.05))  # Kigali example
+    print(hourly_temperature(-1.95, 30.05))
     print(get_current_time_in_timezone("Africa/Kigali"))
-
-    # Launch Gradio UI
     GradioUI(agent).launch()