From 21509e7eca6abbbc59ea63ef2fca5fab4dca8b9a Mon Sep 17 00:00:00 2001 From: CREDO23 Date: Fri, 5 Jun 2026 11:02:26 +0200 Subject: [PATCH] refactor(agents): group filesystem backends under filesystem/backends/ The concrete filesystem backends are consumed only by the MAC filesystem layer (tools, path-resolution middleware, the resolver, skills backend) and tests -- no external app code. Group them next to the filesystem middleware they serve: - filesystem_backends.py -> filesystem/backends/resolver.py - middleware/kb_postgres_backend.py -> filesystem/backends/kb_postgres.py - middleware/local_folder_backend.py -> filesystem/backends/local_folder.py - middleware/multi_root_local_folder_backend.py -> .../multi_root_local_folder.py - document_xml.py -> filesystem/backends/document_xml.py Repoint all 21 importers. No behavior change; import-all + filesystem backend/path-resolution/knowledge-search unit tests stay green (478). --- .../multi_agent_chat/main_agent/runtime/factory.py | 4 +++- .../multi_agent_chat/main_agent/skills/backends.py | 6 ++++-- .../shared/middleware/filesystem/backends/__init__.py | 0 .../middleware/filesystem/backends}/document_xml.py | 0 .../middleware/filesystem/backends/kb_postgres.py} | 4 +++- .../middleware/filesystem/backends/local_folder.py} | 0 .../filesystem/backends/multi_root_local_folder.py} | 4 +++- .../shared/middleware/filesystem/backends/resolver.py} | 8 +++++--- .../filesystem/middleware/path_resolution.py | 6 +++--- .../middleware/filesystem/tools/edit_file/index.py | 4 +++- .../middleware/filesystem/tools/list_tree/index.py | 4 +++- .../shared/middleware/filesystem/tools/ls/index.py | 4 +++- .../middleware/filesystem/tools/move_file/helpers.py | 4 +++- .../middleware/filesystem/tools/read_file/index.py | 4 +++- .../shared/middleware/filesystem/tools/rm/helpers.py | 4 +++- .../middleware/filesystem/tools/rmdir/helpers.py | 4 +++- .../multi_agent_chat/test_kb_filesystem_cloud.py | 4 +++- .../multi_agent_chat/test_kb_filesystem_desktop.py | 4 +++- .../middleware/test_b_filesystem_path_resolution.py | 6 +++--- .../middleware/test_b_filesystem_rm_rmdir_cloud.py | 8 ++++++-- .../tests/unit/middleware/test_filesystem_backends.py | 10 ++++++---- .../tests/unit/middleware/test_knowledge_search.py | 4 +++- .../tests/unit/middleware/test_local_folder_backend.py | 4 +++- .../middleware/test_multi_root_local_folder_backend.py | 2 +- 24 files changed, 70 insertions(+), 32 deletions(-) create mode 100644 surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/backends/__init__.py rename surfsense_backend/app/agents/{shared => multi_agent_chat/shared/middleware/filesystem/backends}/document_xml.py (100%) rename surfsense_backend/app/agents/{shared/middleware/kb_postgres_backend.py => multi_agent_chat/shared/middleware/filesystem/backends/kb_postgres.py} (99%) rename surfsense_backend/app/agents/{shared/middleware/local_folder_backend.py => multi_agent_chat/shared/middleware/filesystem/backends/local_folder.py} (100%) rename surfsense_backend/app/agents/{shared/middleware/multi_root_local_folder_backend.py => multi_agent_chat/shared/middleware/filesystem/backends/multi_root_local_folder.py} (99%) rename surfsense_backend/app/agents/{shared/filesystem_backends.py => multi_agent_chat/shared/middleware/filesystem/backends/resolver.py} (90%) diff --git a/surfsense_backend/app/agents/multi_agent_chat/main_agent/runtime/factory.py b/surfsense_backend/app/agents/multi_agent_chat/main_agent/runtime/factory.py index d692a3d94..2280ec649 100644 --- a/surfsense_backend/app/agents/multi_agent_chat/main_agent/runtime/factory.py +++ b/surfsense_backend/app/agents/multi_agent_chat/main_agent/runtime/factory.py @@ -12,6 +12,9 @@ from langchain_core.tools import BaseTool from langgraph.types import Checkpointer from sqlalchemy.ext.asyncio import AsyncSession +from app.agents.multi_agent_chat.shared.middleware.filesystem.backends.resolver import ( + build_backend_resolver, +) from app.agents.multi_agent_chat.subagents import ( get_subagents_to_exclude, main_prompt_registry_subagent_lines, @@ -20,7 +23,6 @@ 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_backends import build_backend_resolver 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 diff --git a/surfsense_backend/app/agents/multi_agent_chat/main_agent/skills/backends.py b/surfsense_backend/app/agents/multi_agent_chat/main_agent/skills/backends.py index 51528e686..4b6179158 100644 --- a/surfsense_backend/app/agents/multi_agent_chat/main_agent/skills/backends.py +++ b/surfsense_backend/app/agents/multi_agent_chat/main_agent/skills/backends.py @@ -47,7 +47,9 @@ from deepagents.backends.state import StateBackend if TYPE_CHECKING: from langchain.tools import ToolRuntime - from app.agents.shared.middleware.kb_postgres_backend import KBPostgresBackend + from app.agents.multi_agent_chat.shared.middleware.filesystem.backends.kb_postgres import ( + KBPostgresBackend, + ) logger = logging.getLogger(__name__) @@ -308,7 +310,7 @@ def build_skills_backend_factory( # Imported lazily to avoid a hard dependency at module import time: # ``KBPostgresBackend`` pulls in DB models, which are unnecessary for # the unit-tested builtin path. - from app.agents.shared.middleware.kb_postgres_backend import ( + from app.agents.multi_agent_chat.shared.middleware.filesystem.backends.kb_postgres import ( KBPostgresBackend, ) diff --git a/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/backends/__init__.py b/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/backends/__init__.py new file mode 100644 index 000000000..e69de29bb diff --git a/surfsense_backend/app/agents/shared/document_xml.py b/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/backends/document_xml.py similarity index 100% rename from surfsense_backend/app/agents/shared/document_xml.py rename to surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/backends/document_xml.py diff --git a/surfsense_backend/app/agents/shared/middleware/kb_postgres_backend.py b/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/backends/kb_postgres.py similarity index 99% rename from surfsense_backend/app/agents/shared/middleware/kb_postgres_backend.py rename to surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/backends/kb_postgres.py index 9e29ecab4..1111a303d 100644 --- a/surfsense_backend/app/agents/shared/middleware/kb_postgres_backend.py +++ b/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/backends/kb_postgres.py @@ -42,7 +42,9 @@ from langchain.tools import ToolRuntime from sqlalchemy import select from sqlalchemy.ext.asyncio import AsyncSession -from app.agents.shared.document_xml import build_document_xml +from app.agents.multi_agent_chat.shared.middleware.filesystem.backends.document_xml import ( + build_document_xml, +) from app.agents.shared.path_resolver import ( DOCUMENTS_ROOT, build_path_index, diff --git a/surfsense_backend/app/agents/shared/middleware/local_folder_backend.py b/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/backends/local_folder.py similarity index 100% rename from surfsense_backend/app/agents/shared/middleware/local_folder_backend.py rename to surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/backends/local_folder.py diff --git a/surfsense_backend/app/agents/shared/middleware/multi_root_local_folder_backend.py b/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/backends/multi_root_local_folder.py similarity index 99% rename from surfsense_backend/app/agents/shared/middleware/multi_root_local_folder_backend.py rename to surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/backends/multi_root_local_folder.py index 220890fea..e443995f4 100644 --- a/surfsense_backend/app/agents/shared/middleware/multi_root_local_folder_backend.py +++ b/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/backends/multi_root_local_folder.py @@ -15,7 +15,9 @@ from deepagents.backends.protocol import ( WriteResult, ) -from app.agents.shared.middleware.local_folder_backend import LocalFolderBackend +from app.agents.multi_agent_chat.shared.middleware.filesystem.backends.local_folder import ( + LocalFolderBackend, +) _INVALID_PATH = "invalid_path" _FILE_NOT_FOUND = "file_not_found" diff --git a/surfsense_backend/app/agents/shared/filesystem_backends.py b/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/backends/resolver.py similarity index 90% rename from surfsense_backend/app/agents/shared/filesystem_backends.py rename to surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/backends/resolver.py index c20eaea5f..d244a1820 100644 --- a/surfsense_backend/app/agents/shared/filesystem_backends.py +++ b/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/backends/resolver.py @@ -9,11 +9,13 @@ from deepagents.backends.protocol import BackendProtocol from deepagents.backends.state import StateBackend from langgraph.prebuilt.tool_node import ToolRuntime -from app.agents.shared.filesystem_selection import FilesystemMode, FilesystemSelection -from app.agents.shared.middleware.kb_postgres_backend import KBPostgresBackend -from app.agents.shared.middleware.multi_root_local_folder_backend import ( +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) diff --git a/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/middleware/path_resolution.py b/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/middleware/path_resolution.py index a3bd62e72..64a73bc9b 100644 --- a/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/middleware/path_resolution.py +++ b/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.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 app.agents.shared.middleware.multi_root_local_folder_backend import ( - MultiRootLocalFolderBackend, -) from ..shared.paths import ( extract_mount_from_path, diff --git a/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/edit_file/index.py b/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/edit_file/index.py index a13a11f2b..ba489913b 100644 --- a/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/edit_file/index.py +++ b/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/edit_file/index.py @@ -11,10 +11,12 @@ 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.middleware.filesystem.backends.kb_postgres import ( + KBPostgresBackend, +) from app.agents.multi_agent_chat.shared.state.filesystem_state import ( SurfSenseFilesystemState, ) -from app.agents.shared.middleware.kb_postgres_backend import KBPostgresBackend from ...middleware.async_dispatch import run_async_blocking from ...middleware.mode import is_cloud diff --git a/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/list_tree/index.py b/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/list_tree/index.py index 97579351f..f3440f0e9 100644 --- a/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/list_tree/index.py +++ b/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/list_tree/index.py @@ -9,10 +9,12 @@ from deepagents.backends.utils import validate_path from langchain.tools import ToolRuntime from langchain_core.tools import BaseTool, StructuredTool +from app.agents.multi_agent_chat.shared.middleware.filesystem.backends.kb_postgres import ( + KBPostgresBackend, +) from app.agents.multi_agent_chat.shared.state.filesystem_state import ( SurfSenseFilesystemState, ) -from app.agents.shared.middleware.kb_postgres_backend import KBPostgresBackend from ...middleware.async_dispatch import run_async_blocking from ...middleware.path_resolution import resolve_list_target_path diff --git a/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/ls/index.py b/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/ls/index.py index ab1618d01..149a20da8 100644 --- a/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/ls/index.py +++ b/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/ls/index.py @@ -8,10 +8,12 @@ from deepagents.backends.utils import validate_path from langchain.tools import ToolRuntime from langchain_core.tools import BaseTool, StructuredTool +from app.agents.multi_agent_chat.shared.middleware.filesystem.backends.kb_postgres import ( + paginate_listing, +) from app.agents.multi_agent_chat.shared.state.filesystem_state import ( SurfSenseFilesystemState, ) -from app.agents.shared.middleware.kb_postgres_backend import paginate_listing from ...middleware.async_dispatch import run_async_blocking from ...middleware.path_resolution import resolve_list_target_path diff --git a/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/move_file/helpers.py b/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/move_file/helpers.py index 5f9dc1154..cca8ef630 100644 --- a/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/move_file/helpers.py +++ b/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/move_file/helpers.py @@ -8,11 +8,13 @@ from langchain.tools import ToolRuntime from langchain_core.messages import ToolMessage 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.state.filesystem_state import ( SurfSenseFilesystemState, ) from app.agents.multi_agent_chat.shared.state.reducers import _CLEAR -from app.agents.shared.middleware.kb_postgres_backend import KBPostgresBackend from app.agents.shared.path_resolver import DOCUMENTS_ROOT if TYPE_CHECKING: diff --git a/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/read_file/index.py b/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/read_file/index.py index bc673adce..f16332712 100644 --- a/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/read_file/index.py +++ b/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/read_file/index.py @@ -10,10 +10,12 @@ 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.middleware.filesystem.backends.kb_postgres import ( + KBPostgresBackend, +) from app.agents.multi_agent_chat.shared.state.filesystem_state import ( SurfSenseFilesystemState, ) -from app.agents.shared.middleware.kb_postgres_backend import KBPostgresBackend from ...middleware.async_dispatch import run_async_blocking from ...middleware.path_resolution import resolve_relative diff --git a/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/rm/helpers.py b/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/rm/helpers.py index 55530eda0..3b4fd8a6a 100644 --- a/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/rm/helpers.py +++ b/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/rm/helpers.py @@ -12,11 +12,13 @@ from langchain.tools import ToolRuntime from langchain_core.messages import ToolMessage 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.state.filesystem_state import ( SurfSenseFilesystemState, ) from app.agents.multi_agent_chat.shared.state.reducers import _CLEAR -from app.agents.shared.middleware.kb_postgres_backend import KBPostgresBackend from app.agents.shared.path_resolver import DOCUMENTS_ROOT if TYPE_CHECKING: diff --git a/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/rmdir/helpers.py b/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/rmdir/helpers.py index b269e8aa0..080103b7e 100644 --- a/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/rmdir/helpers.py +++ b/surfsense_backend/app/agents/multi_agent_chat/shared/middleware/filesystem/tools/rmdir/helpers.py @@ -13,11 +13,13 @@ from langchain.tools import ToolRuntime from langchain_core.messages import ToolMessage 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.state.filesystem_state import ( SurfSenseFilesystemState, ) from app.agents.multi_agent_chat.shared.state.reducers import _CLEAR -from app.agents.shared.middleware.kb_postgres_backend import KBPostgresBackend from app.agents.shared.path_resolver import DOCUMENTS_ROOT from ...middleware.path_resolution import current_cwd diff --git a/surfsense_backend/tests/integration/agents/multi_agent_chat/test_kb_filesystem_cloud.py b/surfsense_backend/tests/integration/agents/multi_agent_chat/test_kb_filesystem_cloud.py index c6b107bc6..a592e99dc 100644 --- a/surfsense_backend/tests/integration/agents/multi_agent_chat/test_kb_filesystem_cloud.py +++ b/surfsense_backend/tests/integration/agents/multi_agent_chat/test_kb_filesystem_cloud.py @@ -23,7 +23,9 @@ from langgraph.checkpoint.memory import InMemorySaver from app.agents.multi_agent_chat.shared.middleware.filesystem import ( build_filesystem_mw, ) -from app.agents.shared.filesystem_backends import build_backend_resolver +from app.agents.multi_agent_chat.shared.middleware.filesystem.backends.resolver import ( + build_backend_resolver, +) from app.agents.shared.filesystem_selection import FilesystemMode, FilesystemSelection from tests.integration.harness import ScriptedTurn, build_scripted_harness diff --git a/surfsense_backend/tests/integration/agents/multi_agent_chat/test_kb_filesystem_desktop.py b/surfsense_backend/tests/integration/agents/multi_agent_chat/test_kb_filesystem_desktop.py index 4c4f98022..3c43969ac 100644 --- a/surfsense_backend/tests/integration/agents/multi_agent_chat/test_kb_filesystem_desktop.py +++ b/surfsense_backend/tests/integration/agents/multi_agent_chat/test_kb_filesystem_desktop.py @@ -26,7 +26,9 @@ from langgraph.checkpoint.memory import InMemorySaver from app.agents.multi_agent_chat.shared.middleware.filesystem import ( build_filesystem_mw, ) -from app.agents.shared.filesystem_backends import build_backend_resolver +from app.agents.multi_agent_chat.shared.middleware.filesystem.backends.resolver import ( + build_backend_resolver, +) from app.agents.shared.filesystem_selection import ( FilesystemMode, FilesystemSelection, diff --git a/surfsense_backend/tests/unit/middleware/test_b_filesystem_path_resolution.py b/surfsense_backend/tests/unit/middleware/test_b_filesystem_path_resolution.py index d68db8d72..799ce8fc8 100644 --- a/surfsense_backend/tests/unit/middleware/test_b_filesystem_path_resolution.py +++ b/surfsense_backend/tests/unit/middleware/test_b_filesystem_path_resolution.py @@ -20,6 +20,9 @@ from types import SimpleNamespace import pytest +from app.agents.multi_agent_chat.shared.middleware.filesystem.backends.multi_root_local_folder import ( + MultiRootLocalFolderBackend, +) from app.agents.multi_agent_chat.shared.middleware.filesystem.middleware.mode import ( default_cwd, ) @@ -33,9 +36,6 @@ from app.agents.multi_agent_chat.shared.middleware.filesystem.middleware.path_re resolve_relative, ) from app.agents.shared.filesystem_selection import FilesystemMode -from app.agents.shared.middleware.multi_root_local_folder_backend import ( - MultiRootLocalFolderBackend, -) pytestmark = pytest.mark.unit diff --git a/surfsense_backend/tests/unit/middleware/test_b_filesystem_rm_rmdir_cloud.py b/surfsense_backend/tests/unit/middleware/test_b_filesystem_rm_rmdir_cloud.py index 634584cb7..4b43ef3be 100644 --- a/surfsense_backend/tests/unit/middleware/test_b_filesystem_rm_rmdir_cloud.py +++ b/surfsense_backend/tests/unit/middleware/test_b_filesystem_rm_rmdir_cloud.py @@ -22,10 +22,14 @@ import pytest from app.agents.multi_agent_chat.shared.middleware.filesystem import ( build_filesystem_mw, ) +from app.agents.multi_agent_chat.shared.middleware.filesystem.backends.kb_postgres import ( + KBPostgresBackend, +) +from app.agents.multi_agent_chat.shared.middleware.filesystem.backends.resolver import ( + build_backend_resolver, +) from app.agents.multi_agent_chat.shared.state.reducers import _CLEAR -from app.agents.shared.filesystem_backends import build_backend_resolver from app.agents.shared.filesystem_selection import FilesystemMode, FilesystemSelection -from app.agents.shared.middleware.kb_postgres_backend import KBPostgresBackend pytestmark = pytest.mark.unit diff --git a/surfsense_backend/tests/unit/middleware/test_filesystem_backends.py b/surfsense_backend/tests/unit/middleware/test_filesystem_backends.py index 999102859..cafef3a83 100644 --- a/surfsense_backend/tests/unit/middleware/test_filesystem_backends.py +++ b/surfsense_backend/tests/unit/middleware/test_filesystem_backends.py @@ -2,16 +2,18 @@ from pathlib import Path import pytest -from app.agents.shared.filesystem_backends import build_backend_resolver +from app.agents.multi_agent_chat.shared.middleware.filesystem.backends.multi_root_local_folder import ( + MultiRootLocalFolderBackend, +) +from app.agents.multi_agent_chat.shared.middleware.filesystem.backends.resolver import ( + build_backend_resolver, +) from app.agents.shared.filesystem_selection import ( ClientPlatform, FilesystemMode, FilesystemSelection, LocalFilesystemMount, ) -from app.agents.shared.middleware.multi_root_local_folder_backend import ( - MultiRootLocalFolderBackend, -) pytestmark = pytest.mark.unit diff --git a/surfsense_backend/tests/unit/middleware/test_knowledge_search.py b/surfsense_backend/tests/unit/middleware/test_knowledge_search.py index 72f54f400..c97bcde0a 100644 --- a/surfsense_backend/tests/unit/middleware/test_knowledge_search.py +++ b/surfsense_backend/tests/unit/middleware/test_knowledge_search.py @@ -5,7 +5,9 @@ import json import pytest from langchain_core.messages import AIMessage, HumanMessage -from app.agents.shared.document_xml import build_document_xml as _build_document_xml +from app.agents.multi_agent_chat.shared.middleware.filesystem.backends.document_xml import ( + build_document_xml as _build_document_xml, +) from app.agents.shared.middleware.knowledge_search import ( KBSearchPlan, KnowledgeBaseSearchMiddleware, diff --git a/surfsense_backend/tests/unit/middleware/test_local_folder_backend.py b/surfsense_backend/tests/unit/middleware/test_local_folder_backend.py index 2111e2f66..7ceb19fa8 100644 --- a/surfsense_backend/tests/unit/middleware/test_local_folder_backend.py +++ b/surfsense_backend/tests/unit/middleware/test_local_folder_backend.py @@ -2,7 +2,9 @@ from pathlib import Path import pytest -from app.agents.shared.middleware.local_folder_backend import LocalFolderBackend +from app.agents.multi_agent_chat.shared.middleware.filesystem.backends.local_folder import ( + LocalFolderBackend, +) pytestmark = pytest.mark.unit diff --git a/surfsense_backend/tests/unit/middleware/test_multi_root_local_folder_backend.py b/surfsense_backend/tests/unit/middleware/test_multi_root_local_folder_backend.py index 2176698ec..dbfbb6ac7 100644 --- a/surfsense_backend/tests/unit/middleware/test_multi_root_local_folder_backend.py +++ b/surfsense_backend/tests/unit/middleware/test_multi_root_local_folder_backend.py @@ -2,7 +2,7 @@ from pathlib import Path import pytest -from app.agents.shared.middleware.multi_root_local_folder_backend import ( +from app.agents.multi_agent_chat.shared.middleware.filesystem.backends.multi_root_local_folder import ( MultiRootLocalFolderBackend, )