Update app.py

app.py

@@ -9,21 +9,33 @@ HF_TOKEN = os.getenv("HF_TOKEN")
 
 # --- Tools ---
 class WebSearchTool(Tool):
-    """Search the web using DuckDuckGo"""
     name = "web_search"
     description = "Search the web and return concise results. Input: search query string."
+    inputs = {
+        "query": {
+            "type": "string",
+            "description": "The search query to look up on the web"
+        }
+    }
+    output_type = "string"
     
-    def forward(self, query: str):
+    def forward(self, query: str) -> str:
         from smolagents import DuckDuckGoSearchTool
         tool = DuckDuckGoSearchTool()
         return tool.forward(query)
 
 class WikipediaTool(Tool):
-    """Fetch summaries from Wikipedia"""
     name = "wikipedia_search"
     description = "Fetch Wikipedia summary for a topic. Input: topic string."
+    inputs = {
+        "topic": {
+            "type": "string",
+            "description": "The topic to search for on Wikipedia"
+        }
+    }
+    output_type = "string"
     
-    def forward(self, topic: str):
+    def forward(self, topic: str) -> str:
         try:
             summary = wikipedia.summary(topic, sentences=3)
             return summary
@@ -31,11 +43,17 @@ class WikipediaTool(Tool):
             return f"Wikipedia lookup failed: {e}"
 
 class WeatherTool(Tool):
-    """Get weather using Open-Meteo API"""
     name = "weather"
-    description = "Get current weather. Input: city name."
+    description = "Get current weather for a city. Input: city name."
+    inputs = {
+        "city": {
+            "type": "string",
+            "description": "The name of the city to get weather for"
+        }
+    }
+    output_type = "string"
     
-    def forward(self, city: str):
+    def forward(self, city: str) -> str:
         try:
             # Geocoding to get coordinates
             geocode_url = f"https://geocoding-api.open-meteo.com/v1/search?name={city}"