From 8d175082626fb5da9e5f4b1fed39e5487d5afca1 Mon Sep 17 00:00:00 2001
From: "Huu Le (Lee)" <39040748+leehuwuj@users.noreply.github.com>
Date: Mon, 20 May 2024 09:45:14 +0700
Subject: [PATCH] Integration: Bump duckduckgo-search package (#13583)
---
.../llama_index/tools/duckduckgo/base.py | 13 +++++++++++--
.../llama-index-tools-duckduckgo/pyproject.toml | 6 +++---
2 files changed, 14 insertions(+), 5 deletions(-)
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 8555011b00..ff5a847092 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 413597213b..0defe0d8cd 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"
--
GitLab