Skip to content

Commit 12640f7

Browse files
authored
fix(agent, forge): Conform web_search.py to duckduckgo_search v5 (Significant-Gravitas#7045)
Update `web_search` command for both autogpt and forge to adjust for breaking change in v5 of `duckduckgo_search`
1 parent 5f9cc58 commit 12640f7

File tree

4 files changed

+4
-6
lines changed

4 files changed

+4
-6
lines changed

autogpts/autogpt/autogpt/commands/web_search.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -49,8 +49,7 @@ def web_search(query: str, agent: Agent, num_results: int = 8) -> str:
4949
if not query:
5050
return json.dumps(search_results)
5151

52-
results = DDGS().text(query)
53-
search_results = list(islice(results, num_results))
52+
search_results = DDGS().text(query, max_results=num_results)
5453

5554
if search_results:
5655
break

autogpts/autogpt/pyproject.toml

+1-1
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ colorama = "^0.4.6"
3333
demjson3 = "^3.0.0"
3434
distro = "^1.8.0"
3535
docker = "*"
36-
duckduckgo-search = "^4.0.0"
36+
duckduckgo-search = "^5.0.0"
3737
en-core-web-sm = {url = "https://github.com/explosion/spacy-models/releases/download/en_core_web_sm-3.5.0/en_core_web_sm-3.5.0-py3-none-any.whl"}
3838
fastapi = "^0.109.1"
3939
ftfy = "^6.1.1"

autogpts/forge/forge/actions/web/web_search.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -42,8 +42,7 @@ async def web_search(agent, task_id: str, query: str) -> str:
4242
if not query:
4343
return json.dumps(search_results)
4444

45-
results = DDGS().text(query)
46-
search_results = list(islice(results, num_results))
45+
search_results = DDGS().text(query, max_results=num_results)
4746

4847
if search_results:
4948
break

autogpts/forge/pyproject.toml

+1-1
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ toml = "^0.10.2"
2121
jinja2 = "^3.1.2"
2222
uvicorn = "^0.23.2"
2323
litellm = "^1.17.9"
24-
duckduckgo-search = "^4.0.0"
24+
duckduckgo-search = "^5.0.0"
2525
selenium = "^4.13.0"
2626
bs4 = "^0.0.1"
2727
agbenchmark = { path = "../../benchmark", optional = true }

0 commit comments

Comments
 (0)