feat: unify download task tool names

This commit is contained in:
jxxghp
2026-06-15 14:28:18 +08:00
parent 47f6389424
commit bae820a11d
7 changed files with 62 additions and 26 deletions

View File

@@ -0,0 +1,27 @@
from unittest.mock import patch
from app.agent.tools.factory import MoviePilotToolFactory
def test_factory_registers_plural_download_task_tool_names():
"""
下载任务工具应统一使用 *_download_tasks 命名。
"""
with patch(
"app.agent.tools.factory.PluginManager.get_plugin_agent_tools",
return_value=[],
):
tools = MoviePilotToolFactory.create_tools(
session_id="download-task-names",
user_id="10001",
)
tool_names = {tool.name for tool in tools}
assert {
"add_download_tasks",
"query_download_tasks",
"update_download_tasks",
"delete_download_tasks",
} <= tool_names
assert "add_download" not in tool_names
assert "delete_download" not in tool_names