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.
49 lines
1.6 KiB
Python
49 lines
1.6 KiB
Python
"""Single per-thread agent (re)build path.
|
|
|
|
A graph swap mid-turn would corrupt checkpointer state for the same
|
|
``thread_id``, so both the initial build and any mid-stream 429 recovery rebuild
|
|
must funnel through this single function.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
from app.agents.multi_agent_chat.shared.filesystem_selection import FilesystemSelection
|
|
from app.agents.multi_agent_chat.shared.llm_config import AgentConfig
|
|
from app.db import ChatVisibility
|
|
from app.services.connector_service import ConnectorService
|
|
|
|
|
|
async def build_main_agent_for_thread(
|
|
agent_factory: Any,
|
|
*,
|
|
llm: Any,
|
|
search_space_id: int,
|
|
db_session: Any,
|
|
connector_service: ConnectorService,
|
|
checkpointer: Any,
|
|
user_id: str | None,
|
|
thread_id: int | None,
|
|
agent_config: AgentConfig | None,
|
|
firecrawl_api_key: str | None,
|
|
thread_visibility: ChatVisibility | None,
|
|
filesystem_selection: FilesystemSelection | None,
|
|
disabled_tools: list[str] | None = None,
|
|
mentioned_document_ids: list[int] | None = None,
|
|
) -> Any:
|
|
return await agent_factory(
|
|
llm=llm,
|
|
search_space_id=search_space_id,
|
|
db_session=db_session,
|
|
connector_service=connector_service,
|
|
checkpointer=checkpointer,
|
|
user_id=user_id,
|
|
thread_id=thread_id,
|
|
agent_config=agent_config,
|
|
firecrawl_api_key=firecrawl_api_key,
|
|
thread_visibility=thread_visibility,
|
|
filesystem_selection=filesystem_selection,
|
|
disabled_tools=disabled_tools,
|
|
mentioned_document_ids=mentioned_document_ids,
|
|
)
|