56 lines
1.6 KiB
Python
56 lines
1.6 KiB
Python
"""
|
||
公共工具模块 - 联网搜索、可视化图表等公共功能
|
||
Common Tools Module - Web search, visualization, etc.
|
||
"""
|
||
|
||
from langchain_core.tools import tool
|
||
from typing import Optional
|
||
|
||
|
||
@tool
|
||
def web_search_tool(query: str, max_results: int = 5) -> str:
|
||
"""
|
||
联网搜索工具 - 无需 API Key，使用 DuckDuckGo 免费搜索
|
||
|
||
Args:
|
||
query: 搜索关键词或问题
|
||
max_results: 返回结果数量，默认 5 条
|
||
|
||
Returns:
|
||
格式化的搜索结果，包含引用溯源
|
||
"""
|
||
try:
|
||
from app.core import web_search
|
||
return web_search(query, max_results)
|
||
except Exception as e:
|
||
return f"联网搜索出错：{str(e)}"
|
||
|
||
|
||
@tool
|
||
def generate_chart_tool(data_text: str, chart_type: str = "bar") -> str:
|
||
"""
|
||
可视化图表工具 - 生成 Mermaid 格式图表
|
||
|
||
Args:
|
||
data_text: 图表数据文本，格式：标题,标签1:值1,标签2:值2,...
|
||
示例：月度销售额,1月:100,2月:150,3月:200
|
||
chart_type: 图表类型，可选：bar（柱状图）、line（折线图）、pie（饼图）
|
||
|
||
Returns:
|
||
格式化的图表输出（Mermaid 格式）
|
||
"""
|
||
try:
|
||
from app.core import generate_chart
|
||
return generate_chart(data_text, chart_type)
|
||
except Exception as e:
|
||
return f"生成图表出错：{str(e)}\n\n请使用格式：标题,标签1:值1,标签2:值2,..."
|
||
|
||
|
||
# 公共工具列表
|
||
COMMON_TOOLS = [
|
||
web_search_tool,
|
||
generate_chart_tool
|
||
]
|
||
|
||
COMMON_TOOLS_BY_NAME = {tool.name: tool for tool in COMMON_TOOLS}
|