7b3c12ba02
- Introduced a new method `validate_plugin_config` in `PluginRuntimeManager` to validate plugin configurations at runtime. - Implemented the `_normalize_plugin_config` method in `PluginRunner` to normalize and persist plugin configurations. - Enhanced the `PluginRunner` to handle configuration validation requests and return normalized configurations. - Updated the WebUI routes to utilize runtime validation for plugin configurations, ensuring that configurations are validated and normalized before saving. - Added tests for runtime configuration validation and normalization processes, including handling of invalid configurations.
95 lines
4.7 KiB
Python
95 lines
4.7 KiB
Python
from typing import TYPE_CHECKING
|
|
|
|
from src.common.logger import get_logger
|
|
from src.plugin_runtime.host.capability_service import CapabilityImpl
|
|
from src.plugin_runtime.host.supervisor import PluginSupervisor
|
|
|
|
if TYPE_CHECKING:
|
|
from src.plugin_runtime.integration import PluginRuntimeManager
|
|
|
|
logger = get_logger("plugin_runtime.integration")
|
|
|
|
|
|
def register_capability_impls(manager: "PluginRuntimeManager", supervisor: PluginSupervisor) -> None:
|
|
"""向指定 Supervisor 注册主程序提供的能力实现。"""
|
|
cap_service = supervisor.capability_service
|
|
|
|
def _register(name: str, impl: CapabilityImpl) -> None:
|
|
"""注册单个能力实现。
|
|
|
|
Args:
|
|
name: 能力名称。
|
|
impl: 能力实现函数。
|
|
"""
|
|
cap_service.register_capability(name, impl)
|
|
|
|
_register("send.text", manager._cap_send_text)
|
|
_register("send.emoji", manager._cap_send_emoji)
|
|
_register("send.image", manager._cap_send_image)
|
|
_register("send.command", manager._cap_send_command)
|
|
_register("send.custom", manager._cap_send_custom)
|
|
|
|
_register("llm.generate", manager._cap_llm_generate)
|
|
_register("llm.generate_with_tools", manager._cap_llm_generate_with_tools)
|
|
_register("llm.get_available_models", manager._cap_llm_get_available_models)
|
|
|
|
_register("config.get", manager._cap_config_get)
|
|
_register("config.get_plugin", manager._cap_config_get_plugin)
|
|
_register("config.get_all", manager._cap_config_get_all)
|
|
|
|
_register("database.query", manager._cap_database_query)
|
|
_register("database.save", manager._cap_database_save)
|
|
_register("database.get", manager._cap_database_get)
|
|
_register("database.delete", manager._cap_database_delete)
|
|
_register("database.count", manager._cap_database_count)
|
|
|
|
_register("chat.get_all_streams", manager._cap_chat_get_all_streams)
|
|
_register("chat.get_group_streams", manager._cap_chat_get_group_streams)
|
|
_register("chat.get_private_streams", manager._cap_chat_get_private_streams)
|
|
_register("chat.get_stream_by_group_id", manager._cap_chat_get_stream_by_group_id)
|
|
_register("chat.get_stream_by_user_id", manager._cap_chat_get_stream_by_user_id)
|
|
|
|
_register("message.get_by_time", manager._cap_message_get_by_time)
|
|
_register("message.get_by_time_in_chat", manager._cap_message_get_by_time_in_chat)
|
|
_register("message.get_recent", manager._cap_message_get_recent)
|
|
_register("message.count_new", manager._cap_message_count_new)
|
|
_register("message.build_readable", manager._cap_message_build_readable)
|
|
|
|
_register("person.get_id", manager._cap_person_get_id)
|
|
_register("person.get_value", manager._cap_person_get_value)
|
|
_register("person.get_id_by_name", manager._cap_person_get_id_by_name)
|
|
|
|
_register("emoji.get_by_description", manager._cap_emoji_get_by_description)
|
|
_register("emoji.get_random", manager._cap_emoji_get_random)
|
|
_register("emoji.get_count", manager._cap_emoji_get_count)
|
|
_register("emoji.get_emotions", manager._cap_emoji_get_emotions)
|
|
_register("emoji.get_all", manager._cap_emoji_get_all)
|
|
_register("emoji.get_info", manager._cap_emoji_get_info)
|
|
_register("emoji.register", manager._cap_emoji_register)
|
|
_register("emoji.delete", manager._cap_emoji_delete)
|
|
|
|
_register("frequency.get_current_talk_value", manager._cap_frequency_get_current_talk_value)
|
|
_register("frequency.set_adjust", manager._cap_frequency_set_adjust)
|
|
_register("frequency.get_adjust", manager._cap_frequency_get_adjust)
|
|
|
|
_register("tool.get_definitions", manager._cap_tool_get_definitions)
|
|
|
|
_register("api.call", manager._cap_api_call)
|
|
_register("api.get", manager._cap_api_get)
|
|
_register("api.list", manager._cap_api_list)
|
|
_register("api.replace_dynamic", manager._cap_api_replace_dynamic)
|
|
|
|
_register("component.get_all_plugins", manager._cap_component_get_all_plugins)
|
|
_register("component.get_plugin_info", manager._cap_component_get_plugin_info)
|
|
_register("component.get_plugin_config_schema", manager._cap_component_get_plugin_config_schema)
|
|
_register("component.list_loaded_plugins", manager._cap_component_list_loaded_plugins)
|
|
_register("component.list_registered_plugins", manager._cap_component_list_registered_plugins)
|
|
_register("component.enable", manager._cap_component_enable)
|
|
_register("component.disable", manager._cap_component_disable)
|
|
_register("component.load_plugin", manager._cap_component_load_plugin)
|
|
_register("component.unload_plugin", manager._cap_component_unload_plugin)
|
|
_register("component.reload_plugin", manager._cap_component_reload_plugin)
|
|
|
|
_register("knowledge.search", manager._cap_knowledge_search)
|
|
logger.debug("已注册全部主程序能力实现")
|