diff --git a/backend/package/yuxi/agents/__init__.py b/backend/package/yuxi/agents/__init__.py index b2895d53..dd7174a4 100644 --- a/backend/package/yuxi/agents/__init__.py +++ b/backend/package/yuxi/agents/__init__.py @@ -9,7 +9,7 @@ from yuxi.agents.models import load_chat_model from yuxi.agents.state import BaseState # Tools - 核心工具函数 -from yuxi.agents.toolkits.utils import gen_tool_info +from yuxi.agents.toolkits.utils import get_tool_info # MCP - Agent 层统一入口(自动过滤 disabled_tools) from yuxi.services.mcp_service import get_enabled_mcp_tools @@ -22,7 +22,7 @@ __all__ = [ # Model utilities "load_chat_model", # Core tools - "gen_tool_info", + "get_tool_info", # Core MCP "get_enabled_mcp_tools", ] diff --git a/backend/package/yuxi/agents/toolkits/utils.py b/backend/package/yuxi/agents/toolkits/utils.py index 0bc2a1ea..9c3a8d2a 100644 --- a/backend/package/yuxi/agents/toolkits/utils.py +++ b/backend/package/yuxi/agents/toolkits/utils.py @@ -4,7 +4,7 @@ from typing import Any from yuxi.utils import logger -def gen_tool_info(tools) -> list[dict[str, Any]]: +def get_tool_info(tools) -> list[dict[str, Any]]: """获取所有工具的信息(用于前端展示)""" tools_info = [] diff --git a/docs/agents/context-config.md b/docs/agents/context-config.md index 33457c64..20a6b651 100644 --- a/docs/agents/context-config.md +++ b/docs/agents/context-config.md @@ -38,7 +38,7 @@ class MyAgentContext(BaseContext): 有时需要自定义工具选项,比如 ReporterAgent 需要包含 MySQL 工具: ```python -from yuxi.agents import BaseContext, gen_tool_info +from yuxi.agents import BaseContext, get_tool_info from yuxi.agents.toolkits.buildin import calculator, query_knowledge_graph from yuxi.agents.toolkits.mysql import get_mysql_tools @@ -48,7 +48,7 @@ class ReporterContext(BaseContext): default_factory=lambda: [t.name for t in get_mysql_tools()], metadata={ "name": "工具", - "options": lambda: gen_tool_info( + "options": lambda: get_tool_info( [calculator, query_knowledge_graph, _create_tavily_search()] + get_mysql_tools() ), "description": "包含内置工具和 MySQL 工具包。",