From cb06be7685809a34f14b45f48d059a772886d38a Mon Sep 17 00:00:00 2001 From: jxxghp Date: Sun, 23 Aug 2026 09:52:06 +0800 Subject: [PATCH] fix(ci): use concrete agent imports in search test --- tests/test_search_ai_recommend.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_search_ai_recommend.py b/tests/test_search_ai_recommend.py index ba976f2e0..ebb120501 100644 --- a/tests/test_search_ai_recommend.py +++ b/tests/test_search_ai_recommend.py @@ -12,7 +12,8 @@ ensure_optional_stub("transmission_rpc", File=object) ensure_optional_stub("psutil", __spec__=importlib.machinery.ModuleSpec("psutil", loader=None)) from app.agent.tools.factory import MoviePilotToolFactory -from app.agent import ReplyMode +from app.agent.contracts import ReplyMode +from app.agent.orchestrator import agent_manager from app.chain.search import SearchChain from app.runtime.config import settings from app.modules.indexer import IndexerModule @@ -103,7 +104,6 @@ class SearchChainAIRecommendTest(unittest.IsolatedAsyncioTestCase): async def test_invoke_recommend_llm_disables_output_message_persistence(self): chain = self._make_chain() - from app.agent import agent_manager from app.agent.prompt import prompt_manager captured = {}