From 4df9e107689225032aee5309bea07fc10dce32c0 Mon Sep 17 00:00:00 2001 From: Kevin Lin Date: Wed, 16 Apr 2025 13:43:46 -0700 Subject: [PATCH] feat: sync memory tools between the sleeptime-doc agent and sleeptime agent for chat (#1730) --- letta/personas/examples/sleeptime_doc_persona.txt | 2 ++ letta/server/server.py | 11 +++-------- 2 files changed, 5 insertions(+), 8 deletions(-) create mode 100644 letta/personas/examples/sleeptime_doc_persona.txt diff --git a/letta/personas/examples/sleeptime_doc_persona.txt b/letta/personas/examples/sleeptime_doc_persona.txt new file mode 100644 index 00000000..e1f26faa --- /dev/null +++ b/letta/personas/examples/sleeptime_doc_persona.txt @@ -0,0 +1,2 @@ +You are an expert document assistant. When given external data, I will take notes on them and generate memories that help me understand what is in the external data. +When given information about eg. chat logs, results, etc. I generate memories that contain higher level profiles of the user, finding patterns and making inferences based on the data. diff --git a/letta/server/server.py b/letta/server/server.py index a8e82bb9..274db94f 100644 --- a/letta/server/server.py +++ b/letta/server/server.py @@ -97,7 +97,7 @@ from letta.services.user_manager import UserManager from letta.settings import model_settings, settings, tool_settings from letta.sleeptime_agent import SleeptimeAgent from letta.tracing import log_event, trace_method -from letta.utils import get_friendly_error_msg, make_key +from letta.utils import get_friendly_error_msg, get_persona_text, make_key config = LettaConfig.load() logger = get_logger(__name__) @@ -1053,19 +1053,14 @@ class SyncServer(Server): memory_blocks=[ CreateBlock( label="persona", - value=( - "I am an expert document summarizer. " - "I manage the data source blocks such that they " - "contain everything that is important about " - "the corresponding files." - ), + value=get_persona_text("sleeptime_doc_persona"), ), ], llm_config=main_agent.llm_config, embedding_config=main_agent.embedding_config, project_id=main_agent.project_id, include_base_tools=False, - tools=["core_memory_insert", "rethink_memory", "finish_rethinking_memory", "view_core_memory_with_line_numbers"], + tools=constants.BASE_SLEEPTIME_TOOLS, ) return self.agent_manager.create_agent( agent_create=request,