From 63184a49c504c3f23d78eb1dc5b7c1be6eae6553 Mon Sep 17 00:00:00 2001
From: jxxghp <jxxghp@gmail.com>
Date: Sat, 22 Aug 2026 13:35:57 +0800
Subject: [PATCH] fix(ci): use canonical imports in changed files

---
 app/agent/middleware/activity_log.py | 8 ++++++--
 app/chain/_messaging.py              | 2 +-
 tests/test_agent_tokens_events.py    | 2 +-
 tests/test_sunnypt_indexer.py        | 2 +-
 4 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/app/agent/middleware/activity_log.py b/app/agent/middleware/activity_log.py
index d0a25a260..d3422fc46 100644
--- a/app/agent/middleware/activity_log.py
+++ b/app/agent/middleware/activity_log.py
@@ -33,7 +33,11 @@ from langgraph.runtime import Runtime
 from pydantic import BaseModel, Field
 
 from app.agent.middleware.utils import append_to_system_message
-from app.agent.policy import sanitize_for_host, summarize_error, summarize_result
+from app.agent.policy.sanitizer import (
+    sanitize_for_host,
+    summarize_error,
+    summarize_result,
+)
 from app.agent.tools.tags import ToolTag
 from app.runtime.log import logger
 
@@ -459,7 +463,7 @@ async def _summarize_with_llm(conversation_text: str) -> Optional[str]:
         LLM 生成的摘要字符串，失败时返回 None。
     """
     try:
-        from app.agent.llm import LLMHelper
+        from app.agent.llm.helper import LLMHelper
 
         llm = await LLMHelper.get_llm(streaming=False)
         prompt = SUMMARY_PROMPT.format(conversation=conversation_text)
diff --git a/app/chain/_messaging.py b/app/chain/_messaging.py
index 320a8f9f1..6fcaeea09 100644
--- a/app/chain/_messaging.py
+++ b/app/chain/_messaging.py
@@ -18,7 +18,7 @@ from app.runtime.log import logger
 from app.schemas.message import MessageResponse
 from app.schemas.message import Message
 from app.schemas.transfer import TransferInfo
-from app.schemas.message import ChannelCapability, ChannelCapabilityManager
+from app.schemas.notification import ChannelCapability, ChannelCapabilityManager
 from app.schemas.types import EventType, NotificationChannel
 
 
diff --git a/tests/test_agent_tokens_events.py b/tests/test_agent_tokens_events.py
index 7421feafb..3b578ff49 100644
--- a/tests/test_agent_tokens_events.py
+++ b/tests/test_agent_tokens_events.py
@@ -4,7 +4,7 @@ from unittest.mock import AsyncMock, patch
 
 from langchain_core.messages import AIMessage
 
-from app.agent import MoviePilotAgent
+from app.agent.orchestrator import MoviePilotAgent
 from app.agent.memory import memory_manager
 from app.runtime.config import settings
 from app.schemas.types import ChainEventType, EventType
diff --git a/tests/test_sunnypt_indexer.py b/tests/test_sunnypt_indexer.py
index ae1b04bec..69c7b626c 100644
--- a/tests/test_sunnypt_indexer.py
+++ b/tests/test_sunnypt_indexer.py
@@ -13,7 +13,7 @@ from app.db.oper.message import MessageOper
 from app.modules.indexer import IndexerModule
 from app.modules.indexer.parser.sunnypt import SunnyPTSiteUserInfo
 from app.modules.indexer.spider.sunnypt import SunnyPTSpider
-from app.schemas import MediaSource, MediaType, MessageType
+from app.schemas.types import MediaSource, MediaType, MessageType
 
 
 class _FakeResponse: