From 2aabb15e97c5d4817bf7bfeb6dcfac2815eb108b Mon Sep 17 00:00:00 2001 From: cthomas Date: Mon, 8 Sep 2025 22:53:18 -0700 Subject: [PATCH] feat: add template id to create message (#4488) --- letta/agents/letta_agent_v2.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/letta/agents/letta_agent_v2.py b/letta/agents/letta_agent_v2.py index 8fd34a6d..96f8e873 100644 --- a/letta/agents/letta_agent_v2.py +++ b/letta/agents/letta_agent_v2.py @@ -536,6 +536,7 @@ class LettaAgentV2(BaseAgentV2): input_messages_to_persist, actor=self.actor, project_id=self.agent_state.project_id, + template_id=self.agent_state.template_id, ) elif step_progression <= StepProgression.LOGGED_TRACE: if self.stop_reason is None: @@ -835,7 +836,10 @@ class LettaAgentV2(BaseAgentV2): ) messages_to_persist = (initial_messages or []) + tool_call_messages persisted_messages = await self.message_manager.create_many_messages_async( - messages_to_persist, actor=self.actor, embedding_config=agent_state.embedding_config, project_id=agent_state.project_id + messages_to_persist, + actor=self.actor, + project_id=agent_state.project_id, + template_id=agent_state.template_id, ) return persisted_messages, continue_stepping, stop_reason @@ -947,7 +951,7 @@ class LettaAgentV2(BaseAgentV2): messages_to_persist = (initial_messages or []) + tool_call_messages persisted_messages = await self.message_manager.create_many_messages_async( - messages_to_persist, actor=self.actor, embedding_config=agent_state.embedding_config, project_id=agent_state.project_id + messages_to_persist, actor=self.actor, project_id=agent_state.project_id, template_id=agent_state.template_id ) if run_id: