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

refactor: rename and consolidate memory tools to update_memory

Browse source
This commit is contained in:
Anish Sarkar 2026-04-08 23:29:26 +05:30
parent 5ff99115c6
commit 6fc941e4c5
2 changed files with 10 additions and 34 deletions
Download patch file Download diff file Expand all files Collapse all files

9
surfsense_backend/app/agents/new_chat/tools/__init__.py
View file

@ -10,8 +10,7 @@ Available tools:
- generate_video_presentation: Generate video presentations with slides and narration - generate_video_presentation: Generate video presentations with slides and narration
- generate_image: Generate images from text descriptions using AI models - generate_image: Generate images from text descriptions using AI models
- scrape_webpage: Extract content from webpages - scrape_webpage: Extract content from webpages
- save_memory: Store facts/preferences about the user - update_memory: Update the user's / team's memory document
- recall_memory: Retrieve relevant user memories
""" """
# Registry exports # Registry exports
@ -33,7 +32,7 @@ from .registry import (
) )
from .scrape_webpage import create_scrape_webpage_tool from .scrape_webpage import create_scrape_webpage_tool
from .search_surfsense_docs import create_search_surfsense_docs_tool from .search_surfsense_docs import create_search_surfsense_docs_tool
from .user_memory import create_recall_memory_tool, create_save_memory_tool from .update_memory import create_update_memory_tool, create_update_team_memory_tool
from .video_presentation import create_generate_video_presentation_tool from .video_presentation import create_generate_video_presentation_tool
__all__ = [ __all__ = [
@ -47,10 +46,10 @@ __all__ = [
"create_generate_image_tool", "create_generate_image_tool",
"create_generate_podcast_tool", "create_generate_podcast_tool",
"create_generate_video_presentation_tool", "create_generate_video_presentation_tool",
"create_recall_memory_tool",
"create_save_memory_tool",
"create_scrape_webpage_tool", "create_scrape_webpage_tool",
"create_search_surfsense_docs_tool", "create_search_surfsense_docs_tool",
"create_update_memory_tool",
"create_update_team_memory_tool",
"format_documents_for_context", "format_documents_for_context",
"get_all_tool_names", "get_all_tool_names",
"get_default_enabled_tools", "get_default_enabled_tools",

35
surfsense_backend/app/agents/new_chat/tools/registry.py
View file

@ -94,11 +94,7 @@ from .podcast import create_generate_podcast_tool
from .report import create_generate_report_tool from .report import create_generate_report_tool
from .scrape_webpage import create_scrape_webpage_tool from .scrape_webpage import create_scrape_webpage_tool
from .search_surfsense_docs import create_search_surfsense_docs_tool from .search_surfsense_docs import create_search_surfsense_docs_tool
from .shared_memory import ( from .update_memory import create_update_memory_tool, create_update_team_memory_tool
create_recall_shared_memory_tool,
create_save_shared_memory_tool,
)
from .user_memory import create_recall_memory_tool, create_save_memory_tool
from .video_presentation import create_generate_video_presentation_tool from .video_presentation import create_generate_video_presentation_tool
from .web_search import create_web_search_tool from .web_search import create_web_search_tool
@ -214,38 +210,19 @@ BUILTIN_TOOLS: list[ToolDefinition] = [
requires=["db_session"], requires=["db_session"],
), ),
# ========================================================================= # =========================================================================
# USER MEMORY TOOLS - private or team store by thread_visibility # MEMORY TOOL - single update_memory, private or team by thread_visibility
# ========================================================================= # =========================================================================
ToolDefinition( ToolDefinition(
name="save_memory", name="update_memory",
description="Save facts, preferences, or context for personalized or team responses", description="Update the memory document (personal or team) with curated long-term information",
factory=lambda deps: ( factory=lambda deps: (
create_save_shared_memory_tool( create_update_team_memory_tool(
search_space_id=deps["search_space_id"],
created_by_id=deps["user_id"],
db_session=deps["db_session"],
)
if deps["thread_visibility"] == ChatVisibility.SEARCH_SPACE
else create_save_memory_tool(
user_id=deps["user_id"],
search_space_id=deps["search_space_id"],
db_session=deps["db_session"],
)
),
requires=["user_id", "search_space_id", "db_session", "thread_visibility"],
),
ToolDefinition(
name="recall_memory",
description="Recall relevant memories (personal or team) for context",
factory=lambda deps: (
create_recall_shared_memory_tool(
search_space_id=deps["search_space_id"], search_space_id=deps["search_space_id"],
db_session=deps["db_session"], db_session=deps["db_session"],
) )
if deps["thread_visibility"] == ChatVisibility.SEARCH_SPACE if deps["thread_visibility"] == ChatVisibility.SEARCH_SPACE
else create_recall_memory_tool( else create_update_memory_tool(
user_id=deps["user_id"], user_id=deps["user_id"],
search_space_id=deps["search_space_id"],
db_session=deps["db_session"], db_session=deps["db_session"],
) )
), ),