mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-10-29 11:27:43 +08:00
65 lines
3.1 KiB
Python
65 lines
3.1 KiB
Python
"""
|
|
验证 Agent 进度提示与工具展示模式独立，并适配纯文本渠道。
|
|
"""
|
|
|
|
from unittest.mock import patch
|
|
|
|
from app.agent.middleware.subagents import SUBAGENT_PARENT_PROMPT
|
|
from app.agent.prompt import prompt_manager
|
|
from app.runtime.config import settings
|
|
from app.schemas.notification import ChannelCapabilityManager
|
|
from app.schemas.types import NotificationChannel
|
|
|
|
|
|
def test_progress_prompt_is_independent_from_tool_display_mode() -> None:
|
|
"""
|
|
进度沟通规则不应随工具逐条或汇总展示模式变化。
|
|
"""
|
|
with patch.object(settings, "AI_AGENT_VERBOSE", False):
|
|
summary_mode_prompt = prompt_manager.get_agent_prompt(
|
|
channel=NotificationChannel.WebAgent.value
|
|
)
|
|
with patch.object(settings, "AI_AGENT_VERBOSE", True):
|
|
verbose_mode_prompt = prompt_manager.get_agent_prompt(
|
|
channel=NotificationChannel.WebAgent.value
|
|
)
|
|
|
|
assert summary_mode_prompt == verbose_mode_prompt
|
|
assert "progress updates and final answer" in SUBAGENT_PARENT_PROMPT
|
|
assert "only your synthesized final answer" not in SUBAGENT_PARENT_PROMPT
|
|
assert "meaningful changes in understanding or execution" in summary_mode_prompt
|
|
assert "not on elapsed time or the number of tool calls" in summary_mode_prompt
|
|
assert "one or two tools have finished" in summary_mode_prompt
|
|
assert "useful preliminary conclusion" in summary_mode_prompt
|
|
assert "materially changes the working direction" in summary_mode_prompt
|
|
assert "sustained blocker the user should know about" in summary_mode_prompt
|
|
assert "including the key evidence and what you will do next" in summary_mode_prompt
|
|
assert "brevity is not a goal by itself" in summary_mode_prompt
|
|
assert "do not repeat an unchanged status" in summary_mode_prompt.lower()
|
|
assert "Continue working after each update" in summary_mode_prompt
|
|
assert "The final reply must be self-contained" in summary_mode_prompt
|
|
assert "before the first tool call" not in summary_mode_prompt
|
|
assert "approximately every 30 to 60 seconds" not in summary_mode_prompt
|
|
assert "one or two short sentences" not in summary_mode_prompt
|
|
assert "remain completely silent" not in summary_mode_prompt
|
|
assert "DO NOT output any intermediate content" not in summary_mode_prompt
|
|
|
|
|
|
def test_plain_text_channel_uses_consistent_formatting() -> None:
|
|
"""
|
|
微信纯文本规则不再推荐已禁止的 Markdown 符号，且不影响网页渠道。
|
|
"""
|
|
caps = ChannelCapabilityManager.get_capabilities(NotificationChannel.Wechat)
|
|
instructions = prompt_manager._generate_formatting_instructions(caps)
|
|
plain_prompt = prompt_manager.get_agent_prompt(NotificationChannel.Wechat.value)
|
|
web_prompt = prompt_manager.get_agent_prompt(NotificationChannel.WebAgent.value)
|
|
|
|
assert instructions in plain_prompt
|
|
assert "plain text only" in instructions
|
|
assert "1、" in instructions
|
|
assert "*" not in instructions
|
|
assert "`>`" not in instructions
|
|
assert "Paste URLs directly as text" in instructions
|
|
assert "plain text only" not in web_prompt
|
|
assert "even when the persona suggests a conflicting format" in plain_prompt
|