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
# 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",
]

View File

@ -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 = []

View File

@ -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 工具包。",