Update tools.py

tools.py

@@ -80,30 +80,30 @@ def create_search_tool() -> Graph:
     """Creates a search tool using DuckDuckGo that can search for information online."""
     
     def search_function(state: SearchState) -> dict:
-        # Initialize DuckDuckGo search
         with DDGS() as ddgs:
-            # Perform the search
-            search_results = list(ddgs.text(
+            # Run search
+            raw_results = list(ddgs.text(
                 state.input.query,
                 max_results=state.input.max_results
             ))
-            
-            # Convert results to our model
-            results = [
-                SearchResult(
-                    title=result["title"],
-                    link=result["link"],
-                    snippet=result["body"]
-                )
-                for result in search_results
-            ]
-            
-            return {
-                "output": SearchOutput(
-                    results=results,
-                    query=state.input.query
-                )
-            }
+
+        results = []
+        for r in raw_results:
+            try:
+                results.append(SearchResult(
+                    title=r.get("title", ""),
+                    link=r.get("href", r.get("link", "")),  # DuckDuckGo sometimes uses "href"
+                    snippet=r.get("body", r.get("snippet", ""))
+                ))
+            except Exception as e:
+                print("Skipping malformed search result:", r, "Error:", e)
+
+        return {
+            "output": SearchOutput(
+                results=results,
+                query=state.input.query
+            )
+        }
 
     # Create the graph with state schema
     workflow = StateGraph(state_schema=SearchState)