apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-04-28 10:26:33 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

refactor: enhance MemoryInjectionMiddleware to auto-persist user memory seed and improve button styles in MemoryContent and TeamMemoryManager components

Browse source
This commit is contained in:
Anish Sarkar 2026-04-09 13:57:54 +05:30
parent 1f59382ab0
commit d1c6caddba
3 changed files with 21 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

19
surfsense_backend/app/agents/new_chat/middleware/memory_injection.py
View file

@ -19,8 +19,8 @@ from langgraph.runtime import Runtime
from sqlalchemy import select
from sqlalchemy.ext.asyncio import AsyncSession
from app.db import ChatVisibility, SearchSpace, User, shielded_async_session
from app.agents.new_chat.tools.update_memory import MEMORY_HARD_LIMIT
from app.db import ChatVisibility, SearchSpace, User, shielded_async_session
logger = logging.getLogger(__name__)
@ -96,10 +96,9 @@ class MemoryInjectionMiddleware(AgentMiddleware): # type: ignore[type-arg]
) -> tuple[str | None, bool]:
"""Return (memory_content, is_persisted).
When the user has saved memory in the database, ``is_persisted`` is
``True``. When we fall back to a seed (first-name only), it is
``False`` — the system prompt instructs the LLM to call
``update_memory`` once to persist it.
When the user has no saved memory but has a display name, a seed
document is created and **persisted to the database immediately**
so the LLM doesn't need to make a tool call to save it.
"""
try:
result = await session.execute(
@ -118,7 +117,15 @@ class MemoryInjectionMiddleware(AgentMiddleware): # type: ignore[type-arg]
if display_name:
first_name = display_name.split()[0]
return f"## About the user\n- Name: {first_name}", False
seed = f"## About the user\n- Name: {first_name}"
await session.execute(
User.__table__.update()
.where(User.id == self.user_id)
.values(memory_md=seed)
)
await session.commit()
logger.info("Auto-persisted memory seed for user %s", self.user_id)
return seed, True
return None, True
except Exception: