update
This commit is contained in:
@@ -19,8 +19,6 @@ class BaseTool:
|
|||||||
parameters = None
|
parameters = None
|
||||||
# 是否可供LLM使用,默认为False
|
# 是否可供LLM使用,默认为False
|
||||||
available_for_llm = False
|
available_for_llm = False
|
||||||
# 是否启用该工具
|
|
||||||
enabled = True
|
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def get_tool_definition(cls) -> dict[str, Any]:
|
def get_tool_definition(cls) -> dict[str, Any]:
|
||||||
@@ -45,7 +43,6 @@ class BaseTool:
|
|||||||
|
|
||||||
return ToolInfo(
|
return ToolInfo(
|
||||||
name=cls.name,
|
name=cls.name,
|
||||||
enabled=cls.enabled,
|
|
||||||
tool_description=cls.description,
|
tool_description=cls.description,
|
||||||
available_for_llm=cls.available_for_llm,
|
available_for_llm=cls.available_for_llm,
|
||||||
tool_parameters=cls.parameters,
|
tool_parameters=cls.parameters,
|
||||||
|
|||||||
@@ -193,14 +193,11 @@ class ComponentRegistry:
|
|||||||
def _register_tool_component(self, tool_info: ToolInfo, tool_class: BaseTool):
|
def _register_tool_component(self, tool_info: ToolInfo, tool_class: BaseTool):
|
||||||
"""注册Tool组件到Tool特定注册表"""
|
"""注册Tool组件到Tool特定注册表"""
|
||||||
tool_name = tool_info.name
|
tool_name = tool_info.name
|
||||||
if not tool_info.enabled:
|
|
||||||
logger.info(f"Tool组件 {tool_name} 未启用,跳过注册")
|
|
||||||
return False
|
|
||||||
|
|
||||||
self._tool_registry[tool_name] = tool_class
|
self._tool_registry[tool_name] = tool_class
|
||||||
|
|
||||||
# 如果是llm可用的工具,添加到 llm可用工具列表
|
# 如果是llm可用的且启用的工具,添加到 llm可用工具列表
|
||||||
if tool_info.available_for_llm:
|
if tool_info.available_for_llm and tool_info.enabled:
|
||||||
self._llm_available_tools[tool_name] = tool_class
|
self._llm_available_tools[tool_name] = tool_class
|
||||||
|
|
||||||
return True
|
return True
|
||||||
|
|||||||
Reference in New Issue
Block a user