refactor: typo 将 gen_tool_info 函数重命名为 get_tool_info

This commit is contained in:
Wenjie Zhang 2026-03-21 15:35:14 +08:00
parent fc58e8e8ef
commit 73bb5ae6b9
3 changed files with 5 additions and 5 deletions

View File

@ -9,7 +9,7 @@ from yuxi.agents.models import load_chat_model
from yuxi.agents.state import BaseState from yuxi.agents.state import BaseState
# Tools - 核心工具函数 # Tools - 核心工具函数
from yuxi.agents.toolkits.utils import gen_tool_info from yuxi.agents.toolkits.utils import get_tool_info
# MCP - Agent 层统一入口(自动过滤 disabled_tools # MCP - Agent 层统一入口(自动过滤 disabled_tools
from yuxi.services.mcp_service import get_enabled_mcp_tools from yuxi.services.mcp_service import get_enabled_mcp_tools
@ -22,7 +22,7 @@ __all__ = [
# Model utilities # Model utilities
"load_chat_model", "load_chat_model",
# Core tools # Core tools
"gen_tool_info", "get_tool_info",
# Core MCP # Core MCP
"get_enabled_mcp_tools", "get_enabled_mcp_tools",
] ]

View File

@ -4,7 +4,7 @@ from typing import Any
from yuxi.utils import logger 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 = [] tools_info = []

View File

@ -38,7 +38,7 @@ class MyAgentContext(BaseContext):
有时需要自定义工具选项,比如 ReporterAgent 需要包含 MySQL 工具: 有时需要自定义工具选项,比如 ReporterAgent 需要包含 MySQL 工具:
```python ```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.buildin import calculator, query_knowledge_graph
from yuxi.agents.toolkits.mysql import get_mysql_tools 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()], default_factory=lambda: [t.name for t in get_mysql_tools()],
metadata={ metadata={
"name": "工具", "name": "工具",
"options": lambda: gen_tool_info( "options": lambda: get_tool_info(
[calculator, query_knowledge_graph, _create_tavily_search()] + get_mysql_tools() [calculator, query_knowledge_graph, _create_tavily_search()] + get_mysql_tools()
), ),
"description": "包含内置工具和 MySQL 工具包。", "description": "包含内置工具和 MySQL 工具包。",