mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-06-06 20:15:17 +02:00
refactor(agents): move memory middleware into main_agent (owner)
memory (builder) + memory_injection (impl) lived in shared/middleware/ but are consumed only by main_agent (no subagent, no shared plumbing). Colocate with their owner using the folder-per-middleware shape: shared/middleware/memory.py -> main_agent/middleware/memory/builder.py shared/middleware/memory_injection.py -> main_agent/middleware/memory/middleware.py
This commit is contained in:
parent
1a52166145
commit
6b1da64182
4 changed files with 9 additions and 2 deletions
|
|
@ -0,0 +1,5 @@
|
||||||
|
"""User/team memory injection middleware (main-agent only)."""
|
||||||
|
|
||||||
|
from .builder import build_memory_mw
|
||||||
|
|
||||||
|
__all__ = ["build_memory_mw"]
|
||||||
|
|
@ -4,7 +4,7 @@ from __future__ import annotations
|
||||||
|
|
||||||
from app.db import ChatVisibility
|
from app.db import ChatVisibility
|
||||||
|
|
||||||
from .memory_injection import MemoryInjectionMiddleware
|
from .middleware import MemoryInjectionMiddleware
|
||||||
|
|
||||||
|
|
||||||
def build_memory_mw(
|
def build_memory_mw(
|
||||||
|
|
@ -20,6 +20,9 @@ from langchain_core.language_models import BaseChatModel
|
||||||
from langchain_core.tools import BaseTool
|
from langchain_core.tools import BaseTool
|
||||||
from langgraph.types import Checkpointer
|
from langgraph.types import Checkpointer
|
||||||
|
|
||||||
|
from app.agents.chat.multi_agent_chat.main_agent.middleware.memory import (
|
||||||
|
build_memory_mw,
|
||||||
|
)
|
||||||
from app.agents.chat.multi_agent_chat.shared.feature_flags import AgentFeatureFlags
|
from app.agents.chat.multi_agent_chat.shared.feature_flags import AgentFeatureFlags
|
||||||
from app.agents.chat.multi_agent_chat.shared.filesystem_selection import FilesystemMode
|
from app.agents.chat.multi_agent_chat.shared.filesystem_selection import FilesystemMode
|
||||||
from app.agents.chat.multi_agent_chat.shared.middleware.anthropic_cache import (
|
from app.agents.chat.multi_agent_chat.shared.middleware.anthropic_cache import (
|
||||||
|
|
@ -31,7 +34,6 @@ from app.agents.chat.multi_agent_chat.shared.middleware.compaction import (
|
||||||
from app.agents.chat.multi_agent_chat.shared.middleware.kb_context_projection import (
|
from app.agents.chat.multi_agent_chat.shared.middleware.kb_context_projection import (
|
||||||
build_kb_context_projection_mw,
|
build_kb_context_projection_mw,
|
||||||
)
|
)
|
||||||
from app.agents.chat.multi_agent_chat.shared.middleware.memory import build_memory_mw
|
|
||||||
from app.agents.chat.multi_agent_chat.shared.middleware.patch_tool_calls import (
|
from app.agents.chat.multi_agent_chat.shared.middleware.patch_tool_calls import (
|
||||||
build_patch_tool_calls_mw,
|
build_patch_tool_calls_mw,
|
||||||
)
|
)
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue