chore: import deployable mai-bot source tree

2026-05-11 00:51:12 +00:00
parent 4813699b3e
commit 7a54015f94
1009 changed files with 312999 additions and 16 deletions
--- a/pytests/test_plugin_runtime_action_bridge.py
+++ b/pytests/test_plugin_runtime_action_bridge.py
@@ -0,0 +1,284 @@
+"""核心组件查询层与插件运行时聚合测试。"""
+
+from types import SimpleNamespace
+from typing import Any
+
+import pytest
+
+import src.plugin_runtime.integration as integration_module
+
+from src.core.types import ActionInfo, ToolInfo
+from src.plugin_runtime.component_query import component_query_service
+from src.plugin_runtime.host.supervisor import PluginSupervisor
+
+
+class _FakeRuntimeManager:
+    """测试用插件运行时管理器。"""
+
+    def __init__(self, supervisor: PluginSupervisor, plugin_id: str, plugin_config: dict[str, Any]) -> None:
+        """初始化测试用运行时管理器。
+
+        Args:
+            supervisor: 持有测试组件的监督器。
+            plugin_id: 目标插件 ID。
+            plugin_config: 需要返回的插件配置。
+        """
+
+        self.supervisors = [supervisor]
+        self._plugin_id = plugin_id
+        self._plugin_config = plugin_config
+
+    def _get_supervisor_for_plugin(self, plugin_id: str) -> PluginSupervisor | None:
+        """按插件 ID 返回对应监督器。
+
+        Args:
+            plugin_id: 目标插件 ID。
+
+        Returns:
+            PluginSupervisor | None: 命中时返回监督器。
+        """
+
+        return self.supervisors[0] if plugin_id == self._plugin_id else None
+
+    def _load_plugin_config_for_supervisor(self, supervisor: Any, plugin_id: str) -> dict[str, Any]:
+        """返回测试配置。
+
+        Args:
+            supervisor: 监督器实例。
+            plugin_id: 目标插件 ID。
+
+        Returns:
+            dict[str, Any]: 测试配置内容。
+        """
+
+        del supervisor
+        if plugin_id != self._plugin_id:
+            return {}
+        return dict(self._plugin_config)
+
+
+def _install_runtime_manager(
+    monkeypatch: pytest.MonkeyPatch,
+    supervisor: PluginSupervisor,
+    plugin_id: str,
+    plugin_config: dict[str, Any] | None = None,
+) -> None:
+    """为测试安装假的运行时管理器。
+
+    Args:
+        monkeypatch: pytest monkeypatch 对象。
+        supervisor: 持有测试组件的监督器。
+        plugin_id: 测试插件 ID。
+        plugin_config: 可选的测试配置内容。
+    """
+
+    fake_manager = _FakeRuntimeManager(supervisor, plugin_id, plugin_config or {"enabled": True})
+    monkeypatch.setattr(integration_module, "get_plugin_runtime_manager", lambda: fake_manager)
+
+
+@pytest.mark.asyncio
+async def test_core_component_registry_reads_runtime_action_and_executor(
+    monkeypatch: pytest.MonkeyPatch,
+) -> None:
+    """核心查询层应直接读取运行时 Action，并返回 RPC 执行闭包。"""
+
+    plugin_id = "runtime_action_bridge_plugin"
+    action_name = "runtime_action_bridge_test"
+    supervisor = PluginSupervisor(plugin_dirs=[])
+    captured: dict[str, Any] = {}
+
+    supervisor.component_registry.register_component(
+        name=action_name,
+        component_type="ACTION",
+        plugin_id=plugin_id,
+        metadata={
+            "description": "发送一个测试回复",
+            "enabled": True,
+            "activation_type": "keyword",
+            "activation_probability": 0.25,
+            "activation_keywords": ["测试", "hello"],
+            "action_parameters": {"target": "目标对象"},
+            "action_require": ["需要发送回复时使用"],
+            "associated_types": ["text"],
+            "parallel_action": True,
+        },
+    )
+    _install_runtime_manager(monkeypatch, supervisor, plugin_id, {"enabled": True, "mode": "test"})
+
+    async def fake_invoke_plugin(
+        method: str,
+        plugin_id: str,
+        component_name: str,
+        args: dict[str, Any] | None = None,
+        timeout_ms: int = 30000,
+    ) -> Any:
+        """模拟动作 RPC 调用。"""
+
+        captured["method"] = method
+        captured["plugin_id"] = plugin_id
+        captured["component_name"] = component_name
+        captured["args"] = args or {}
+        captured["timeout_ms"] = timeout_ms
+        return SimpleNamespace(payload={"success": True, "result": (True, "runtime action executed")})
+
+    monkeypatch.setattr(supervisor, "invoke_plugin", fake_invoke_plugin)
+
+    action_info = component_query_service.get_action_info(action_name)
+    assert isinstance(action_info, ActionInfo)
+    assert action_info.plugin_name == plugin_id
+    assert action_info.description == "发送一个测试回复"
+    assert action_info.activation_keywords == ["测试", "hello"]
+    assert action_info.random_activation_probability == 0.25
+    assert action_info.parallel_action is True
+    assert action_name in component_query_service.get_default_actions()
+    assert component_query_service.get_plugin_config(plugin_id) == {"enabled": True, "mode": "test"}
+
+    executor = component_query_service.get_action_executor(action_name)
+    assert executor is not None
+
+    success, reason = await executor(
+        action_data={"target": "MaiBot"},
+        action_reasoning="当前适合使用这个动作",
+        cycle_timers={"planner": 0.1},
+        thinking_id="tid-1",
+        chat_stream=SimpleNamespace(session_id="stream-1"),
+        log_prefix="[test]",
+        shutting_down=False,
+        plugin_config={"enabled": True},
+    )
+
+    assert success is True
+    assert reason == "runtime action executed"
+    assert captured["method"] == "plugin.invoke_action"
+    assert captured["plugin_id"] == plugin_id
+    assert captured["component_name"] == action_name
+    assert captured["args"]["stream_id"] == "stream-1"
+    assert captured["args"]["chat_id"] == "stream-1"
+    assert captured["args"]["reasoning"] == "当前适合使用这个动作"
+    assert captured["args"]["target"] == "MaiBot"
+    assert captured["args"]["action_data"] == {"target": "MaiBot"}
+
+
+@pytest.mark.asyncio
+async def test_core_component_registry_reads_runtime_command_and_executor(
+    monkeypatch: pytest.MonkeyPatch,
+) -> None:
+    """核心查询层应直接使用运行时命令匹配与执行闭包。"""
+
+    plugin_id = "runtime_command_bridge_plugin"
+    command_name = "runtime_command_bridge_test"
+    supervisor = PluginSupervisor(plugin_dirs=[])
+    captured: dict[str, Any] = {}
+
+    supervisor.component_registry.register_component(
+        name=command_name,
+        component_type="COMMAND",
+        plugin_id=plugin_id,
+        metadata={
+            "description": "测试命令",
+            "enabled": True,
+            "command_pattern": r"^/test(?:\s+.+)?$",
+            "aliases": ["/hello"],
+            "intercept_message_level": 1,
+        },
+    )
+    _install_runtime_manager(monkeypatch, supervisor, plugin_id, {"mode": "command"})
+
+    async def fake_invoke_plugin(
+        method: str,
+        plugin_id: str,
+        component_name: str,
+        args: dict[str, Any] | None = None,
+        timeout_ms: int = 30000,
+    ) -> Any:
+        """模拟命令 RPC 调用。"""
+
+        captured["method"] = method
+        captured["plugin_id"] = plugin_id
+        captured["component_name"] = component_name
+        captured["args"] = args or {}
+        captured["timeout_ms"] = timeout_ms
+        return SimpleNamespace(payload={"success": True, "result": (True, "command ok", True)})
+
+    monkeypatch.setattr(supervisor, "invoke_plugin", fake_invoke_plugin)
+
+    matched = component_query_service.find_command_by_text("/test hello")
+    assert matched is not None
+    command_executor, matched_groups, command_info = matched
+
+    assert matched_groups == {}
+    assert command_info.plugin_name == plugin_id
+    assert command_info.command_pattern == r"^/test(?:\s+.+)?$"
+
+    success, response_text, intercept = await command_executor(
+        message=SimpleNamespace(processed_plain_text="/test hello", session_id="stream-2"),
+        plugin_config={"mode": "command"},
+        matched_groups=matched_groups,
+    )
+
+    assert success is True
+    assert response_text == "command ok"
+    assert intercept is True
+    assert captured["method"] == "plugin.invoke_command"
+    assert captured["plugin_id"] == plugin_id
+    assert captured["component_name"] == command_name
+    assert captured["args"]["text"] == "/test hello"
+    assert captured["args"]["stream_id"] == "stream-2"
+    assert captured["args"]["plugin_config"] == {"mode": "command"}
+
+
+@pytest.mark.asyncio
+async def test_core_component_registry_reads_runtime_tools_and_executor(
+    monkeypatch: pytest.MonkeyPatch,
+) -> None:
+    """核心查询层应直接读取运行时 Tool，并返回 RPC 执行闭包。"""
+
+    plugin_id = "runtime_tool_bridge_plugin"
+    tool_name = "runtime_tool_bridge_test"
+    supervisor = PluginSupervisor(plugin_dirs=[])
+
+    supervisor.component_registry.register_component(
+        name=tool_name,
+        component_type="TOOL",
+        plugin_id=plugin_id,
+        metadata={
+            "description": "测试工具",
+            "enabled": True,
+            "parameters": [
+                {
+                    "name": "query",
+                    "param_type": "string",
+                    "description": "查询词",
+                    "required": True,
+                }
+            ],
+        },
+    )
+    _install_runtime_manager(monkeypatch, supervisor, plugin_id)
+
+    async def fake_invoke_plugin(
+        method: str,
+        plugin_id: str,
+        component_name: str,
+        args: dict[str, Any] | None = None,
+        timeout_ms: int = 30000,
+    ) -> Any:
+        """模拟工具 RPC 调用。"""
+
+        del timeout_ms
+        assert method == "plugin.invoke_tool"
+        assert plugin_id == "runtime_tool_bridge_plugin"
+        assert component_name == "runtime_tool_bridge_test"
+        assert args == {"query": "MaiBot"}
+        return SimpleNamespace(payload={"success": True, "result": {"content": "tool ok"}})
+
+    monkeypatch.setattr(supervisor, "invoke_plugin", fake_invoke_plugin)
+
+    tool_info = component_query_service.get_tool_info(tool_name)
+    assert isinstance(tool_info, ToolInfo)
+    assert tool_info.tool_description == "测试工具"
+    assert tool_name in component_query_service.get_llm_available_tools()
+
+    executor = component_query_service.get_tool_executor(tool_name)
+    assert executor is not None
+    assert await executor({"query": "MaiBot"}) == {"content": "tool ok"}