| from typing import Any, Optional |
| from smolagents.tools import Tool |
| from ddgs import DDGS |
|
|
| class DuckDuckGoSearchTool(Tool): |
| name = "web_search" |
|
|
| description = ( |
| "Searches the web for relevant sources. " |
| "Returns titles, URLs, and snippets from search results." |
| ) |
|
|
| inputs = { |
| "query": { |
| "type": "string", |
| "description": "The web search query.", |
| } |
| } |
|
|
| output_type = "string" |
|
|
| def __init__(self, max_results=10, **kwargs): |
| super().__init__() |
| self.max_results = max_results |
| self.ddgs = DDGS(**kwargs) |
|
|
| def forward(self, query: str) -> str: |
| try: |
| results = list( |
| self.ddgs.text( |
| query, |
| max_results=self.max_results, |
| ) |
| ) |
|
|
| if not results: |
| return ( |
| f"No search results were returned for: {query}. " |
| "Try a broader or differently worded query." |
| ) |
|
|
| formatted = [] |
|
|
| for result in results: |
| title = result.get("title", "Untitled") |
| url = result.get("href", "") |
| body = result.get("body", "") |
|
|
| formatted.append( |
| f"[{title}]({url})\n{body}" |
| ) |
|
|
| return "## Search Results\n\n" + "\n\n".join(formatted) |
|
|
| except Exception as e: |
| return f"Search failed for '{query}': {e}" |
|
|
