mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-09-04 23:17:20 +08:00
Route autonomous agent tasks through notification broadcasting
This commit is contained in:
jxxghp
@@ -1,14 +1,24 @@
|
||||
import json
|
||||
import threading
|
||||
from datetime import datetime, timedelta
|
||||
from unittest.mock import AsyncMock
|
||||
from pathlib import Path
|
||||
from types import SimpleNamespace
|
||||
from unittest.mock import AsyncMock, Mock, patch
|
||||
from uuid import uuid4
|
||||
|
||||
import pytest
|
||||
import pytz
|
||||
from apscheduler.schedulers.background import BackgroundScheduler
|
||||
from langchain_core.messages import AIMessage
|
||||
|
||||
from app.agent import AgentChain, AgentManager, ReplyMode
|
||||
from app.agent import (
|
||||
AgentChain,
|
||||
AgentManager,
|
||||
MoviePilotAgent,
|
||||
ReplyMode,
|
||||
_MessageTask,
|
||||
)
|
||||
from app.agent.middleware.tool_selection import ToolSelectorMiddleware
|
||||
from app.agent.tools.factory import MoviePilotToolFactory
|
||||
from app.agent.tools.impl.create_agent_task import (
|
||||
CreateAgentTaskInput,
|
||||
@@ -16,9 +26,15 @@ from app.agent.tools.impl.create_agent_task import (
|
||||
)
|
||||
from app.agent.tools.impl.delete_agent_task import DeleteAgentTaskTool
|
||||
from app.agent.tools.impl.query_agent_tasks import QueryAgentTasksTool
|
||||
from app.agent.tools.impl.query_schedulers import QuerySchedulersTool
|
||||
from app.agent.tools.impl.run_agent_task import RunAgentTaskTool
|
||||
from app.agent.tools.impl.run_scheduler import RunSchedulerTool
|
||||
from app.agent.tools.impl.send_message import SendMessageTool
|
||||
from app.agent.tools.impl.update_agent_task import UpdateAgentTaskTool
|
||||
from app.agent.tools.tags import ToolTag
|
||||
from app.core.config import settings
|
||||
from app.db.agenttask_oper import AgentTaskOper
|
||||
from app.schemas import ScheduleInfo
|
||||
from app.scheduler import Scheduler
|
||||
from app.utils.timer import TimerUtils
|
||||
|
||||
@@ -30,6 +46,7 @@ class _FakeAgentTaskScheduler:
|
||||
"""初始化运行时调度记录。"""
|
||||
self.updated = []
|
||||
self.removed = []
|
||||
self.started = []
|
||||
|
||||
def update_agent_task_job(self, task_id: int) -> str:
|
||||
"""记录任务重载并返回固定的下一次执行时间。"""
|
||||
@@ -44,6 +61,11 @@ class _FakeAgentTaskScheduler:
|
||||
"""返回固定的下一次执行时间。"""
|
||||
return "2099-01-01T00:00:00+08:00"
|
||||
|
||||
def start_agent_task(self, task_id: int) -> bool:
|
||||
"""记录 Agent 任务立即执行投递。"""
|
||||
self.started.append(task_id)
|
||||
return True
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def anyio_backend() -> str:
|
||||
@@ -114,10 +136,81 @@ def test_agent_task_tools_are_registered_with_relative_delay_schema() -> None:
|
||||
"create_agent_task",
|
||||
"query_agent_tasks",
|
||||
"update_agent_task",
|
||||
"run_agent_task",
|
||||
"delete_agent_task",
|
||||
}.issubset(tool_names)
|
||||
agent_task_tool_names = {
|
||||
"create_agent_task",
|
||||
"query_agent_tasks",
|
||||
"update_agent_task",
|
||||
"run_agent_task",
|
||||
"delete_agent_task",
|
||||
}
|
||||
assert [
|
||||
tool_class.model_fields["name"].default
|
||||
for tool_class in MoviePilotToolFactory.BUILTIN_TOOL_CLASSES
|
||||
if tool_class.model_fields["name"].default in agent_task_tool_names
|
||||
] == [
|
||||
"create_agent_task",
|
||||
"query_agent_tasks",
|
||||
"update_agent_task",
|
||||
"run_agent_task",
|
||||
"delete_agent_task",
|
||||
]
|
||||
assert "delay_minutes" in CreateAgentTaskInput.model_json_schema()["properties"]
|
||||
|
||||
agent_task_tools = [
|
||||
_build_tool(tool_class, "admin-user")
|
||||
for tool_class in (
|
||||
CreateAgentTaskTool,
|
||||
QueryAgentTasksTool,
|
||||
UpdateAgentTaskTool,
|
||||
RunAgentTaskTool,
|
||||
DeleteAgentTaskTool,
|
||||
)
|
||||
]
|
||||
assert all(ToolTag.AgentTask.value in tool.tags for tool in agent_task_tools)
|
||||
assert all(ToolTag.Scheduler.value not in tool.tags for tool in agent_task_tools)
|
||||
scheduler_tool = _build_tool(QuerySchedulersTool, "admin-user")
|
||||
assert ToolTag.Scheduler.value in scheduler_tool.tags
|
||||
assert ToolTag.AgentTask.value not in scheduler_tool.tags
|
||||
runtime_scheduler_tools = [
|
||||
scheduler_tool,
|
||||
_build_tool(RunSchedulerTool, "admin-user"),
|
||||
]
|
||||
all_scheduler_tools = [*agent_task_tools, *runtime_scheduler_tools]
|
||||
tool_groups = dict(
|
||||
ToolSelectorMiddleware._build_tool_groups(
|
||||
available_tools=all_scheduler_tools,
|
||||
valid_tool_names=[tool.name for tool in all_scheduler_tools],
|
||||
)
|
||||
)
|
||||
assert tool_groups[ToolTag.AgentTask.value] == [
|
||||
"create_agent_task",
|
||||
"query_agent_tasks",
|
||||
"update_agent_task",
|
||||
"run_agent_task",
|
||||
"delete_agent_task",
|
||||
]
|
||||
assert tool_groups[ToolTag.Scheduler.value] == [
|
||||
"query_schedulers",
|
||||
"run_scheduler",
|
||||
]
|
||||
|
||||
|
||||
def test_agent_prompt_declares_scheduler_tool_boundaries() -> None:
|
||||
"""核心提示应明确自主任务与运行时调度服务的工具和 ID 边界。"""
|
||||
project_root = Path(__file__).resolve().parents[1]
|
||||
core_prompt = (project_root / "app/agent/prompt/System Core Prompt.txt").read_text(
|
||||
encoding="utf-8"
|
||||
)
|
||||
|
||||
assert "Manage existing autonomous tasks with `query_agent_tasks`" in core_prompt
|
||||
assert "`run_agent_task`" in core_prompt
|
||||
assert "Use `query_schedulers` and `run_scheduler` only" in core_prompt
|
||||
assert "integer `task_id`" in core_prompt
|
||||
assert "string `job_id`" in core_prompt
|
||||
|
||||
|
||||
def test_agent_task_oper_persists_and_scopes_tasks() -> None:
|
||||
"""AgentTaskOper 应持久化任务并按创建用户隔离查询和修改。"""
|
||||
@@ -155,7 +248,7 @@ def test_agent_task_oper_persists_and_scopes_tasks() -> None:
|
||||
|
||||
|
||||
def test_scheduler_registers_and_removes_agent_task_job() -> None:
|
||||
"""Scheduler 应把数据库任务注册为精确 APScheduler Job 并可动态移除。"""
|
||||
"""Scheduler 应注册并完整列出 Agent 任务，供前端展示和动态移除。"""
|
||||
user_id = f"user-{uuid4().hex}"
|
||||
task = AgentTaskOper().add(
|
||||
name="定时检查",
|
||||
@@ -181,10 +274,111 @@ def test_scheduler_registers_and_removes_agent_task_job() -> None:
|
||||
assert next_run_at
|
||||
assert scheduler._scheduler.get_job(job_id) is not None
|
||||
assert scheduler._jobs[job_id]["kwargs"] == {"task_id": task.id}
|
||||
scheduler._scheduler.start(paused=True)
|
||||
try:
|
||||
scheduler_items = scheduler.list()
|
||||
agent_scheduler_item = next(
|
||||
item for item in scheduler_items if item.id == job_id
|
||||
)
|
||||
assert agent_scheduler_item.name == task.name
|
||||
assert agent_scheduler_item.provider == "[Agent]"
|
||||
|
||||
scheduler.remove_agent_task_job(task.id)
|
||||
assert scheduler._scheduler.get_job(job_id) is None
|
||||
assert job_id not in scheduler._jobs
|
||||
scheduler.remove_agent_task_job(task.id)
|
||||
assert scheduler._scheduler.get_job(job_id) is None
|
||||
assert job_id not in scheduler._jobs
|
||||
finally:
|
||||
scheduler._scheduler.shutdown(wait=False)
|
||||
|
||||
|
||||
def test_scheduler_starts_registered_agent_task_without_waiting() -> None:
|
||||
"""Agent 任务立即执行入口应只调用运行时 start，并拒绝缺失或运行中的任务。"""
|
||||
scheduler = object.__new__(Scheduler)
|
||||
scheduler._lock = threading.RLock()
|
||||
scheduler._jobs = {
|
||||
"agent-task-7": {
|
||||
"name": "测试 Agent 任务",
|
||||
"running": False,
|
||||
}
|
||||
}
|
||||
scheduler.start = Mock()
|
||||
|
||||
assert scheduler.start_agent_task(7) is True
|
||||
scheduler.start.assert_called_once_with("agent-task-7")
|
||||
|
||||
scheduler._jobs["agent-task-7"]["running"] = True
|
||||
assert scheduler.start_agent_task(7) is False
|
||||
assert scheduler.start_agent_task(8) is False
|
||||
|
||||
|
||||
@pytest.mark.anyio
|
||||
async def test_dashboard_schedule_keeps_agent_tasks(monkeypatch) -> None:
|
||||
"""前端后台服务接口必须保留 Agent 自主任务。"""
|
||||
from app.api.endpoints.dashboard import schedule
|
||||
|
||||
scheduler_items = [
|
||||
ScheduleInfo(
|
||||
id="agent-task-7",
|
||||
name="检查资源",
|
||||
provider="[Agent]",
|
||||
status="等待",
|
||||
next_run="20 分钟后",
|
||||
)
|
||||
]
|
||||
monkeypatch.setattr(
|
||||
"app.api.endpoints.dashboard.Scheduler",
|
||||
lambda: SimpleNamespace(list=lambda: scheduler_items),
|
||||
)
|
||||
|
||||
result = await schedule(None)
|
||||
|
||||
assert result == scheduler_items
|
||||
assert result[0].id == "agent-task-7"
|
||||
assert result[0].provider == "[Agent]"
|
||||
|
||||
|
||||
@pytest.mark.anyio
|
||||
async def test_scheduler_tools_exclude_agent_tasks(monkeypatch) -> None:
|
||||
"""运行时调度查询应过滤 Agent 任务，并把两类查询边界写入工具描述。"""
|
||||
scheduler = SimpleNamespace(
|
||||
list=lambda: [
|
||||
ScheduleInfo(
|
||||
id="subscribe_search_all",
|
||||
name="订阅搜索",
|
||||
provider="[系统]",
|
||||
status="等待",
|
||||
next_run="10 分钟后",
|
||||
),
|
||||
ScheduleInfo(
|
||||
id="agent-task-7",
|
||||
name="检查资源",
|
||||
provider="[Agent]",
|
||||
status="等待",
|
||||
next_run="20 分钟后",
|
||||
),
|
||||
]
|
||||
)
|
||||
monkeypatch.setattr("app.scheduler.Scheduler", lambda: scheduler)
|
||||
tool = _build_tool(QuerySchedulersTool, "admin-user")
|
||||
|
||||
result = json.loads(await tool.run())
|
||||
|
||||
assert [item["id"] for item in result] == ["subscribe_search_all"]
|
||||
assert tool.require_admin is True
|
||||
assert "excludes user-created autonomous agent tasks" in tool.description
|
||||
agent_query_tool = _build_tool(QueryAgentTasksTool, "admin-user")
|
||||
assert "owned by the current user" in agent_query_tool.description
|
||||
|
||||
|
||||
@pytest.mark.anyio
|
||||
async def test_run_scheduler_rejects_agent_task_job_id() -> None:
|
||||
"""run_scheduler 不应接受 Agent 任务的运行时 job_id。"""
|
||||
tool = _build_tool(RunSchedulerTool, "admin-user")
|
||||
tool._run_scheduler_sync = Mock()
|
||||
|
||||
result = await tool.run(job_id="agent-task-7")
|
||||
|
||||
assert "run_agent_task" in result
|
||||
tool._run_scheduler_sync.assert_not_called()
|
||||
|
||||
|
||||
@pytest.mark.anyio
|
||||
@@ -243,14 +437,60 @@ async def test_agent_task_tools_manage_persistent_schedule(monkeypatch) -> None:
|
||||
assert running_update == f"Agent 定时任务 {task_id} 正在执行，请稍后再修改"
|
||||
AgentTaskOper().finish(task_id, success=True, result="完成")
|
||||
|
||||
run_result = await _build_tool(RunAgentTaskTool, user_id).run(task_id=task_id)
|
||||
assert f"Agent 定时任务 {task_id} 已提交立即执行" in run_result
|
||||
assert fake_scheduler.started == [task_id]
|
||||
|
||||
deleted = await _build_tool(DeleteAgentTaskTool, user_id).run(task_id=task_id)
|
||||
assert deleted == f"Agent 定时任务 {task_id} 已删除"
|
||||
assert fake_scheduler.removed == [task_id]
|
||||
|
||||
|
||||
@pytest.mark.anyio
|
||||
async def test_agent_manager_executes_task_in_original_session() -> None:
|
||||
"""定时触发应复用原 Agent 会话和渠道，并在单次执行后停用任务。"""
|
||||
async def test_run_agent_task_enforces_owner_and_enabled_state(monkeypatch) -> None:
|
||||
"""立即执行 Agent 任务时应校验当前用户归属和启用状态。"""
|
||||
owner_id = f"owner-{uuid4().hex}"
|
||||
task = AgentTaskOper().add(
|
||||
name="检查资源",
|
||||
content="检查资源并报告",
|
||||
trigger_type="cron",
|
||||
cron_expression="0 * * * *",
|
||||
run_at=None,
|
||||
user_id=owner_id,
|
||||
username="admin",
|
||||
session_id=f"session-{owner_id}",
|
||||
channel="Telegram",
|
||||
source="telegram-test",
|
||||
original_chat_id="chat-1",
|
||||
)
|
||||
fake_scheduler = _FakeAgentTaskScheduler()
|
||||
monkeypatch.setattr("app.scheduler.Scheduler", lambda: fake_scheduler)
|
||||
|
||||
other_user_result = await _build_tool(
|
||||
RunAgentTaskTool,
|
||||
"another-user",
|
||||
).run(task_id=task.id)
|
||||
assert "不存在或不属于当前用户" in other_user_result
|
||||
assert fake_scheduler.started == []
|
||||
|
||||
AgentTaskOper().update(
|
||||
task_id=task.id,
|
||||
user_id=owner_id,
|
||||
payload={"enabled": False},
|
||||
)
|
||||
disabled_result = await _build_tool(
|
||||
RunAgentTaskTool,
|
||||
owner_id,
|
||||
).run(task_id=task.id)
|
||||
assert "已暂停" in disabled_result
|
||||
assert fake_scheduler.started == []
|
||||
|
||||
|
||||
@pytest.mark.anyio
|
||||
async def test_agent_manager_executes_task_with_broadcast_delivery(
|
||||
monkeypatch,
|
||||
) -> None:
|
||||
"""定时触发应复用原 Agent 会话、广播结果并在单次执行后停用任务。"""
|
||||
user_id = f"user-{uuid4().hex}"
|
||||
task = AgentTaskOper().add(
|
||||
name="检查电影资源",
|
||||
@@ -267,6 +507,8 @@ async def test_agent_manager_executes_task_in_original_session() -> None:
|
||||
)
|
||||
manager = AgentManager()
|
||||
manager.process_message = AsyncMock(return_value="已找到 2 个资源")
|
||||
post_message = AsyncMock()
|
||||
monkeypatch.setattr(AgentChain, "async_post_message", post_message)
|
||||
|
||||
success, result = await manager.execute_scheduled_task(task.id)
|
||||
|
||||
@@ -275,12 +517,14 @@ async def test_agent_manager_executes_task_in_original_session() -> None:
|
||||
kwargs = manager.process_message.await_args.kwargs
|
||||
assert kwargs["session_id"] == task.session_id
|
||||
assert kwargs["user_id"] == user_id
|
||||
assert kwargs["channel"] == "Telegram"
|
||||
assert kwargs["source"] == "telegram-test"
|
||||
assert kwargs["original_chat_id"] == "chat-123"
|
||||
assert kwargs["channel"] is None
|
||||
assert kwargs["source"] is None
|
||||
assert kwargs["original_chat_id"] is None
|
||||
assert kwargs["reply_mode"] == ReplyMode.DISPATCH
|
||||
assert kwargs["allow_message_tools"] is True
|
||||
assert kwargs["wait_for_completion"] is True
|
||||
assert "搜索示例电影是否已有资源" in kwargs["message"]
|
||||
post_message.assert_not_awaited()
|
||||
|
||||
completed = AgentTaskOper().get(task.id)
|
||||
assert completed.enabled is False
|
||||
@@ -312,10 +556,10 @@ async def test_agent_manager_executes_task_in_original_session() -> None:
|
||||
|
||||
|
||||
@pytest.mark.anyio
|
||||
async def test_agent_manager_dispatches_contextless_result_to_admin(
|
||||
monkeypatch,
|
||||
async def test_agent_manager_runs_contextless_task_in_broadcast_mode(
|
||||
monkeypatch,
|
||||
) -> None:
|
||||
"""无原消息渠道的 MCP 任务应捕获结果并发送到管理员通知渠道。"""
|
||||
"""无原消息渠道的任务也应由 Agent 广播结果且不由调度器重复补发。"""
|
||||
user_id = f"api-{uuid4().hex}"
|
||||
task = AgentTaskOper().add(
|
||||
name="后台检查资源",
|
||||
@@ -340,9 +584,163 @@ async def test_agent_manager_dispatches_contextless_result_to_admin(
|
||||
assert success is True
|
||||
assert result == "后台检查完成"
|
||||
kwargs = manager.process_message.await_args.kwargs
|
||||
assert kwargs["reply_mode"] == ReplyMode.CAPTURE_ONLY
|
||||
assert kwargs["allow_message_tools"] is False
|
||||
assert kwargs["channel"] is None
|
||||
assert kwargs["source"] is None
|
||||
assert kwargs["reply_mode"] == ReplyMode.DISPATCH
|
||||
assert kwargs["allow_message_tools"] is True
|
||||
post_message.assert_not_awaited()
|
||||
|
||||
|
||||
@pytest.mark.anyio
|
||||
async def test_agent_manager_broadcasts_empty_task_result(monkeypatch) -> None:
|
||||
"""Agent 未返回内容时，调度器应广播一次兜底消息。"""
|
||||
user_id = f"empty-{uuid4().hex}"
|
||||
task = AgentTaskOper().add(
|
||||
name="空结果检查",
|
||||
content="执行检查",
|
||||
trigger_type="cron",
|
||||
cron_expression="0 * * * *",
|
||||
run_at=None,
|
||||
user_id=user_id,
|
||||
username="admin",
|
||||
session_id=f"session-{user_id}",
|
||||
channel="Telegram",
|
||||
source="telegram-test",
|
||||
original_chat_id="chat-123",
|
||||
)
|
||||
manager = AgentManager()
|
||||
manager.process_message = AsyncMock(return_value="")
|
||||
post_message = AsyncMock()
|
||||
monkeypatch.setattr(AgentChain, "async_post_message", post_message)
|
||||
|
||||
success, result = await manager.execute_scheduled_task(task.id)
|
||||
|
||||
assert success is False
|
||||
assert result == "定时任务已执行，但 Agent 未返回结果"
|
||||
notification = post_message.await_args.args[0]
|
||||
assert notification.channel is None
|
||||
assert notification.source is None
|
||||
assert notification.userid is None
|
||||
assert notification.username == settings.SUPERUSER
|
||||
assert notification.text == "后台检查完成"
|
||||
assert notification.original_chat_id is None
|
||||
assert notification.username == "admin"
|
||||
|
||||
|
||||
@pytest.mark.anyio
|
||||
async def test_cached_agent_clears_channel_for_background_task() -> None:
|
||||
"""复用会话 Agent 时，后台任务必须覆盖上一轮保留的渠道信息。"""
|
||||
manager = AgentManager()
|
||||
agent = MoviePilotAgent(
|
||||
session_id="scheduled-cached-session",
|
||||
user_id="user-1",
|
||||
channel="Telegram",
|
||||
source="telegram-test",
|
||||
username="admin",
|
||||
original_chat_id="chat-123",
|
||||
)
|
||||
agent.process = AsyncMock(return_value="完成")
|
||||
manager.active_agents[agent.session_id] = agent
|
||||
task = _MessageTask(
|
||||
session_id=agent.session_id,
|
||||
user_id="user-1",
|
||||
message="执行后台任务",
|
||||
channel=None,
|
||||
source=None,
|
||||
username="admin",
|
||||
original_chat_id=None,
|
||||
reply_mode=ReplyMode.DISPATCH,
|
||||
allow_message_tools=True,
|
||||
)
|
||||
|
||||
result = await manager._process_message_internal(task)
|
||||
|
||||
assert result == "完成"
|
||||
assert agent.channel is None
|
||||
assert agent.source is None
|
||||
assert agent.original_chat_id is None
|
||||
|
||||
|
||||
@pytest.mark.anyio
|
||||
async def test_background_agent_final_message_is_broadcast() -> None:
|
||||
"""后台 Agent 的最终消息应清空渠道及渠道用户定位后广播。"""
|
||||
agent = MoviePilotAgent(
|
||||
session_id="scheduled-broadcast-session",
|
||||
user_id="telegram-user-id",
|
||||
channel=None,
|
||||
source=None,
|
||||
username="admin",
|
||||
original_message_id="message-1",
|
||||
original_chat_id="chat-1",
|
||||
)
|
||||
|
||||
with patch(
|
||||
"app.agent.AgentChain.async_post_message",
|
||||
new_callable=AsyncMock,
|
||||
) as post_message:
|
||||
await agent.send_agent_message("任务完成", title="MoviePilot助手")
|
||||
|
||||
notification = post_message.await_args.args[0]
|
||||
assert notification.channel is None
|
||||
assert notification.source is None
|
||||
assert notification.userid is None
|
||||
assert notification.original_message_id is None
|
||||
assert notification.original_chat_id is None
|
||||
assert notification.username == "admin"
|
||||
|
||||
|
||||
@pytest.mark.anyio
|
||||
async def test_background_send_message_tool_broadcasts() -> None:
|
||||
"""后台 send_message 工具应广播消息并记录本轮已经回复。"""
|
||||
tool = SendMessageTool(
|
||||
session_id="scheduled-tool-session",
|
||||
user_id="telegram-user-id",
|
||||
)
|
||||
tool.set_message_attr(channel=None, source=None, username="admin")
|
||||
agent_context = {}
|
||||
tool.set_agent_context(agent_context)
|
||||
|
||||
with patch(
|
||||
"app.agent.tools.base.ToolChain.async_post_message",
|
||||
new_callable=AsyncMock,
|
||||
) as post_message:
|
||||
result = await tool.run(message="工具已完成任务")
|
||||
|
||||
assert result == "消息已发送"
|
||||
assert agent_context["user_reply_sent"] is True
|
||||
notification = post_message.await_args.args[0]
|
||||
assert notification.channel is None
|
||||
assert notification.source is None
|
||||
assert notification.userid is None
|
||||
assert notification.original_message_id is None
|
||||
assert notification.original_chat_id is None
|
||||
assert notification.username == "admin"
|
||||
|
||||
|
||||
@pytest.mark.anyio
|
||||
async def test_background_agent_does_not_repeat_tool_message() -> None:
|
||||
"""消息工具已完成回复后，后台 Agent 不应再次发送最终文本。"""
|
||||
agent = MoviePilotAgent(
|
||||
session_id="scheduled-no-repeat-session",
|
||||
user_id="user-1",
|
||||
channel=None,
|
||||
source=None,
|
||||
username="admin",
|
||||
replay_mode=ReplyMode.DISPATCH,
|
||||
)
|
||||
agent._tool_context = {"user_reply_sent": True}
|
||||
agent._streamed_output = ""
|
||||
agent.stream_handler = SimpleNamespace(
|
||||
stop_streaming=AsyncMock(return_value=(False, ""))
|
||||
)
|
||||
agent._should_stream = lambda: False
|
||||
completed_agent = SimpleNamespace(
|
||||
ainvoke=AsyncMock(return_value=None),
|
||||
get_state=lambda _config: SimpleNamespace(
|
||||
values={"messages": [AIMessage(content="消息已发送")]}
|
||||
),
|
||||
)
|
||||
agent._create_agent = AsyncMock(return_value=completed_agent)
|
||||
agent.send_agent_message = AsyncMock()
|
||||
|
||||
await agent._execute_agent([])
|
||||
|
||||
agent.send_agent_message.assert_not_awaited()
|
||||
|
||||
Reference in New Issue
Block a user