fix(ci): restore lint and architecture gates

app/agent/callback/__init__.py

@@ -5,15 +5,11 @@ from typing import Any, Optional, Tuple
 
 from fastapi.concurrency import run_in_threadpool
 
-from app.agent.policy import sanitize_for_host
+from app.agent.policy.sanitizer import sanitize_for_host
 from app.chain import ChainBase
 from app.runtime.log import logger
-from app.schemas.message import Message
-from app.schemas.message import (
-    MessageResponse,
-    ChannelCapabilityManager,
-    ChannelCapability,
-)
+from app.schemas.message import Message, MessageResponse
+from app.schemas.notification import ChannelCapabilityManager, ChannelCapability
 from app.schemas.types import NotificationChannel, MessageType
 
 

app/agent/orchestrator.py

@@ -21,7 +21,7 @@ from langgraph.checkpoint.memory import InMemorySaver
 
 from app.agent.callback import StreamingHandler
 from app.agent.contracts import ReplyMode, build_display_message
-from app.agent.llm import LLMHelper
+from app.agent.llm.helper import LLMHelper
 from app.agent.llm.server_tools import ServerToolRegistry
 from app.agent.memory import memory_manager
 from app.agent.middleware.activity_log import (
@@ -51,7 +51,7 @@ from app.agent.middleware.subagents import (
 from app.agent.middleware.tool_selection import ToolSelectorMiddleware
 from app.agent.middleware.usage import UsageMiddleware
 from app.agent.prompt import prompt_manager
-from app.agent.policy import (
+from app.agent.policy.contracts import (
     AuthSource,
     PrincipalType,
     ToolOrigin,