66 lines
2.5 KiB
Python
66 lines
2.5 KiB
Python
"""网络搜索工具"""
|
||
|
||
# ════════════════════════════════════════════════════════════════
|
||
# tools/web_search.py — 网络搜索工具（模拟）
|
||
# ════════════════════════════════════════════════════════════════
|
||
"""
|
||
tools/web_search.py
|
||
网络搜索工具（Demo 中使用模拟数据，生产环境可替换为真实 API）
|
||
"""
|
||
|
||
import time
|
||
from tools.base_tool import BaseTool, ToolResult
|
||
|
||
|
||
# 模拟搜索结果数据库
|
||
_MOCK_RESULTS: dict[str, list[dict]] = {
|
||
"天气": [
|
||
{"title": "今日天气预报", "snippet": "晴转多云，气温 15°C ~ 24°C，东南风 3 级"},
|
||
{"title": "未来 7 天天气", "snippet": "本周整体晴好，周末有小雨"},
|
||
],
|
||
"python": [
|
||
{"title": "Python 官方文档", "snippet": "Python 3.12 新特性：改进的错误提示、更快的启动速度"},
|
||
{"title": "Python 教程", "snippet": "从零开始学 Python，包含 300+ 实战案例"},
|
||
],
|
||
}
|
||
_DEFAULT_RESULTS = [
|
||
{"title": "搜索结果 1", "snippet": "找到相关内容，请查看详情"},
|
||
{"title": "搜索结果 2", "snippet": "更多相关信息可通过链接访问"},
|
||
]
|
||
|
||
|
||
class WebSearchTool(BaseTool):
|
||
name = "web_search"
|
||
description = "在互联网上搜索信息，返回相关网页摘要"
|
||
parameters = {
|
||
"query": {
|
||
"type": "string",
|
||
"description": "搜索关键词或问题",
|
||
},
|
||
"max_results": {
|
||
"type": "integer",
|
||
"description": "返回结果数量，默认 3",
|
||
},
|
||
}
|
||
|
||
def execute(self, query: str, max_results: int = 3, **_) -> ToolResult:
|
||
time.sleep(0.1) # 模拟网络延迟
|
||
|
||
# 关键词匹配模拟结果
|
||
results = _DEFAULT_RESULTS
|
||
for keyword, data in _MOCK_RESULTS.items():
|
||
if keyword in query:
|
||
results = data
|
||
break
|
||
|
||
results = results[:max_results]
|
||
formatted = "\n".join(
|
||
f"[{i+1}] {r['title']}\n {r['snippet']}"
|
||
for i, r in enumerate(results)
|
||
)
|
||
return ToolResult(
|
||
success=True,
|
||
output=f"搜索「{query}」，共 {len(results)} 条结果:\n{formatted}",
|
||
metadata={"query": query, "count": len(results)},
|
||
)
|