refactor: typo 将 gen_tool_info 函数重命名为 get_tool_info
This commit is contained in:
parent
fc58e8e8ef
commit
73bb5ae6b9
@ -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",
|
||||
]
|
||||
|
||||
@ -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 = []
|
||||
|
||||
|
||||
@ -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 工具包。",
|
||||
|
||||
Loading…
Reference in New Issue
Block a user