| 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}" |
