46 lines
1.6 KiB
Python
46 lines
1.6 KiB
Python
from typing import Optional, Type, TYPE_CHECKING
|
||
from src.plugin_system.base.base_tool import BaseTool
|
||
from src.plugin_system.base.component_types import ComponentType
|
||
|
||
from src.common.logger import get_logger
|
||
|
||
if TYPE_CHECKING:
|
||
from src.chat.message_receive.chat_stream import ChatStream
|
||
|
||
logger = get_logger("tool_api")
|
||
|
||
|
||
def get_tool_instance(tool_name: str, chat_stream: Optional["ChatStream"] = None) -> Optional[BaseTool]:
|
||
"""获取公开工具实例
|
||
|
||
Args:
|
||
tool_name: 工具名称
|
||
chat_stream: 聊天流对象，用于传递聊天上下文信息
|
||
|
||
Returns:
|
||
Optional[BaseTool]: 工具实例，如果未找到则返回None
|
||
"""
|
||
from src.plugin_system.core import component_registry
|
||
|
||
# 获取插件配置
|
||
tool_info = component_registry.get_component_info(tool_name, ComponentType.TOOL)
|
||
if tool_info:
|
||
plugin_config = component_registry.get_plugin_config(tool_info.plugin_name)
|
||
else:
|
||
plugin_config = None
|
||
|
||
tool_class: Type[BaseTool] = component_registry.get_component_class(tool_name, ComponentType.TOOL) # type: ignore
|
||
return tool_class(plugin_config, chat_stream) if tool_class else None
|
||
|
||
|
||
def get_llm_available_tool_definitions():
|
||
"""获取LLM可用的工具定义列表
|
||
|
||
Returns:
|
||
List[Tuple[str, Dict[str, Any]]]: 工具定义列表，为[("tool_name", 定义)]
|
||
"""
|
||
from src.plugin_system.core import component_registry
|
||
|
||
llm_available_tools = component_registry.get_llm_available_tools()
|
||
return [(name, tool_class.get_tool_definition()) for name, tool_class in llm_available_tools.items()]
|