diff --git a/llama-index-integrations/tools/llama-index-tools-duckduckgo/llama_index/tools/duckduckgo/base.py b/llama-index-integrations/tools/llama-index-tools-duckduckgo/llama_index/tools/duckduckgo/base.py index 8555011b005023eb1f5b73d3e21d79c12dd550eb..ff5a8470924daee44477241b215eefdf8cb79692 100644 --- a/llama-index-integrations/tools/llama-index-tools-duckduckgo/llama_index/tools/duckduckgo/base.py +++ b/llama-index-integrations/tools/llama-index-tools-duckduckgo/llama_index/tools/duckduckgo/base.py @@ -28,7 +28,10 @@ class DuckDuckGoSearchToolSpec(BaseToolSpec): return list(ddg.answers(query)) def duckduckgo_full_search( - self, query: str, region: Optional[str] = None, max_results: Optional[int] = 10 + self, + query: str, + region: Optional[str] = "wt-wt", + max_results: Optional[int] = 10, ) -> List[Dict]: """ Make a query to DuckDuckGo search to receive a full search results. @@ -40,5 +43,11 @@ class DuckDuckGoSearchToolSpec(BaseToolSpec): """ from duckduckgo_search import DDGS + params = { + "keywords": query, + "region": region, + "max_results": max_results, + } + with DDGS() as ddg: - return list(ddg.text(query, region=region, max_results=max_results)) + return list(ddg.text(**params)) diff --git a/llama-index-integrations/tools/llama-index-tools-duckduckgo/pyproject.toml b/llama-index-integrations/tools/llama-index-tools-duckduckgo/pyproject.toml index 413597213b1ba9f6d3bd330a040728a78ec11e63..0defe0d8cdab948fe108b702bf20ce709d838e76 100644 --- a/llama-index-integrations/tools/llama-index-tools-duckduckgo/pyproject.toml +++ b/llama-index-integrations/tools/llama-index-tools-duckduckgo/pyproject.toml @@ -29,12 +29,12 @@ license = "MIT" maintainers = ["leehuwuj"] name = "llama-index-tools-duckduckgo" readme = "README.md" -version = "0.1.0" +version = "0.1.1" [tool.poetry.dependencies] -python = ">=3.8.1,<3.12" +python = ">=3.8.1,<4.0" llama-index = "^0.10.1" -duckduckgo-search = "4.5.0" +duckduckgo-search = "^6.1.0" [tool.poetry.group.dev.dependencies] ipython = "8.10.0"