feat: integrate replace flag in core_memory_insert (#1612)

Co-authored-by: Kevin Lin <klin5061@gmail.com>
Co-authored-by: Kevin Lin <kl2806@columbia.edu>
This commit is contained in:
cthomas
2025-04-08 10:46:32 -07:00
committed by GitHub
parent 88cb293676
commit d3d7288ec8

View File

@@ -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