aljaz/dev-gpt
1
0
Fork 0
mirror of https://github.com/aljazceru/dev-gpt.git synced 2025-12-21 15:44:19 +01:00
Code Issues Packages Projects Releases Wiki Activity

🔎 feat: search fix web search

Browse Source
This commit is contained in:
Florian Hönicke
2023-05-18 20:08:41 +02:00
parent 3101bd1978
commit e1f0205663
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
dev_gpt/options/generate/static_files/microservice/apis.py
View File

@@ -38,7 +38,7 @@ def google_search(search_term, search_type, top_n):
'q': search_term, 'q': search_term,
'key': google_api_key, 'key': google_api_key,
'cx': google_cse_id, 'cx': google_cse_id,
'searchType': search_type, **({'searchType': search_type} if search_type == 'image' else {}),
'num': top_n 'num': top_n
} }
response = requests.get(url, params=params) response = requests.get(url, params=params)

2
test/unit/test_search.py
View File

@@ -4,7 +4,7 @@ from dev_gpt.options.generate.static_files.microservice.apis import search_web,
def test_web_search(): def test_web_search():
results = search_web("jina", 10) results = search_web("jina", 10)
assert len(results) == 10 assert len(results) == 10
assert "jina" in results[0] assert "jina" in results[0].lower()
assert not results[0].startswith("http") assert not results[0].startswith("http")
def test_image_search(): def test_image_search():