Skip to content
Snippets Groups Projects
Unverified Commit 8d175082 authored by Huu Le (Lee)'s avatar Huu Le (Lee) Committed by GitHub
Browse files

Integration: Bump duckduckgo-search package (#13583)

parent 993d6094
No related branches found
No related tags found
No related merge requests found
...@@ -28,7 +28,10 @@ class DuckDuckGoSearchToolSpec(BaseToolSpec): ...@@ -28,7 +28,10 @@ class DuckDuckGoSearchToolSpec(BaseToolSpec):
return list(ddg.answers(query)) return list(ddg.answers(query))
def duckduckgo_full_search( 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]: ) -> List[Dict]:
""" """
Make a query to DuckDuckGo search to receive a full search results. Make a query to DuckDuckGo search to receive a full search results.
...@@ -40,5 +43,11 @@ class DuckDuckGoSearchToolSpec(BaseToolSpec): ...@@ -40,5 +43,11 @@ class DuckDuckGoSearchToolSpec(BaseToolSpec):
""" """
from duckduckgo_search import DDGS from duckduckgo_search import DDGS
params = {
"keywords": query,
"region": region,
"max_results": max_results,
}
with DDGS() as ddg: with DDGS() as ddg:
return list(ddg.text(query, region=region, max_results=max_results)) return list(ddg.text(**params))
...@@ -29,12 +29,12 @@ license = "MIT" ...@@ -29,12 +29,12 @@ license = "MIT"
maintainers = ["leehuwuj"] maintainers = ["leehuwuj"]
name = "llama-index-tools-duckduckgo" name = "llama-index-tools-duckduckgo"
readme = "README.md" readme = "README.md"
version = "0.1.0" version = "0.1.1"
[tool.poetry.dependencies] [tool.poetry.dependencies]
python = ">=3.8.1,<3.12" python = ">=3.8.1,<4.0"
llama-index = "^0.10.1" llama-index = "^0.10.1"
duckduckgo-search = "4.5.0" duckduckgo-search = "^6.1.0"
[tool.poetry.group.dev.dependencies] [tool.poetry.group.dev.dependencies]
ipython = "8.10.0" ipython = "8.10.0"
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment