Spaces:
Runtime error
Runtime error
File size: 2,315 Bytes
105b369 |
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 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 |
import json
from typing import Any, Optional
from phi.tools import Toolkit
from phi.utils.log import logger
try:
from duckduckgo_search import DDGS
except ImportError:
raise ImportError("`duckduckgo-search` not installed. Please install using `pip install duckduckgo-search`")
class DuckDuckGo(Toolkit):
def __init__(
self,
search: bool = True,
news: bool = True,
fixed_max_results: Optional[int] = None,
headers: Optional[Any] = None,
proxy: Optional[str] = None,
proxies: Optional[Any] = None,
timeout: Optional[int] = 10,
):
super().__init__(name="duckduckgo")
self.headers: Optional[Any] = headers
self.proxy: Optional[str] = proxy
self.proxies: Optional[Any] = proxies
self.timeout: Optional[int] = timeout
self.fixed_max_results: Optional[int] = fixed_max_results
if search:
self.register(self.duckduckgo_search)
if news:
self.register(self.duckduckgo_news)
def duckduckgo_search(self, query: str, max_results: int = 5) -> str:
"""Use this function to search DuckDuckGo for a query.
Args:
query(str): The query to search for.
max_results (optional, default=5): The maximum number of results to return.
Returns:
The result from DuckDuckGo.
"""
logger.debug(f"Searching DDG for: {query}")
ddgs = DDGS(headers=self.headers, proxy=self.proxy, proxies=self.proxies, timeout=self.timeout)
return json.dumps(ddgs.text(keywords=query, max_results=(self.fixed_max_results or max_results)), indent=2)
def duckduckgo_news(self, query: str, max_results: int = 5) -> str:
"""Use this function to get the latest news from DuckDuckGo.
Args:
query(str): The query to search for.
max_results (optional, default=5): The maximum number of results to return.
Returns:
The latest news from DuckDuckGo.
"""
logger.debug(f"Searching DDG news for: {query}")
ddgs = DDGS(headers=self.headers, proxy=self.proxy, proxies=self.proxies, timeout=self.timeout)
return json.dumps(ddgs.news(keywords=query, max_results=(self.fixed_max_results or max_results)), indent=2)
|