mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-06-06 20:15:17 +02:00
82c5dc5b02
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.
70 lines
2.4 KiB
Python
70 lines
2.4 KiB
Python
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from app.agents.multi_agent_chat.shared.filesystem_selection import (
|
|
ClientPlatform,
|
|
FilesystemMode,
|
|
FilesystemSelection,
|
|
LocalFilesystemMount,
|
|
)
|
|
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,
|
|
)
|
|
|
|
pytestmark = pytest.mark.unit
|
|
|
|
|
|
class _RuntimeStub:
|
|
state = {"files": {}}
|
|
|
|
|
|
def test_backend_resolver_returns_multi_root_backend_for_single_root(tmp_path: Path):
|
|
selection = FilesystemSelection(
|
|
mode=FilesystemMode.DESKTOP_LOCAL_FOLDER,
|
|
client_platform=ClientPlatform.DESKTOP,
|
|
local_mounts=(LocalFilesystemMount(mount_id="tmp", root_path=str(tmp_path)),),
|
|
)
|
|
resolver = build_backend_resolver(selection)
|
|
|
|
backend = resolver(_RuntimeStub())
|
|
assert isinstance(backend, MultiRootLocalFolderBackend)
|
|
assert backend.list_mounts() == ("tmp",)
|
|
|
|
|
|
def test_backend_resolver_uses_cloud_mode_by_default():
|
|
resolver = build_backend_resolver(FilesystemSelection())
|
|
backend = resolver(_RuntimeStub())
|
|
# When no search_space_id is provided we fall back to StateBackend so
|
|
# sub-agents / tests without DB access still work.
|
|
assert backend.__class__.__name__ == "StateBackend"
|
|
|
|
|
|
def test_backend_resolver_uses_kb_postgres_in_cloud_with_search_space():
|
|
resolver = build_backend_resolver(FilesystemSelection(), search_space_id=42)
|
|
backend = resolver(_RuntimeStub())
|
|
assert backend.__class__.__name__ == "KBPostgresBackend"
|
|
assert backend.search_space_id == 42
|
|
|
|
|
|
def test_backend_resolver_returns_multi_root_backend_for_multiple_roots(tmp_path: Path):
|
|
root_one = tmp_path / "resume"
|
|
root_two = tmp_path / "notes"
|
|
root_one.mkdir()
|
|
root_two.mkdir()
|
|
selection = FilesystemSelection(
|
|
mode=FilesystemMode.DESKTOP_LOCAL_FOLDER,
|
|
client_platform=ClientPlatform.DESKTOP,
|
|
local_mounts=(
|
|
LocalFilesystemMount(mount_id="resume", root_path=str(root_one)),
|
|
LocalFilesystemMount(mount_id="notes", root_path=str(root_two)),
|
|
),
|
|
)
|
|
resolver = build_backend_resolver(selection)
|
|
|
|
backend = resolver(_RuntimeStub())
|
|
assert isinstance(backend, MultiRootLocalFolderBackend)
|
|
assert backend.list_mounts() == ("resume", "notes")
|