arsam
fa54cc0b7f
Import the existing local project snapshot. This commit organizes existing files; it does not represent newly developed functionality.
56 lines
1.6 KiB
Python
56 lines
1.6 KiB
Python
import logging
|
|
|
|
import httpx
|
|
|
|
from axon.config import settings
|
|
from axon.models import SearchResult
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
_TBS = {"day": "qdr:d", "week": "qdr:w", "month": "qdr:m", "year": "qdr:y"}
|
|
|
|
|
|
async def search_serper(
|
|
query: str,
|
|
count: int = 10,
|
|
topic: str = "general",
|
|
time_range: str | None = None,
|
|
) -> list[SearchResult]:
|
|
if not settings.serper_api_key:
|
|
return []
|
|
|
|
endpoint = (
|
|
"https://google.serper.dev/news"
|
|
if topic == "news"
|
|
else "https://google.serper.dev/search"
|
|
)
|
|
payload: dict = {"q": query, "num": min(count, 10)}
|
|
if time_range and time_range in _TBS:
|
|
payload["tbs"] = _TBS[time_range]
|
|
|
|
headers = {"X-API-KEY": settings.serper_api_key, "Content-Type": "application/json"}
|
|
|
|
try:
|
|
async with httpx.AsyncClient() as client:
|
|
response = await client.post(endpoint, json=payload, headers=headers, timeout=10.0)
|
|
response.raise_for_status()
|
|
data = response.json()
|
|
|
|
raw = data.get("organic") or data.get("news") or []
|
|
results = []
|
|
for item in raw:
|
|
results.append(
|
|
SearchResult(
|
|
title=item.get("title", ""),
|
|
url=item.get("link", ""),
|
|
content=item.get("snippet", ""),
|
|
score=0.5,
|
|
published_date=item.get("date"),
|
|
favicon=None,
|
|
)
|
|
)
|
|
return results
|
|
except Exception as exc:
|
|
logger.error("Serper search error: %s", exc)
|
|
return []
|