eb1feeeb0b
将工具缓存的实现从`ToolExecutor`的装饰器模式重构为直接集成。缓存逻辑被移出`cache_manager.py`并整合进`ToolExecutor.execute_tool_call`方法中，简化了代码结构并使其更易于维护。 主要变更： - 从`cache_manager.py`中移除了`wrap_tool_executor`函数。 - 在`tool_use.py`中，`execute_tool_call`现在包含完整的缓存检查和设置逻辑。 - 调整了`llm_models/utils_model.py`中的LLM请求逻辑，为模型生成的空回复或截断响应增加了内部重试机制，增强了稳定性。 - 清理了项目中未使用的导入和过时的文档文件，以保持代码库的整洁。
79 lines
2.8 KiB
Python
79 lines
2.8 KiB
Python
from abc import abstractmethod
|
||
from typing import List, Type, Tuple, Union
|
||
from .plugin_base import PluginBase
|
||
|
||
from src.common.logger import get_logger
|
||
from src.plugin_system.base.component_types import ActionInfo, CommandInfo, PlusCommandInfo, EventHandlerInfo, ToolInfo
|
||
from .base_action import BaseAction
|
||
from .base_command import BaseCommand
|
||
from .base_events_handler import BaseEventHandler
|
||
from .base_tool import BaseTool
|
||
from .plus_command import PlusCommand
|
||
|
||
logger = get_logger("base_plugin")
|
||
|
||
|
||
class BasePlugin(PluginBase):
|
||
"""基于Action和Command的插件基类
|
||
|
||
所有上述类型的插件都应该继承这个基类，一个插件可以包含多种组件：
|
||
- Action组件：处理聊天中的动作
|
||
- Command组件：处理命令请求
|
||
- 未来可扩展：Scheduler、Listener等
|
||
"""
|
||
|
||
def __init__(self, *args, **kwargs):
|
||
super().__init__(*args, **kwargs)
|
||
|
||
@abstractmethod
|
||
def get_plugin_components(
|
||
self,
|
||
) -> List[
|
||
Union[
|
||
Tuple[ActionInfo, Type[BaseAction]],
|
||
Tuple[CommandInfo, Type[BaseCommand]],
|
||
Tuple[PlusCommandInfo, Type[PlusCommand]],
|
||
Tuple[EventHandlerInfo, Type[BaseEventHandler]],
|
||
Tuple[ToolInfo, Type[BaseTool]],
|
||
]
|
||
]:
|
||
"""获取插件包含的组件列表
|
||
|
||
子类必须实现此方法，返回组件信息和组件类的列表
|
||
|
||
Returns:
|
||
List[tuple[ComponentInfo, Type]]: [(组件信息, 组件类), ...]
|
||
"""
|
||
raise NotImplementedError("Subclasses must implement this method")
|
||
|
||
def register_plugin(self) -> bool:
|
||
"""注册插件及其所有组件"""
|
||
from src.plugin_system.core.component_registry import component_registry
|
||
|
||
components = self.get_plugin_components()
|
||
|
||
# 检查依赖
|
||
if not self._check_dependencies():
|
||
logger.error(f"{self.log_prefix} 依赖检查失败，跳过注册")
|
||
return False
|
||
|
||
# 注册所有组件
|
||
registered_components = []
|
||
for component_info, component_class in components:
|
||
component_info.plugin_name = self.plugin_name
|
||
if component_registry.register_component(component_info, component_class):
|
||
registered_components.append(component_info)
|
||
else:
|
||
logger.warning(f"{self.log_prefix} 组件 {component_info.name} 注册失败")
|
||
|
||
# 更新插件信息中的组件列表
|
||
self.plugin_info.components = registered_components
|
||
|
||
# 注册插件
|
||
if component_registry.register_plugin(self.plugin_info):
|
||
logger.debug(f"{self.log_prefix} 插件注册成功，包含 {len(registered_components)} 个组件")
|
||
return True
|
||
else:
|
||
logger.error(f"{self.log_prefix} 插件注册失败")
|
||
return False
|