GithubBackup/MoviePilot
1
0
Fork 0
mirror of https://github.com/jxxghp/MoviePilot.git synced 2026-09-05 15:38:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor(agent): 按需加载 Agent 运行时 (#6336)

Browse Source
This commit is contained in:
InfinityPacer
2026-08-16 19:25:24 +08:00
committed by GitHub
parent e5f0c53069
commit 5b367011c5
56 changed files with 5059 additions and 628 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_transfer_failed_retry_buttons.py
+15 -29
View File
@@ -1,18 +1,3 @@
# 把真实 Agent 服务注册进 application 门面（幂等），供测试 patch 门面背后的单例方法。
from app.agent.llm import AgentCapabilityManager, LLMHelper
from app.agent.orchestrator import agent_manager
from app.agent.prompt import prompt_manager
from app.agent.prompt.transfer_redo import build_manual_redo_prompt
from app.application.agent import register_agent_services
register_agent_services(
agent_manager=agent_manager,
prompt_manager=prompt_manager,
capability_manager=AgentCapabilityManager,
llm_helper=LLMHelper,
manual_redo_prompt_builder=build_manual_redo_prompt,
)
import unittest
import asyncio
import sys
@@ -147,16 +132,15 @@ class TestTransferFailedRetryButtons(unittest.TestCase):
with patch.object(settings, "AI_AGENT_ENABLE", True):
with patch(
"app.chain.transfer.TransferHistoryOper"
) as history_oper_cls, patch(
# mixin 中按自身模块命名空间解析 TransferHistoryOper，需同步镜像
"app.chain._transfer.TransferHistoryOper"
) as mixins_history_oper_cls, patch(
"app.chain.transfer.asyncio.run_coroutine_threadsafe",
) as history_oper_cls, patch(
"app.chain._transfer.build_manual_redo_prompt",
return_value="retry transfer prompt",
), patch(
"app.chain._transfer.asyncio.run_coroutine_threadsafe",
side_effect=_close_pending_coro,
) as run_task:
history_oper_cls.return_value.get.return_value = history
mixins_history_oper_cls.return_value.get.return_value = history
with patch.object(chain, "post_message") as post_message:
chain.handle_failed_transfer_callback(
callback_data="transfer_ai_retry_34",
@@ -219,21 +203,23 @@ class TestTransferFailedRetryButtons(unittest.TestCase):
async def fake_async_post_message(*args, **kwargs):
return None
from app.agent.prompt.transfer_redo import build_manual_redo_prompt
manager = SimpleNamespace(run_background_prompt=fake_run_background_prompt)
with patch.object(settings, "AI_AGENT_ENABLE", True):
with patch(
"app.chain.transfer.TransferHistoryOper"
) as history_oper_cls, patch(
# mixin 中按自身模块命名空间解析 TransferHistoryOper，需同步镜像
"app.chain._transfer.TransferHistoryOper"
) as mixins_history_oper_cls, patch(
"app.application.agent._agent_manager.run_background_prompt",
side_effect=fake_run_background_prompt,
) as history_oper_cls, patch(
"app.chain._transfer.build_manual_redo_prompt",
side_effect=build_manual_redo_prompt,
), patch(
"app.chain.transfer.asyncio.run_coroutine_threadsafe",
"app.chain._transfer.get_running_agent_manager",
return_value=manager,
), patch(
"app.chain._transfer.asyncio.run_coroutine_threadsafe",
side_effect=_run_pending_coro,
):
history_oper_cls.return_value.get.return_value = history
mixins_history_oper_cls.return_value.get.return_value = history
with patch.object(chain, "post_message"), patch.object(
chain, "async_post_message", side_effect=fake_async_post_message
):