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

将发送消息工具设为必选工具

Browse Source
This commit is contained in:
jxxghp
2026-05-27 08:48:56 +08:00
parent 990a28b51b
commit 5b6b4c9744
3 changed files with 8 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
tests/test_agent_background_output.py
View File

@@ -12,6 +12,7 @@ from app.agent import (
UNSUPPORTED_IMAGE_INPUT_MESSAGE,
)
from app.agent.memory import memory_manager
from app.agent.tools.factory import MoviePilotToolFactory
from app.core.config import settings
from app.utils.identity import SYSTEM_INTERNAL_USER_ID
@@ -320,42 +321,14 @@ class AgentBackgroundOutputTest(unittest.IsolatedAsyncioTestCase):
created["middleware"],
)
async def test_heartbeat_session_always_keeps_send_message_tool(self):
agent = MoviePilotAgent(
session_id=f"{HEARTBEAT_SESSION_PREFIX}test__",
user_id="system",
)
def test_send_message_tool_is_always_included_by_tool_selector(self):
send_message_tool = SimpleNamespace(name="send_message")
agent._initialize_tools = lambda: [send_message_tool]
captured = {}
always_include = MoviePilotToolFactory.get_tool_selector_always_include_names(
[send_message_tool]
)
def fake_tool_selector(*args, **kwargs):
captured["always_include"] = kwargs["always_include"]
return "selector"
with (
patch.object(settings, "LLM_MAX_TOOLS", 1),
patch.object(agent, "_initialize_llm", new=AsyncMock(return_value=object())),
patch("app.agent.prompt_manager.get_agent_prompt", return_value="PROMPT"),
patch(
"app.agent.MoviePilotToolFactory.get_tool_selector_always_include_names",
return_value=[],
),
patch("app.agent.SkillsMiddleware", side_effect=lambda *args, **kwargs: "skills"),
patch("app.agent.JobsMiddleware", side_effect=lambda *args, **kwargs: "jobs"),
patch("app.agent.RuntimeConfigMiddleware", side_effect=lambda *args, **kwargs: "runtime"),
patch("app.agent.MemoryMiddleware", side_effect=lambda *args, **kwargs: "memory"),
patch("app.agent.SummarizationMiddleware", side_effect=lambda *args, **kwargs: "summary"),
patch("app.agent.PatchToolCallsMiddleware", side_effect=lambda *args, **kwargs: "patch"),
patch("app.agent.UsageMiddleware", side_effect=lambda *args, **kwargs: "usage"),
patch("app.agent.ToolSelectorMiddleware", side_effect=fake_tool_selector),
patch("app.agent.InMemorySaver", return_value="checkpointer"),
patch("app.agent.create_agent", side_effect=lambda **kwargs: kwargs),
):
await agent._create_agent(streaming=False)
self.assertIn("send_message", captured["always_include"])
self.assertIn("send_message", always_include)
async def test_create_agent_keeps_activity_log_for_normal_session(self):
agent = MoviePilotAgent(session_id="normal-session", user_id="system")