mirror of
https://github.com/bytedance/deer-flow.git
synced 2026-09-25 14:06:18 +00:00
8c8c5ac246
* feat(search): add native recency filters * fix(search): enforce recency across backends * docs(search): record recency provider contract
72 lines
2.3 KiB
Python
72 lines
2.3 KiB
Python
import logging
|
|
from typing import Any
|
|
|
|
import httpx
|
|
|
|
from deerflow.community.search_time_range import SearchTimeRange
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class SearxngClient:
|
|
"""Client for SearXNG meta search engine API."""
|
|
|
|
def __init__(self, base_url: str) -> None:
|
|
self.base_url = base_url.rstrip("/")
|
|
|
|
async def search(
|
|
self,
|
|
query: str,
|
|
max_results: int = 5,
|
|
categories: list[str] | None = None,
|
|
time_range: SearchTimeRange | None = None,
|
|
) -> list[dict[str, Any]]:
|
|
"""Search the web using SearXNG.
|
|
|
|
Args:
|
|
query: The search query.
|
|
max_results: Maximum number of results to return.
|
|
categories: Search categories to use.
|
|
time_range: Optional relative publication/update window.
|
|
|
|
Returns:
|
|
List of search result dictionaries.
|
|
"""
|
|
params: dict[str, Any] = {
|
|
"q": query,
|
|
"format": "json",
|
|
"language": "auto",
|
|
"pageno": 1,
|
|
}
|
|
if max_results:
|
|
params["limit"] = max_results
|
|
if categories:
|
|
params["categories"] = ",".join(categories)
|
|
if time_range is not None:
|
|
params["time_range"] = time_range
|
|
|
|
logger.debug(f"Searching SearXNG at {self.base_url} with query: {query}")
|
|
try:
|
|
async with httpx.AsyncClient(timeout=30) as client:
|
|
resp = await client.get(
|
|
f"{self.base_url}/search",
|
|
params=params,
|
|
headers={
|
|
"User-Agent": "Mozilla/5.0 (compatible; DeerFlow/1.0)",
|
|
"Accept": "application/json",
|
|
},
|
|
)
|
|
resp.raise_for_status()
|
|
data = resp.json()
|
|
results = data.get("results", [])
|
|
return results[:max_results] if max_results else results
|
|
except httpx.HTTPStatusError as e:
|
|
logger.error(f"SearXNG search returned error status: {e}")
|
|
raise
|
|
except httpx.RequestError as e:
|
|
logger.error(f"SearXNG search request failed: {e}")
|
|
raise
|
|
except Exception as e:
|
|
logger.error(f"An unexpected error occurred during SearXNG search: {e}")
|
|
raise
|