Update tools/web_search.py

tools/web_search.py

@@ -1,26 +1,26 @@
-from typing import Any, Optional
+from typing import Any
 from smolagents.tools import Tool
 from ddgs import DDGS
 
+
 class DuckDuckGoSearchTool(Tool):
     name = "web_search"
-    description = "Performs a duckduckgo web search based on your query and returns the top results."
-    inputs: Dict[str, Dict[str, Any]] = {
-        'query': {'type': 'string', 'description': 'The search query to perform.'}
+    description = "Performs a DuckDuckGo web search based on your query and returns the top results."
+    inputs: dict[str, dict[str, Any]] = {
+        "query": {"type": "string", "description": "The search query to perform."}
     }
-    output_type = "string"
+    output_type = "string"  # you can switch to "json" if your smolagents version supports it
 
     def __init__(self, max_results: int = 10, **kwargs: Any) -> None:
         super().__init__()
         self.max_results = max_results
         self.ddgs = DDGS(**kwargs)
 
-    def forward(self, query: str) -> str:
+    def forward(self, query: str) -> list[dict[str, str]]:
         q = (query or "").strip()
         if not q:
             raise ValueError("Query must be a non-empty string.")
-    
-        # 1) ddgs.text: correct args + materialize generator
+
         items = list(
             self.ddgs.text(
                 keywords=q,
@@ -30,27 +30,24 @@ class DuckDuckGoSearchTool(Tool):
                 max_results=self.max_results,
             )
         )
-    
-        # 3) Better emptiness check + error type
+
         if not items:
             raise RuntimeError("No results found. Try a broader/simpler query.")
-    
-        results: List[Dict[str, str]] = []
+
+        results: list[dict[str, str]] = []
         for it in items:
             href = (it.get("href") or "").strip()
             if not href:
-                continue  # skip malformed entries without a URL
+                continue
             results.append(
                 {
                     "title": str(it.get("title", "")),
-                    "url": href,                       # normalized key name
+                    "url": href,
                     "snippet": str(it.get("body", "")),
                 }
             )
-  
-        
+
         if not results:
             raise RuntimeError("Results were returned but none contained valid URLs.")
-    
-        return results
 
+        return results