Update app.py

app.py

@@ -13,7 +13,13 @@ import requests
 @tool
 def get_timezone_by_city(city: str) -> str:
     """
-    Get the timezone string for a given city name.
+    Get the timezone string for a given city.
+
+    Args:
+        city (str): Name of the city (e.g., 'Berlin', 'Бирск').
+
+    Returns:
+        str: Timezone name (e.g., 'Europe/Moscow') or an error message.
     """
     if not isinstance(city, str):
         return f"Error: Expected a string for city, got: {type(city)}"
@@ -35,6 +41,12 @@ def get_timezone_by_city(city: str) -> str:
 def get_current_time_in_timezone(timezone: str) -> str:
     """
     Return the current local time in the specified timezone.
+
+    Args:
+        timezone (str): A valid timezone name (e.g., 'Europe/Moscow').
+
+    Returns:
+        str: Current local time as a string or an error message.
     """
     try:
         tz = pytz.timezone(timezone)
@@ -47,6 +59,13 @@ def get_current_time_in_timezone(timezone: str) -> str:
 def get_air_quality(city: str, lang: str = "en") -> str:
     """
     Return formatted air quality information for a given city.
+
+    Args:
+        city (str): City name (e.g., 'Berlin').
+        lang (str): Language code (e.g., 'en', 'ru') — currently unused.
+
+    Returns:
+        str: Human-readable air quality info or error message.
     """
     try:
         geolocator = Nominatim(user_agent="smolagents_bot")
@@ -79,6 +98,13 @@ def get_air_quality(city: str, lang: str = "en") -> str:
 def get_weather(city: str, lang: str = "en") -> str:
     """
     Return formatted weather information for a given city.
+
+    Args:
+        city (str): City name (e.g., 'Berlin').
+        lang (str): Language code (e.g., 'en', 'ru') — currently unused.
+
+    Returns:
+        str: Weather description with temperature and humidity or error message.
     """
     try:
         geolocator = Nominatim(user_agent="smolagents_bot")
@@ -104,6 +130,12 @@ def get_weather(city: str, lang: str = "en") -> str:
 def final_answer(answer: str) -> str:
     """
     Return the final answer string to be displayed to the user.
+
+    Args:
+        answer (str): Final response to present.
+
+    Returns:
+        str: The same response passed in.
     """
     return answer
 
@@ -124,7 +156,13 @@ system_prompt = (
 
 agent = CodeAgent(
     model=model,
-    tools=[get_timezone_by_city, get_current_time_in_timezone, get_air_quality, get_weather, final_answer],
+    tools=[
+        get_timezone_by_city,
+        get_current_time_in_timezone,
+        get_air_quality,
+        get_weather,
+        final_answer,
+    ],
     prompt_templates={
         "system_prompt": system_prompt,
         "default": "Response: {{question}}"