diff --git a/dev_gpt/options/generate/static_files/microservice/apis.py b/dev_gpt/options/generate/static_files/microservice/apis.py index 1c529da..527d13a 100644 --- a/dev_gpt/options/generate/static_files/microservice/apis.py +++ b/dev_gpt/options/generate/static_files/microservice/apis.py @@ -38,7 +38,7 @@ def google_search(search_term, search_type, top_n): 'q': search_term, 'key': google_api_key, 'cx': google_cse_id, - 'searchType': search_type, + **({'searchType': search_type} if search_type == 'image' else {}), 'num': top_n } response = requests.get(url, params=params) diff --git a/test/unit/test_search.py b/test/unit/test_search.py index 7f16179..b5839fd 100644 --- a/test/unit/test_search.py +++ b/test/unit/test_search.py @@ -4,7 +4,7 @@ from dev_gpt.options.generate.static_files.microservice.apis import search_web, def test_web_search(): results = search_web("jina", 10) assert len(results) == 10 - assert "jina" in results[0] + assert "jina" in results[0].lower() assert not results[0].startswith("http") def test_image_search():