Spaces:
Sleeping
Sleeping
File size: 1,875 Bytes
fd18d93 8fe992b fa65674 8fe992b fa65674 8fe992b 79a8ee7 8fe992b 79a8ee7 8fe992b fa65674 8fe992b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 | from typing import Any, Optional
from smolagents.tools import Tool
import duckduckgo_search
class DuckDuckGoSearchTool(Tool):
name = "web_search"
description = "Performs a duckduckgo web search based on your query (think a Google search) then returns the top search results."
inputs = {
'query': {'type': 'string', 'description': 'The search query to perform.'},
'max_results': {
'type': 'integer',
'description': 'Maximum number of search results to return',
'nullable': True
}
}
output_type = "string"
def __init__(self, max_results=10, **kwargs):
super().__init__()
self.default_max_results = max_results
try:
from duckduckgo_search import DDGS
except ImportError as e:
raise ImportError(
"You must install package `duckduckgo_search` to run this tool: for instance run `pip install duckduckgo-search`."
) from e
self.ddgs = DDGS(**kwargs)
def forward(self, query: str, max_results: int = None) -> str:
"""Performs a web search
Args:
query: The search query to perform
max_results: Maximum number of results to return. If not specified, uses the value from __init__
Returns:
String containing formatted search results
"""
results_limit = max_results if max_results is not None else self.default_max_results
results = self.ddgs.text(query, max_results=results_limit)
if len(results) == 0:
raise Exception(
"No results found! Try a less restrictive/shorter query.")
postprocessed_results = [
f"[{result['title']}]({result['href']})\n{result['body']}" for result in results]
return "## Search Results\n\n" + "\n\n".join(postprocessed_results)
|