Create web_search.py

web_search.py

@@ -0,0 +1,56 @@
+from smolagents import Tool
+from ddgs import DDGS
+
+
+class WebSearchTool(Tool):
+    name = "web_search"
+
+    description = (
+        "Searches the web for relevant information. "
+        "Use this to find sources, facts, dates, statistics, "
+        "and other information needed to answer the question. "
+        "Search results include titles, URLs, and snippets."
+    )
+
+    inputs = {
+        "query": {
+            "type": "string",
+            "description": "The search query to run.",
+        }
+    }
+
+    output_type = "string"
+
+    def __init__(self, max_results=8):
+        super().__init__()
+        self.max_results = max_results
+
+    def forward(self, query: str) -> str:
+        try:
+            results = list(
+                DDGS().text(
+                    query,
+                    max_results=self.max_results,
+                )
+            )
+
+            if not results:
+                return "No search results found."
+
+            output = []
+
+            for result in results:
+                title = result.get("title", "Untitled")
+                url = result.get("href", "")
+                body = result.get("body", "")
+
+                output.append(
+                    f"TITLE: {title}\n"
+                    f"URL: {url}\n"
+                    f"SNIPPET: {body}"
+                )
+
+            return "\n\n".join(output)
+
+        except Exception as e:
+            return f"Search failed: {e}"