mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-06-12 03:01:45 +08:00
feat(agent): add ToolTag-based tags to all agent tools; implement tags.py for unified tool capability tagging
This commit is contained in:
@@ -35,6 +35,9 @@ class TestAgentSummarizationStreaming(unittest.TestCase):
|
||||
patch.object(
|
||||
agent_module.prompt_manager, "get_agent_prompt", return_value="prompt"
|
||||
),
|
||||
patch.object(
|
||||
agent_module, "create_subagent_middlewares", return_value=([], [])
|
||||
),
|
||||
patch.object(agent_module, "create_agent", side_effect=_fake_create_agent),
|
||||
patch.object(agent_module.settings, "LLM_MAX_TOOLS", 0),
|
||||
):
|
||||
@@ -93,6 +96,9 @@ class TestAgentSummarizationStreaming(unittest.TestCase):
|
||||
patch.object(
|
||||
agent_module.prompt_manager, "get_agent_prompt", return_value="prompt"
|
||||
),
|
||||
patch.object(
|
||||
agent_module, "create_subagent_middlewares", return_value=([], [])
|
||||
),
|
||||
patch.object(
|
||||
agent_module,
|
||||
"ToolSelectorMiddleware",
|
||||
@@ -138,6 +144,9 @@ class TestAgentSummarizationStreaming(unittest.TestCase):
|
||||
patch.object(
|
||||
agent_module.prompt_manager, "get_agent_prompt", return_value="prompt"
|
||||
),
|
||||
patch.object(
|
||||
agent_module, "create_subagent_middlewares", return_value=([], [])
|
||||
),
|
||||
patch.object(agent_module, "create_agent", side_effect=_fake_create_agent),
|
||||
patch.object(agent_module.settings, "LLM_MAX_TOOLS", 0),
|
||||
):
|
||||
@@ -167,6 +176,9 @@ class TestAgentSummarizationStreaming(unittest.TestCase):
|
||||
patch.object(
|
||||
agent_module.prompt_manager, "get_agent_prompt", return_value="prompt"
|
||||
),
|
||||
patch.object(
|
||||
agent_module, "create_subagent_middlewares", return_value=([], [])
|
||||
),
|
||||
patch.object(agent_module, "create_agent", side_effect=_fake_create_agent),
|
||||
patch.object(agent_module.settings, "LLM_MAX_TOOLS", 0),
|
||||
):
|
||||
|
||||
Reference in New Issue
Block a user