mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-06-01 07:26:50 +00:00
88 lines
3.1 KiB
Python
88 lines
3.1 KiB
Python
import unittest
|
|
from pathlib import Path
|
|
from types import SimpleNamespace
|
|
from unittest.mock import patch
|
|
|
|
from langchain_core.language_models.fake_chat_models import FakeListChatModel
|
|
|
|
import app.agent.middleware.subagents as subagent_module
|
|
from app.agent.middleware.subagents import (
|
|
MoviePilotSubAgentMiddleware,
|
|
SUBAGENT_TASK_TOOL_NAME,
|
|
create_subagent_middlewares,
|
|
)
|
|
from app.agent.tools.tags import ToolTag
|
|
|
|
|
|
class TestAgentSubagents(unittest.TestCase):
|
|
def test_create_subagent_middlewares_registers_task_tool(self):
|
|
"""子代理中间件应向主 Agent 注册 task 委派工具。"""
|
|
model = FakeListChatModel(responses=["ok"])
|
|
|
|
middlewares, task_tools = create_subagent_middlewares(
|
|
model=model,
|
|
tools=[],
|
|
stream_handler=None,
|
|
)
|
|
|
|
self.assertEqual(len(middlewares), 2)
|
|
self.assertEqual([tool.name for tool in task_tools], [SUBAGENT_TASK_TOOL_NAME])
|
|
self.assertIn("media-researcher", task_tools[0].description)
|
|
self.assertIn("system-diagnostician", task_tools[0].description)
|
|
|
|
def test_subagent_tools_are_selected_by_tags(self):
|
|
"""子代理应根据工具标签筛选工具，而不是依赖工具名名单。"""
|
|
model = FakeListChatModel(responses=["ok"])
|
|
tools = [
|
|
SimpleNamespace(
|
|
name="custom_media_lookup",
|
|
tags=[ToolTag.Read.value, ToolTag.Media.value],
|
|
),
|
|
SimpleNamespace(
|
|
name="custom_media_writer",
|
|
tags=[ToolTag.Read.value, ToolTag.Write.value, ToolTag.Media.value],
|
|
),
|
|
SimpleNamespace(
|
|
name="custom_site_lookup",
|
|
tags=[ToolTag.Read.value, ToolTag.Site.value],
|
|
),
|
|
]
|
|
captured = {}
|
|
|
|
def _fake_create_agent(**kwargs):
|
|
captured.update(kwargs)
|
|
return kwargs
|
|
|
|
middleware = MoviePilotSubAgentMiddleware(
|
|
model=model,
|
|
profiles=subagent_module._builtin_subagent_profiles(),
|
|
tools=tools,
|
|
)
|
|
|
|
with patch.object(subagent_module, "create_agent", side_effect=_fake_create_agent):
|
|
middleware._get_agent("media-researcher")
|
|
|
|
self.assertEqual(
|
|
[tool.name for tool in captured["tools"]],
|
|
["custom_media_lookup"],
|
|
)
|
|
|
|
def test_builtin_tools_declare_tags_in_implementation(self):
|
|
"""所有内置工具实现都应显式声明 tags。"""
|
|
impl_dir = Path(__file__).resolve().parents[1] / "app" / "agent" / "tools" / "impl"
|
|
missing_tools = []
|
|
for path in sorted(impl_dir.glob("*.py")):
|
|
text = path.read_text()
|
|
for block in text.split("\nclass "):
|
|
if "(MoviePilotTool)" not in block:
|
|
continue
|
|
class_name = block.split("(", 1)[0].strip()
|
|
if "tags: list[str]" not in block:
|
|
missing_tools.append(f"{path.name}:{class_name}")
|
|
|
|
self.assertEqual([], missing_tools)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|