wehub-resource-sync
6188617037
Environment Corruption Check / test-python-versions (3.12.8) (push) Failing after 2s
Environment Corruption Check / test-python-versions (3.11.11) (push) Failing after 1s
Pre-commit checks / pre-commit-check (push) Failing after 1s
Environment Corruption Check / test-python-versions (3.13.2) (push) Failing after 4s
34 lines
987 B
Python
34 lines
987 B
Python
from typing import List
|
|
|
|
from googlesearch import search
|
|
|
|
from app.tool.search.base import SearchItem, WebSearchEngine
|
|
|
|
|
|
class GoogleSearchEngine(WebSearchEngine):
|
|
def perform_search(
|
|
self, query: str, num_results: int = 10, *args, **kwargs
|
|
) -> List[SearchItem]:
|
|
"""
|
|
Google search engine.
|
|
|
|
Returns results formatted according to SearchItem model.
|
|
"""
|
|
raw_results = search(query, num_results=num_results, advanced=True)
|
|
|
|
results = []
|
|
for i, item in enumerate(raw_results):
|
|
if isinstance(item, str):
|
|
# If it's just a URL
|
|
results.append(
|
|
{"title": f"Google Result {i+1}", "url": item, "description": ""}
|
|
)
|
|
else:
|
|
results.append(
|
|
SearchItem(
|
|
title=item.title, url=item.url, description=item.description
|
|
)
|
|
)
|
|
|
|
return results
|