From 4d9725750d64e4165e6fcf424fce1e560e2fbc33 Mon Sep 17 00:00:00 2001 From: cthomas Date: Tue, 8 Apr 2025 10:46:32 -0700 Subject: [PATCH] feat: integrate replace flag in core_memory_insert (#1612) Co-authored-by: Kevin Lin Co-authored-by: Kevin Lin --- letta/functions/function_sets/base.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/letta/functions/function_sets/base.py b/letta/functions/function_sets/base.py index 02c507d9..44163deb 100644 --- a/letta/functions/function_sets/base.py +++ b/letta/functions/function_sets/base.py @@ -225,7 +225,10 @@ def core_memory_insert(agent_state: "AgentState", target_block_label: str, new_m current_value_list = current_value.split("\n") if line_number is None: line_number = len(current_value_list) - current_value_list.insert(line_number, new_memory) + if replace: + current_value_list[line_number] = new_memory + else: + current_value_list.insert(line_number, new_memory) new_value = "\n".join(current_value_list) agent_state.memory.update_block_value(label=target_block_label, value=new_value) return None