refactor(agents): move mac-only modules out of the cross-agent shared kernel

app/agents/shared/ is a sibling of anonymous_chat/podcaster/multi_agent_chat/
video_presentation, so it should only hold code shared across 2+ of those
agents. In practice podcaster and video_presentation import nothing from it,
and anonymous_chat needs only context + compaction + retry_after + web_search.
Everything else was multi_agent_chat-only (the boundary just passes through).

Move the multi_agent_chat-only cluster into multi_agent_chat/shared/ (files
moved verbatim via git rename; ~116 import sites rewritten):

  errors, feature_flags, filesystem_selection, path_resolver, prompt_caching,
  sandbox, llm_config, mention_resolver
  middleware/busy_mutex, middleware/kb_persistence

busy_mutex/llm_config/mention_resolver are boundary-only but import the moved
modules, so they were folded in to avoid a backwards shared -> multi_agent_chat
dependency. main_agent builders now import the impls directly; the shared
middleware barrel keeps only the genuinely-shared compaction + retry_after.

Also delete the dead leftover shared/plugins and shared/skills dirs (live
copies already live under main_agent/).

Remaining in app/agents/shared/: context, system_prompt(+prompts), checkpointer,
middleware/{compaction,retry_after,dedup_tool_calls}, tools/. checkpointer and
system_prompt are boundary-only infra pending a dedicated home decision.

surfsense_backend/app/agents/multi_agent_chat/main_agent/graph/compile_graph_sync.py

@@ -14,9 +14,9 @@ from langgraph.types import Checkpointer
 from app.agents.multi_agent_chat.main_agent.middleware.stack import (
     build_main_agent_deepagent_middleware,
 )
+from app.agents.multi_agent_chat.shared.feature_flags import AgentFeatureFlags
+from app.agents.multi_agent_chat.shared.filesystem_selection import FilesystemMode
 from app.agents.shared.context import SurfSenseContextSchema
-from app.agents.shared.feature_flags import AgentFeatureFlags
-from app.agents.shared.filesystem_selection import FilesystemMode
 from app.db import ChatVisibility
 
 

surfsense_backend/app/agents/multi_agent_chat/main_agent/middleware/action_log/builder.py

@@ -4,8 +4,8 @@ from __future__ import annotations
 
 import logging
 
+from app.agents.multi_agent_chat.shared.feature_flags import AgentFeatureFlags
 from app.agents.multi_agent_chat.shared.middleware.flags import enabled
-from app.agents.shared.feature_flags import AgentFeatureFlags
 
 from .middleware import ActionLogMiddleware
 

surfsense_backend/app/agents/multi_agent_chat/main_agent/middleware/action_log/middleware.py

@@ -34,7 +34,7 @@ from langchain.agents.middleware import AgentMiddleware
 from langchain_core.callbacks import adispatch_custom_event
 from langchain_core.messages import ToolMessage
 
-from app.agents.shared.feature_flags import get_flags
+from app.agents.multi_agent_chat.shared.feature_flags import get_flags
 
 if TYPE_CHECKING:  # pragma: no cover - type-only
     from langchain.agents.middleware.types import ToolCallRequest

surfsense_backend/app/agents/multi_agent_chat/main_agent/middleware/anonymous_document/builder.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from app.agents.shared.filesystem_selection import FilesystemMode
+from app.agents.multi_agent_chat.shared.filesystem_selection import FilesystemMode
 
 from .middleware import AnonymousDocumentMiddleware
 

surfsense_backend/app/agents/multi_agent_chat/main_agent/middleware/anonymous_document/middleware.py

@@ -24,10 +24,13 @@ from typing import Any
 from langchain.agents.middleware import AgentMiddleware, AgentState
 from langgraph.runtime import Runtime
 
+from app.agents.multi_agent_chat.shared.path_resolver import (
+    DOCUMENTS_ROOT,
+    safe_filename,
+)
 from app.agents.multi_agent_chat.shared.state.filesystem_state import (
     SurfSenseFilesystemState,
 )
-from app.agents.shared.path_resolver import DOCUMENTS_ROOT, safe_filename
 
 logger = logging.getLogger(__name__)
 

surfsense_backend/app/agents/multi_agent_chat/main_agent/middleware/busy_mutex.py

@@ -2,9 +2,8 @@
 
 from __future__ import annotations
 
-from app.agents.shared.feature_flags import AgentFeatureFlags
-from app.agents.shared.middleware import BusyMutexMiddleware
-
+from app.agents.multi_agent_chat.shared.feature_flags import AgentFeatureFlags
+from app.agents.multi_agent_chat.shared.middleware.busy_mutex import BusyMutexMiddleware
 from app.agents.multi_agent_chat.shared.middleware.flags import enabled
 
 

surfsense_backend/app/agents/multi_agent_chat/main_agent/middleware/context_editing/builder.py

@@ -10,8 +10,8 @@ from langchain_core.tools import BaseTool
 from app.agents.multi_agent_chat.main_agent.context_prune.prune_tool_names import (
     safe_exclude_tools,
 )
+from app.agents.multi_agent_chat.shared.feature_flags import AgentFeatureFlags
 from app.agents.multi_agent_chat.shared.middleware.flags import enabled
-from app.agents.shared.feature_flags import AgentFeatureFlags
 
 from .middleware import (
     ClearToolUsesEdit,

surfsense_backend/app/agents/multi_agent_chat/main_agent/middleware/doom_loop/builder.py

@@ -2,8 +2,8 @@
 
 from __future__ import annotations
 
+from app.agents.multi_agent_chat.shared.feature_flags import AgentFeatureFlags
 from app.agents.multi_agent_chat.shared.middleware.flags import enabled
-from app.agents.shared.feature_flags import AgentFeatureFlags
 
 from .middleware import DoomLoopMiddleware
 

surfsense_backend/app/agents/multi_agent_chat/main_agent/middleware/kb_persistence.py

@@ -2,8 +2,10 @@
 
 from __future__ import annotations
 
-from app.agents.shared.filesystem_selection import FilesystemMode
-from app.agents.shared.middleware import KnowledgeBasePersistenceMiddleware
+from app.agents.multi_agent_chat.shared.filesystem_selection import FilesystemMode
+from app.agents.multi_agent_chat.shared.middleware.kb_persistence import (
+    KnowledgeBasePersistenceMiddleware,
+)
 
 
 def build_kb_persistence_mw(

surfsense_backend/app/agents/multi_agent_chat/main_agent/middleware/knowledge_priority.py

@@ -4,10 +4,10 @@ from __future__ import annotations
 
 from langchain_core.language_models import BaseChatModel
 
+from app.agents.multi_agent_chat.shared.filesystem_selection import FilesystemMode
 from app.agents.multi_agent_chat.shared.middleware.knowledge_search import (
     KnowledgePriorityMiddleware,
 )
-from app.agents.shared.filesystem_selection import FilesystemMode
 from app.services.llm_service import get_planner_llm
 
 

surfsense_backend/app/agents/multi_agent_chat/main_agent/middleware/knowledge_tree/builder.py

@@ -4,7 +4,7 @@ from __future__ import annotations
 
 from langchain_core.language_models import BaseChatModel
 
-from app.agents.shared.filesystem_selection import FilesystemMode
+from app.agents.multi_agent_chat.shared.filesystem_selection import FilesystemMode
 
 from .middleware import KnowledgeTreeMiddleware
 

surfsense_backend/app/agents/multi_agent_chat/main_agent/middleware/knowledge_tree/middleware.py

@@ -33,16 +33,16 @@ from langchain_core.messages import SystemMessage
 from langgraph.runtime import Runtime
 from sqlalchemy import select
 
-from app.agents.multi_agent_chat.shared.state.filesystem_state import (
-    SurfSenseFilesystemState,
-)
-from app.agents.shared.filesystem_selection import FilesystemMode
-from app.agents.shared.path_resolver import (
+from app.agents.multi_agent_chat.shared.filesystem_selection import FilesystemMode
+from app.agents.multi_agent_chat.shared.path_resolver import (
     DOCUMENTS_ROOT,
     PathIndex,
     build_path_index,
     doc_to_virtual_path,
 )
+from app.agents.multi_agent_chat.shared.state.filesystem_state import (
+    SurfSenseFilesystemState,
+)
 from app.db import Document, shielded_async_session
 from app.utils.perf import get_perf_logger
 

surfsense_backend/app/agents/multi_agent_chat/main_agent/middleware/noop_injection/builder.py

@@ -2,8 +2,8 @@
 
 from __future__ import annotations
 
+from app.agents.multi_agent_chat.shared.feature_flags import AgentFeatureFlags
 from app.agents.multi_agent_chat.shared.middleware.flags import enabled
-from app.agents.shared.feature_flags import AgentFeatureFlags
 
 from .middleware import NoopInjectionMiddleware
 

surfsense_backend/app/agents/multi_agent_chat/main_agent/middleware/otel_span/builder.py

@@ -2,8 +2,8 @@
 
 from __future__ import annotations
 
+from app.agents.multi_agent_chat.shared.feature_flags import AgentFeatureFlags
 from app.agents.multi_agent_chat.shared.middleware.flags import enabled
-from app.agents.shared.feature_flags import AgentFeatureFlags
 
 from .middleware import OtelSpanMiddleware
 

surfsense_backend/app/agents/multi_agent_chat/main_agent/middleware/plugins.py

@@ -7,8 +7,8 @@ from typing import Any
 
 from langchain_core.language_models import BaseChatModel
 
+from app.agents.multi_agent_chat.shared.feature_flags import AgentFeatureFlags
 from app.agents.multi_agent_chat.shared.middleware.flags import enabled
-from app.agents.shared.feature_flags import AgentFeatureFlags
 from app.db import ChatVisibility
 
 from ..plugins.loader import (

surfsense_backend/app/agents/multi_agent_chat/main_agent/middleware/skills.py

@@ -6,9 +6,9 @@ import logging
 
 from deepagents.middleware.skills import SkillsMiddleware
 
+from app.agents.multi_agent_chat.shared.feature_flags import AgentFeatureFlags
+from app.agents.multi_agent_chat.shared.filesystem_selection import FilesystemMode
 from app.agents.multi_agent_chat.shared.middleware.flags import enabled
-from app.agents.shared.feature_flags import AgentFeatureFlags
-from app.agents.shared.filesystem_selection import FilesystemMode
 
 from ..skills.backends import build_skills_backend_factory, default_skills_sources
 

surfsense_backend/app/agents/multi_agent_chat/main_agent/middleware/stack.py

@@ -20,6 +20,8 @@ from langchain_core.language_models import BaseChatModel
 from langchain_core.tools import BaseTool
 from langgraph.types import Checkpointer
 
+from app.agents.multi_agent_chat.shared.feature_flags import AgentFeatureFlags
+from app.agents.multi_agent_chat.shared.filesystem_selection import FilesystemMode
 from app.agents.multi_agent_chat.shared.middleware.anthropic_cache import (
     build_anthropic_cache_mw,
 )
@@ -52,8 +54,6 @@ from app.agents.multi_agent_chat.subagents.builtins.knowledge_base.ask_knowledge
 from app.agents.multi_agent_chat.subagents.shared.middleware.middleware_stack import (
     build_subagent_middleware_stack,
 )
-from app.agents.shared.feature_flags import AgentFeatureFlags
-from app.agents.shared.filesystem_selection import FilesystemMode
 from app.db import ChatVisibility
 
 from .action_log import build_action_log_mw

surfsense_backend/app/agents/multi_agent_chat/main_agent/middleware/tool_call_repair/builder.py

@@ -6,8 +6,8 @@ from collections.abc import Sequence
 
 from langchain_core.tools import BaseTool
 
+from app.agents.multi_agent_chat.shared.feature_flags import AgentFeatureFlags
 from app.agents.multi_agent_chat.shared.middleware.flags import enabled
-from app.agents.shared.feature_flags import AgentFeatureFlags
 
 from .middleware import ToolCallNameRepairMiddleware
 

surfsense_backend/app/agents/multi_agent_chat/main_agent/runtime/agent_cache.py

@@ -10,8 +10,8 @@ from langchain_core.language_models import BaseChatModel
 from langchain_core.tools import BaseTool
 from langgraph.types import Checkpointer
 
-from app.agents.shared.feature_flags import AgentFeatureFlags
-from app.agents.shared.filesystem_selection import FilesystemMode
+from app.agents.multi_agent_chat.shared.feature_flags import AgentFeatureFlags
+from app.agents.multi_agent_chat.shared.filesystem_selection import FilesystemMode
 from app.db import ChatVisibility
 
 from ..graph.compile_graph_sync import build_compiled_agent_graph_sync

surfsense_backend/app/agents/multi_agent_chat/main_agent/runtime/factory.py

@@ -12,9 +12,21 @@ from langchain_core.tools import BaseTool
 from langgraph.types import Checkpointer
 from sqlalchemy.ext.asyncio import AsyncSession
 
+from app.agents.multi_agent_chat.shared.feature_flags import (
+    AgentFeatureFlags,
+    get_flags,
+)
+from app.agents.multi_agent_chat.shared.filesystem_selection import (
+    FilesystemMode,
+    FilesystemSelection,
+)
+from app.agents.multi_agent_chat.shared.llm_config import AgentConfig
 from app.agents.multi_agent_chat.shared.middleware.filesystem.backends.resolver import (
     build_backend_resolver,
 )
+from app.agents.multi_agent_chat.shared.prompt_caching import (
+    apply_litellm_prompt_caching,
+)
 from app.agents.multi_agent_chat.subagents import (
     get_subagents_to_exclude,
     main_prompt_registry_subagent_lines,
@@ -22,10 +34,6 @@ from app.agents.multi_agent_chat.subagents import (
 from app.agents.multi_agent_chat.subagents.mcp_tools.index import (
     load_mcp_tools_by_connector,
 )
-from app.agents.shared.feature_flags import AgentFeatureFlags, get_flags
-from app.agents.shared.filesystem_selection import FilesystemMode, FilesystemSelection
-from app.agents.shared.llm_config import AgentConfig
-from app.agents.shared.prompt_caching import apply_litellm_prompt_caching
 from app.db import ChatVisibility
 from app.services.connector_service import ConnectorService
 from app.services.user_tool_allowlist import (

surfsense_backend/app/agents/multi_agent_chat/shared/llm_config.py

@@ -27,7 +27,9 @@ from litellm import get_model_info
 from sqlalchemy import select
 from sqlalchemy.ext.asyncio import AsyncSession
 
-from app.agents.shared.prompt_caching import apply_litellm_prompt_caching
+from app.agents.multi_agent_chat.shared.prompt_caching import (
+    apply_litellm_prompt_caching,
+)
 from app.services.llm_router_service import (
     AUTO_MODE_ID,
     ChatLiteLLMRouter,

surfsense_backend/app/agents/multi_agent_chat/shared/mention_resolver.py

@@ -36,7 +36,7 @@ from dataclasses import dataclass, field
 from sqlalchemy import select
 from sqlalchemy.ext.asyncio import AsyncSession
 
-from app.agents.shared.path_resolver import (
+from app.agents.multi_agent_chat.shared.path_resolver import (
     DOCUMENTS_ROOT,
     build_path_index,
     doc_to_virtual_path,

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/busy_mutex.py

@@ -20,7 +20,7 @@ What this provides:
   tools can poll to abort cooperatively. The event is reset between
   turns. Tools should check ``runtime.context.cancel_event.is_set()``
   in tight inner loops.
-- A typed :class:`~app.agents.shared.errors.BusyError` raised when a
+- A typed :class:`~app.agents.multi_agent_chat.shared.errors.BusyError` raised when a
   second turn arrives while the lock is held.
 
 Note: SurfSense's ``stream_new_chat`` is the call site that should
@@ -46,7 +46,7 @@ from langchain.agents.middleware.types import (
 from langgraph.config import get_config
 from langgraph.runtime import Runtime
 
-from app.agents.shared.errors import BusyError
+from app.agents.multi_agent_chat.shared.errors import BusyError
 
 logger = logging.getLogger(__name__)
 

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/backends/kb_postgres.py

@@ -45,7 +45,7 @@ from sqlalchemy.ext.asyncio import AsyncSession
 from app.agents.multi_agent_chat.shared.middleware.filesystem.backends.document_xml import (
     build_document_xml,
 )
-from app.agents.shared.path_resolver import (
+from app.agents.multi_agent_chat.shared.path_resolver import (
     DOCUMENTS_ROOT,
     build_path_index,
     doc_to_virtual_path,

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/backends/resolver.py

@@ -9,13 +9,16 @@ from deepagents.backends.protocol import BackendProtocol
 from deepagents.backends.state import StateBackend
 from langgraph.prebuilt.tool_node import ToolRuntime
 
+from app.agents.multi_agent_chat.shared.filesystem_selection import (
+    FilesystemMode,
+    FilesystemSelection,
+)
 from app.agents.multi_agent_chat.shared.middleware.filesystem.backends.kb_postgres import (
     KBPostgresBackend,
 )
 from app.agents.multi_agent_chat.shared.middleware.filesystem.backends.multi_root_local_folder import (
     MultiRootLocalFolderBackend,
 )
-from app.agents.shared.filesystem_selection import FilesystemMode, FilesystemSelection
 
 
 @lru_cache(maxsize=64)

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/index.py

@@ -4,7 +4,7 @@ from __future__ import annotations
 
 from typing import Any
 
-from app.agents.shared.filesystem_selection import FilesystemMode
+from app.agents.multi_agent_chat.shared.filesystem_selection import FilesystemMode
 
 from .middleware import SurfSenseFilesystemMiddleware
 

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/middleware/middleware.py

@@ -7,11 +7,11 @@ from typing import Any
 from deepagents import FilesystemMiddleware
 from langchain_core.tools import BaseTool
 
+from app.agents.multi_agent_chat.shared.filesystem_selection import FilesystemMode
+from app.agents.multi_agent_chat.shared.sandbox import is_sandbox_enabled
 from app.agents.multi_agent_chat.shared.state.filesystem_state import (
     SurfSenseFilesystemState,
 )
-from app.agents.shared.filesystem_selection import FilesystemMode
-from app.agents.shared.sandbox import is_sandbox_enabled
 
 from ..system_prompt import build_system_prompt
 from ..tools import (

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/middleware/mode.py

@@ -2,8 +2,8 @@
 
 from __future__ import annotations
 
-from app.agents.shared.filesystem_selection import FilesystemMode
-from app.agents.shared.path_resolver import DOCUMENTS_ROOT
+from app.agents.multi_agent_chat.shared.filesystem_selection import FilesystemMode
+from app.agents.multi_agent_chat.shared.path_resolver import DOCUMENTS_ROOT
 
 
 def is_cloud(mode: FilesystemMode) -> bool:

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/middleware/namespace_policy.py

@@ -11,10 +11,10 @@ from typing import TYPE_CHECKING
 
 from langchain.tools import ToolRuntime
 
+from app.agents.multi_agent_chat.shared.path_resolver import DOCUMENTS_ROOT
 from app.agents.multi_agent_chat.shared.state.filesystem_state import (
     SurfSenseFilesystemState,
 )
-from app.agents.shared.path_resolver import DOCUMENTS_ROOT
 
 from ..shared.paths import TEMP_PREFIX, basename
 from .mode import is_cloud

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/middleware/path_resolution.py

@@ -7,13 +7,13 @@ from typing import TYPE_CHECKING
 
 from langchain.tools import ToolRuntime
 
+from app.agents.multi_agent_chat.shared.filesystem_selection import FilesystemMode
 from app.agents.multi_agent_chat.shared.middleware.filesystem.backends.multi_root_local_folder import (
     MultiRootLocalFolderBackend,
 )
 from app.agents.multi_agent_chat.shared.state.filesystem_state import (
     SurfSenseFilesystemState,
 )
-from app.agents.shared.filesystem_selection import FilesystemMode
 
 from ..shared.paths import (
     extract_mount_from_path,

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/system_prompt/index.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from app.agents.shared.filesystem_selection import FilesystemMode
+from app.agents.multi_agent_chat.shared.filesystem_selection import FilesystemMode
 
 from .cloud import BODY as CLOUD_BODY
 from .common import HEADER, SANDBOX_ADDENDUM

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/cd/description.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from app.agents.shared.filesystem_selection import FilesystemMode
+from app.agents.multi_agent_chat.shared.filesystem_selection import FilesystemMode
 
 _DESCRIPTION = """Changes the current working directory (cwd).
 

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/cd/index.py

@@ -10,10 +10,10 @@ from langchain_core.messages import ToolMessage
 from langchain_core.tools import BaseTool, StructuredTool
 from langgraph.types import Command
 
+from app.agents.multi_agent_chat.shared.path_resolver import DOCUMENTS_ROOT
 from app.agents.multi_agent_chat.shared.state.filesystem_state import (
     SurfSenseFilesystemState,
 )
-from app.agents.shared.path_resolver import DOCUMENTS_ROOT
 
 from ...middleware.async_dispatch import run_async_blocking
 from ...middleware.path_resolution import resolve_relative

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/edit_file/description.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from app.agents.shared.filesystem_selection import FilesystemMode
+from app.agents.multi_agent_chat.shared.filesystem_selection import FilesystemMode
 
 _CLOUD_DESCRIPTION = """Performs exact string replacements in files.
 

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/execute_code/description.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from app.agents.shared.filesystem_selection import FilesystemMode
+from app.agents.multi_agent_chat.shared.filesystem_selection import FilesystemMode
 
 _DESCRIPTION = """Executes Python code in an isolated sandbox environment.
 

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/execute_code/helpers.py

@@ -14,14 +14,14 @@ from typing import TYPE_CHECKING
 from daytona.common.errors import DaytonaError
 from langchain.tools import ToolRuntime
 
-from app.agents.multi_agent_chat.shared.state.filesystem_state import (
-    SurfSenseFilesystemState,
-)
-from app.agents.shared.sandbox import (
+from app.agents.multi_agent_chat.shared.sandbox import (
     _evict_sandbox_cache,
     delete_sandbox,
     get_or_create_sandbox,
 )
+from app.agents.multi_agent_chat.shared.state.filesystem_state import (
+    SurfSenseFilesystemState,
+)
 
 if TYPE_CHECKING:
     from ...middleware import SurfSenseFilesystemMiddleware

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/glob/description.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from app.agents.shared.filesystem_selection import FilesystemMode
+from app.agents.multi_agent_chat.shared.filesystem_selection import FilesystemMode
 
 _DESCRIPTION = """Find files matching a glob pattern.
 

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/grep/description.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from app.agents.shared.filesystem_selection import FilesystemMode
+from app.agents.multi_agent_chat.shared.filesystem_selection import FilesystemMode
 
 _CLOUD_DESCRIPTION = """Search for a literal text pattern across files.
 

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/list_tree/description.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from app.agents.shared.filesystem_selection import FilesystemMode
+from app.agents.multi_agent_chat.shared.filesystem_selection import FilesystemMode
 
 _CLOUD_DESCRIPTION = """Lists files/folders recursively in a single bounded call.
 

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/ls/description.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from app.agents.shared.filesystem_selection import FilesystemMode
+from app.agents.multi_agent_chat.shared.filesystem_selection import FilesystemMode
 
 _CLOUD_DESCRIPTION = """Lists files and directories at the given path.
 

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/mkdir/description.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from app.agents.shared.filesystem_selection import FilesystemMode
+from app.agents.multi_agent_chat.shared.filesystem_selection import FilesystemMode
 
 _CLOUD_DESCRIPTION = """Creates a directory under `/documents/`.
 

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/mkdir/index.py

@@ -11,10 +11,10 @@ from langchain_core.messages import ToolMessage
 from langchain_core.tools import BaseTool, StructuredTool
 from langgraph.types import Command
 
+from app.agents.multi_agent_chat.shared.path_resolver import DOCUMENTS_ROOT
 from app.agents.multi_agent_chat.shared.state.filesystem_state import (
     SurfSenseFilesystemState,
 )
-from app.agents.shared.path_resolver import DOCUMENTS_ROOT
 
 from ...middleware.async_dispatch import run_async_blocking
 from ...middleware.mode import is_cloud

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/move_file/description.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from app.agents.shared.filesystem_selection import FilesystemMode
+from app.agents.multi_agent_chat.shared.filesystem_selection import FilesystemMode
 
 _CLOUD_DESCRIPTION = """Moves or renames a file or folder.
 

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/move_file/helpers.py

@@ -11,11 +11,11 @@ from langgraph.types import Command
 from app.agents.multi_agent_chat.shared.middleware.filesystem.backends.kb_postgres import (
     KBPostgresBackend,
 )
+from app.agents.multi_agent_chat.shared.path_resolver import DOCUMENTS_ROOT
 from app.agents.multi_agent_chat.shared.state.filesystem_state import (
     SurfSenseFilesystemState,
 )
 from app.agents.multi_agent_chat.shared.state.reducers import _CLEAR
-from app.agents.shared.path_resolver import DOCUMENTS_ROOT
 
 if TYPE_CHECKING:
     from ...middleware import SurfSenseFilesystemMiddleware

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/pwd/description.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from app.agents.shared.filesystem_selection import FilesystemMode
+from app.agents.multi_agent_chat.shared.filesystem_selection import FilesystemMode
 
 _DESCRIPTION = """Prints the current working directory."""
 

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/read_file/description.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from app.agents.shared.filesystem_selection import FilesystemMode
+from app.agents.multi_agent_chat.shared.filesystem_selection import FilesystemMode
 
 _DESCRIPTION = """Reads a file from the filesystem.
 

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/rm/description.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from app.agents.shared.filesystem_selection import FilesystemMode
+from app.agents.multi_agent_chat.shared.filesystem_selection import FilesystemMode
 
 _CLOUD_DESCRIPTION = """Deletes a single file under `/documents/`.
 

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/rm/helpers.py

@@ -15,11 +15,11 @@ from langgraph.types import Command
 from app.agents.multi_agent_chat.shared.middleware.filesystem.backends.kb_postgres import (
     KBPostgresBackend,
 )
+from app.agents.multi_agent_chat.shared.path_resolver import DOCUMENTS_ROOT
 from app.agents.multi_agent_chat.shared.state.filesystem_state import (
     SurfSenseFilesystemState,
 )
 from app.agents.multi_agent_chat.shared.state.reducers import _CLEAR
-from app.agents.shared.path_resolver import DOCUMENTS_ROOT
 
 if TYPE_CHECKING:
     from ...middleware import SurfSenseFilesystemMiddleware

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/rmdir/description.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from app.agents.shared.filesystem_selection import FilesystemMode
+from app.agents.multi_agent_chat.shared.filesystem_selection import FilesystemMode
 
 _CLOUD_DESCRIPTION = """Deletes an empty directory under `/documents/`.
 

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/rmdir/helpers.py

@@ -16,11 +16,11 @@ from langgraph.types import Command
 from app.agents.multi_agent_chat.shared.middleware.filesystem.backends.kb_postgres import (
     KBPostgresBackend,
 )
+from app.agents.multi_agent_chat.shared.path_resolver import DOCUMENTS_ROOT
 from app.agents.multi_agent_chat.shared.state.filesystem_state import (
     SurfSenseFilesystemState,
 )
 from app.agents.multi_agent_chat.shared.state.reducers import _CLEAR
-from app.agents.shared.path_resolver import DOCUMENTS_ROOT
 
 from ...middleware.path_resolution import current_cwd
 from ...shared.paths import is_ancestor_of

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/write_file/description.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from app.agents.shared.filesystem_selection import FilesystemMode
+from app.agents.multi_agent_chat.shared.filesystem_selection import FilesystemMode
 
 _CLOUD_DESCRIPTION = """Writes a new text file to the workspace.
 

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/flags.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from app.agents.shared.feature_flags import AgentFeatureFlags
+from app.agents.multi_agent_chat.shared.feature_flags import AgentFeatureFlags
 
 
 def enabled(flags: AgentFeatureFlags, attr: str) -> bool:

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/kb_persistence.py

@@ -45,19 +45,19 @@ from sqlalchemy import delete, select, update
 from sqlalchemy.exc import IntegrityError
 from sqlalchemy.ext.asyncio import AsyncSession
 
-from app.agents.multi_agent_chat.shared.receipts.receipt import Receipt, make_receipt
-from app.agents.multi_agent_chat.shared.state.filesystem_state import (
-    SurfSenseFilesystemState,
-)
-from app.agents.multi_agent_chat.shared.state.reducers import _CLEAR
-from app.agents.shared.feature_flags import get_flags
-from app.agents.shared.filesystem_selection import FilesystemMode
-from app.agents.shared.path_resolver import (
+from app.agents.multi_agent_chat.shared.feature_flags import get_flags
+from app.agents.multi_agent_chat.shared.filesystem_selection import FilesystemMode
+from app.agents.multi_agent_chat.shared.path_resolver import (
     DOCUMENTS_ROOT,
     parse_documents_path,
     safe_folder_segment,
     virtual_path_to_doc,
 )
+from app.agents.multi_agent_chat.shared.receipts.receipt import Receipt, make_receipt
+from app.agents.multi_agent_chat.shared.state.filesystem_state import (
+    SurfSenseFilesystemState,
+)
+from app.agents.multi_agent_chat.shared.state.reducers import _CLEAR
 from app.db import (
     AgentActionLog,
     Chunk,

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/knowledge_search.py

@@ -45,16 +45,16 @@ from app.agents.multi_agent_chat.shared.date_filters import (
     parse_date_or_datetime,
     resolve_date_range,
 )
-from app.agents.multi_agent_chat.shared.state.filesystem_state import (
-    SurfSenseFilesystemState,
-)
-from app.agents.shared.feature_flags import get_flags
-from app.agents.shared.filesystem_selection import FilesystemMode
-from app.agents.shared.path_resolver import (
+from app.agents.multi_agent_chat.shared.feature_flags import get_flags
+from app.agents.multi_agent_chat.shared.filesystem_selection import FilesystemMode
+from app.agents.multi_agent_chat.shared.path_resolver import (
     PathIndex,
     build_path_index,
     doc_to_virtual_path,
 )
+from app.agents.multi_agent_chat.shared.state.filesystem_state import (
+    SurfSenseFilesystemState,
+)
 from app.db import (
     NATIVE_TO_LEGACY_DOCTYPE,
     Chunk,

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/permissions/deny.py

@@ -11,8 +11,8 @@ from typing import Any
 
 from langchain_core.messages import ToolMessage
 
+from app.agents.multi_agent_chat.shared.errors import StreamingError
 from app.agents.multi_agent_chat.shared.permissions import Rule
-from app.agents.shared.errors import StreamingError
 
 
 def build_deny_message(tool_call: dict[str, Any], rule: Rule) -> ToolMessage:

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/permissions/middleware/core.py

@@ -26,8 +26,8 @@ from langchain_core.messages import AIMessage, ToolMessage
 from langchain_core.tools import BaseTool
 from langgraph.runtime import Runtime
 
+from app.agents.multi_agent_chat.shared.errors import CorrectedError, RejectedError
 from app.agents.multi_agent_chat.shared.permissions import Ruleset
-from app.agents.shared.errors import CorrectedError, RejectedError
 from app.services.user_tool_allowlist import TrustedToolSaver
 
 from ..ask.edit import merge_edited_args

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/permissions/middleware/factory.py

@@ -27,8 +27,8 @@ from collections.abc import Sequence
 
 from langchain_core.tools import BaseTool
 
+from app.agents.multi_agent_chat.shared.feature_flags import AgentFeatureFlags
 from app.agents.multi_agent_chat.shared.permissions import Rule, Ruleset
-from app.agents.shared.feature_flags import AgentFeatureFlags
 from app.services.user_tool_allowlist import TrustedToolSaver
 
 from .core import PermissionMiddleware

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/resilience/bundle.py

@@ -10,7 +10,7 @@ from langchain.agents.middleware import (
     ToolCallLimitMiddleware,
 )
 
-from app.agents.shared.feature_flags import AgentFeatureFlags
+from app.agents.multi_agent_chat.shared.feature_flags import AgentFeatureFlags
 from app.agents.shared.middleware import RetryAfterMiddleware
 
 from .fallback import build_fallback_mw

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/resilience/fallback.py

@@ -4,7 +4,7 @@ from __future__ import annotations
 
 import logging
 
-from app.agents.shared.feature_flags import AgentFeatureFlags
+from app.agents.multi_agent_chat.shared.feature_flags import AgentFeatureFlags
 
 from ..flags import enabled
 from .scoped_model_fallback import (

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/resilience/model_call_limit.py

@@ -4,7 +4,7 @@ from __future__ import annotations
 
 from langchain.agents.middleware import ModelCallLimitMiddleware
 
-from app.agents.shared.feature_flags import AgentFeatureFlags
+from app.agents.multi_agent_chat.shared.feature_flags import AgentFeatureFlags
 
 from ..flags import enabled
 

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/resilience/retry.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from app.agents.shared.feature_flags import AgentFeatureFlags
+from app.agents.multi_agent_chat.shared.feature_flags import AgentFeatureFlags
 from app.agents.shared.middleware import RetryAfterMiddleware
 
 from ..flags import enabled

surfsense_backend/app/agents/multi_agent_chat/shared/middleware/resilience/tool_call_limit.py

@@ -4,7 +4,7 @@ from __future__ import annotations
 
 from langchain.agents.middleware import ToolCallLimitMiddleware
 
-from app.agents.shared.feature_flags import AgentFeatureFlags
+from app.agents.multi_agent_chat.shared.feature_flags import AgentFeatureFlags
 
 from ..flags import enabled
 

surfsense_backend/app/agents/multi_agent_chat/shared/prompt_caching.py

@@ -68,7 +68,7 @@ from typing import TYPE_CHECKING, Any
 from langchain_core.language_models import BaseChatModel
 
 if TYPE_CHECKING:
-    from app.agents.shared.llm_config import AgentConfig
+    from app.agents.multi_agent_chat.shared.llm_config import AgentConfig
 
 logger = logging.getLogger(__name__)
 

surfsense_backend/app/agents/multi_agent_chat/shared/receipts/receipt.py

@@ -23,7 +23,7 @@ the receipt into the parent's ``receipts`` state via the append reducer.
 
 The KB write path is the one exception: file-tool calls cannot emit a
 durable receipt because the actual DB writes happen end-of-turn inside
-:class:`app.agents.shared.middleware.kb_persistence.KnowledgeBasePersistenceMiddleware`.
+:class:`app.agents.multi_agent_chat.shared.middleware.kb_persistence.KnowledgeBasePersistenceMiddleware`.
 KB tools therefore emit a *provisional* receipt with ``status="pending"``;
 the persistence middleware flips it to ``"success"`` or ``"failed"``
 before returning control to the parent.

surfsense_backend/app/agents/multi_agent_chat/subagents/builtins/knowledge_base/agent.py

@@ -13,9 +13,9 @@ from deepagents import SubAgent
 from langchain_core.language_models import BaseChatModel
 from langchain_core.tools import BaseTool
 
+from app.agents.multi_agent_chat.shared.filesystem_selection import FilesystemMode
 from app.agents.multi_agent_chat.shared.permissions import Rule, Ruleset
 from app.agents.multi_agent_chat.subagents.shared.spec import SurfSenseSubagentSpec
-from app.agents.shared.filesystem_selection import FilesystemMode
 
 from .middleware_stack import build_kb_middleware
 from .prompts import load_description, load_readonly_system_prompt, load_system_prompt

surfsense_backend/app/agents/multi_agent_chat/subagents/builtins/knowledge_base/middleware_stack.py

@@ -10,6 +10,8 @@ from typing import Any
 
 from langchain_core.language_models import BaseChatModel
 
+from app.agents.multi_agent_chat.shared.feature_flags import AgentFeatureFlags
+from app.agents.multi_agent_chat.shared.filesystem_selection import FilesystemMode
 from app.agents.multi_agent_chat.shared.middleware.anthropic_cache import (
     build_anthropic_cache_mw,
 )
@@ -29,8 +31,6 @@ from app.agents.multi_agent_chat.shared.middleware.permissions import (
     build_permission_mw,
 )
 from app.agents.multi_agent_chat.shared.permissions import Ruleset
-from app.agents.shared.feature_flags import AgentFeatureFlags
-from app.agents.shared.filesystem_selection import FilesystemMode
 
 
 def _kb_user_allowlist(

surfsense_backend/app/agents/multi_agent_chat/subagents/builtins/knowledge_base/prompts.py

@@ -2,8 +2,8 @@
 
 from __future__ import annotations
 
+from app.agents.multi_agent_chat.shared.filesystem_selection import FilesystemMode
 from app.agents.multi_agent_chat.subagents.shared.md_file_reader import read_md_file
-from app.agents.shared.filesystem_selection import FilesystemMode
 
 
 def load_system_prompt(filesystem_mode: FilesystemMode) -> str:

surfsense_backend/app/agents/multi_agent_chat/subagents/shared/middleware/middleware_stack.py

@@ -14,8 +14,7 @@ from __future__ import annotations
 
 from typing import Any
 
-from app.agents.shared.feature_flags import AgentFeatureFlags
-
+from app.agents.multi_agent_chat.shared.feature_flags import AgentFeatureFlags
 from app.agents.multi_agent_chat.shared.middleware.permissions import (
     build_permission_mw,
 )

surfsense_backend/app/agents/shared/middleware/__init__.py

@@ -1,21 +1,13 @@
 """Shared middleware components for the SurfSense chat agents."""
 
-from app.agents.shared.middleware.busy_mutex import BusyMutexMiddleware
 from app.agents.shared.middleware.compaction import (
     SurfSenseCompactionMiddleware,
     create_surfsense_compaction_middleware,
 )
-from app.agents.shared.middleware.kb_persistence import (
-    KnowledgeBasePersistenceMiddleware,
-    commit_staged_filesystem_state,
-)
 from app.agents.shared.middleware.retry_after import RetryAfterMiddleware
 
 __all__ = [
-    "BusyMutexMiddleware",
-    "KnowledgeBasePersistenceMiddleware",
     "RetryAfterMiddleware",
     "SurfSenseCompactionMiddleware",
-    "commit_staged_filesystem_state",
     "create_surfsense_compaction_middleware",
 ]

surfsense_backend/app/agents/shared/tools/mcp/tool.py

@@ -35,12 +35,12 @@ from sqlalchemy.ext.asyncio import AsyncSession
 
 from app.agents.shared.middleware.dedup_tool_calls import dedup_key_full_args
 from app.agents.shared.tools.hitl import request_approval
-from app.agents.shared.tools.mcp.client import MCPClient
 from app.agents.shared.tools.mcp.cache import (
     CachedMCPTools,
     read_cached_tools,
     write_cached_tools,
 )
+from app.agents.shared.tools.mcp.client import MCPClient
 from app.db import SearchSourceConnector
 from app.services.mcp_oauth.registry import MCP_SERVICES, get_service_by_connector_type
 from app.utils.perf import get_perf_logger