mirror of
https://github.com/bytedance/deer-flow.git
synced 2026-09-19 02:56:17 +00:00
8c8c5ac246
* feat(search): add native recency filters * fix(search): enforce recency across backends * docs(search): record recency provider contract
64 lines
2.1 KiB
Python
64 lines
2.1 KiB
Python
import json
|
|
import logging
|
|
|
|
from langchain.tools import tool
|
|
|
|
from deerflow.community.search_time_range import SearchTimeRange
|
|
from deerflow.config import get_app_config
|
|
|
|
from .searxng_client import SearxngClient
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def _get_tool_config(tool_name: str) -> dict | None:
|
|
"""Get tool config extras safely, returning None if not configured."""
|
|
config = get_app_config().get_tool_config(tool_name)
|
|
if config is None:
|
|
return None
|
|
extras = config.model_extra
|
|
return extras if extras is not None else {}
|
|
|
|
|
|
def _get_searxng_client() -> SearxngClient:
|
|
cfg = _get_tool_config("web_search")
|
|
base_url = "http://localhost:8088"
|
|
if cfg is not None:
|
|
base_url = cfg.get("base_url", base_url)
|
|
return SearxngClient(base_url=base_url)
|
|
|
|
|
|
@tool("web_search", parse_docstring=True)
|
|
async def web_search_tool(query: str, time_range: SearchTimeRange | None = None) -> str:
|
|
"""Search the web using SearXNG.
|
|
|
|
Args:
|
|
query: The query to search for.
|
|
time_range: Optional relative publication/update window. Use only when the request requires recent results.
|
|
"""
|
|
try:
|
|
cfg = _get_tool_config("web_search")
|
|
max_results = 5
|
|
if cfg is not None:
|
|
raw = cfg.get("max_results", max_results)
|
|
max_results = int(raw) if not isinstance(raw, int) else raw
|
|
|
|
client = _get_searxng_client()
|
|
search_kwargs: dict[str, object] = {"max_results": max_results}
|
|
if time_range is not None:
|
|
search_kwargs["time_range"] = time_range
|
|
results = await client.search(query, **search_kwargs)
|
|
|
|
normalized = [
|
|
{
|
|
"title": r.get("title", ""),
|
|
"url": r.get("url", ""),
|
|
"snippet": r.get("content", ""),
|
|
}
|
|
for r in results
|
|
]
|
|
return json.dumps(normalized, indent=2, ensure_ascii=False)
|
|
except Exception as e:
|
|
logger.error(f"Error in web_search_tool: {e}")
|
|
return json.dumps({"error": str(e), "query": query}, ensure_ascii=False)
|