github/AutoGPT
1
1
Fork 0
mirror of https://github.com/Significant-Gravitas/AutoGPT.git synced 2026-04-08 03:00:28 -04:00
Code Issues Packages Projects Releases Wiki Activity

Fix web_search tests

Browse Source
This commit is contained in:
Reinier van der Leer
2023-10-17 17:07:37 -07:00
parent 7e7365eac2
commit 3cc8b2abf7
2 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
autogpts/autogpt/autogpt/commands/web_search.py
View File

@@ -58,7 +58,11 @@ def web_search(query: str, agent: Agent, num_results: int = 8) -> str:
attempts += 1
search_results = [
{"title": r["title"], "url": r["href"], "description": r["body"]}
{
"title": r["title"],
"url": r["href"],
**({"description": r["body"]} if r.get("body") else {}),
}
for r in search_results
]

4
autogpts/autogpt/tests/unit/test_web_search.py
View File

@@ -29,8 +29,8 @@ def test_safe_google_results_invalid_input():
(
"test",
1,
'[\n {\n "title": "Result 1",\n "link": "https://example.com/result1"\n }\n]',
[{"title": "Result 1", "link": "https://example.com/result1"}],
'[\n {\n "title": "Result 1",\n "url": "https://example.com/result1"\n }\n]',
[{"title": "Result 1", "href": "https://example.com/result1"}],
),
("", 1, "[]", []),
("no results", 1, "[]", []),