apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-05-05 22:02:39 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

feat: integrate MemoryInjectionMiddleware and update memory tool instructions

Browse source
This commit is contained in:
Anish Sarkar 2026-04-08 23:23:06 +05:30
parent ba7e288879
commit 702d955e79
2 changed files with 65 additions and 95 deletions
Download patch file Download diff file Expand all files Collapse all files

9
surfsense_backend/app/agents/new_chat/chat_deepagent.py
View file

@ -38,6 +38,7 @@ from app.agents.new_chat.llm_config import AgentConfig
from app.agents.new_chat.middleware import (
DedupHITLToolCallsMiddleware,
KnowledgeBaseSearchMiddleware,
MemoryInjectionMiddleware,
SurfSenseFilesystemMiddleware,
)
from app.agents.new_chat.system_prompt import (
@ -425,9 +426,16 @@ async def create_surfsense_deep_agent(
)
# -- Build the middleware stack (mirrors create_deep_agent internals) ------
_memory_middleware = MemoryInjectionMiddleware(
user_id=user_id,
search_space_id=search_space_id,
thread_visibility=visibility,
)
# General-purpose subagent middleware
gp_middleware = [
TodoListMiddleware(),
_memory_middleware,
SurfSenseFilesystemMiddleware(
search_space_id=search_space_id,
created_by_id=user_id,
@ -447,6 +455,7 @@ async def create_surfsense_deep_agent(
# Main agent middleware
deepagent_middleware = [
TodoListMiddleware(),
_memory_middleware,
KnowledgeBaseSearchMiddleware(
llm=llm,
search_space_id=search_space_id,