GithubBackup/MoviePilot
1
0
Fork 0
mirror of https://github.com/jxxghp/MoviePilot.git synced 2026-06-11 10:40:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: rename MoviePilotToolSelectorMiddleware to ToolSelectorMiddleware and enhance tool selection logic

Browse Source
This commit is contained in:
jxxghp
2026-04-30 19:05:49 +08:00
parent afcc071d07
commit 53bf81aede
4 changed files with 211 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_agent_summarization_streaming.py
View File

@@ -95,7 +95,7 @@ class TestAgentSummarizationStreaming(unittest.TestCase):
),
patch.object(
agent_module,
"MoviePilotToolSelectorMiddleware",
"ToolSelectorMiddleware",
_FakeToolSelectorMiddleware,
),
patch.object(agent_module, "create_agent", side_effect=_fake_create_agent),

8
tests/test_agent_tool_selector_middleware.py
View File

@@ -100,7 +100,7 @@ class ToolSelectorMiddlewareTest(unittest.TestCase):
SimpleNamespace(name="translate", description="Translate text"),
]
model = _FakeModel()
middleware = tool_selector_module.MoviePilotToolSelectorMiddleware(
middleware = tool_selector_module.ToolSelectorMiddleware(
max_tools=2,
selection_tools=tools,
)
@@ -144,7 +144,7 @@ class ToolSelectorMiddlewareTest(unittest.TestCase):
SimpleNamespace(name="translate", description="Translate text"),
]
model = _FakeModel(content='{"tools": ["calendar", "search"]}')
middleware = tool_selector_module.MoviePilotToolSelectorMiddleware(
middleware = tool_selector_module.ToolSelectorMiddleware(
max_tools=2,
selection_tools=tools,
)
@@ -192,7 +192,7 @@ class ToolSelectorMiddlewareTest(unittest.TestCase):
model_name="gpt-4o-mini",
base_url="https://api.openai.com/v1",
)
middleware = tool_selector_module.MoviePilotToolSelectorMiddleware(
middleware = tool_selector_module.ToolSelectorMiddleware(
max_tools=2,
selection_tools=tools,
)
@@ -240,7 +240,7 @@ class ToolSelectorMiddlewareTest(unittest.TestCase):
)
def test_normalize_selection_response_accepts_code_fence_json(self):
middleware = tool_selector_module.MoviePilotToolSelectorMiddleware()
middleware = tool_selector_module.ToolSelectorMiddleware()
response = SimpleNamespace(
content=[
{