Deploy GAIA agent

app.py

@@ -22,24 +22,26 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Enhanced Tools for GAIA ---
 
+from smolagents import tool
 from duckduckgo_search import DDGS
 
+@tool
 def web_search_tool(query: str) -> str:
     """
-    Performs a web search using DuckDuckGo.
+    Perform a web search using DuckDuckGo and return top results.
 
     Args:
-        query (str): The search query string.
+        query (str): Search query.
 
     Returns:
-        str: A formatted string of search results.
+        str: Formatted search result string.
     """
     try:
         with DDGS() as ddgs:
-            results = ddgs.text(query, max_results=5)
+            results = ddgs.text(query, max_results=3)
             output = []
-            for result in results:
-                output.append(f"Title: {result['title']}\nURL: {result['href']}\nSnippet: {result['body']}")
+            for r in results:
+                output.append(f"Title: {r['title']}\nURL: {r['href']}\nSnippet: {r['body']}")
             return "\n\n".join(output)
     except Exception as e:
         return f"Web search failed: {str(e)}"