From 9a69afb32b3da40ec4f656ca87eea1cd5f66cf54 Mon Sep 17 00:00:00 2001 From: Andy Li <55300002+cliandy@users.noreply.github.com> Date: Wed, 23 Apr 2025 21:48:34 -0700 Subject: [PATCH] fix: create response_format for creation (#1870) Co-authored-by: Sarah Wooders --- letta/client/client.py | 3 +++ letta/services/agent_manager.py | 1 + 2 files changed, 4 insertions(+) diff --git a/letta/client/client.py b/letta/client/client.py index 04572a6b..a9ba7b93 100644 --- a/letta/client/client.py +++ b/letta/client/client.py @@ -85,6 +85,7 @@ class AbstractClient(object): description: Optional[str] = None, tags: Optional[List[str]] = None, message_buffer_autoclear: bool = False, + response_format: Optional[ResponseFormatUnion] = None, ) -> AgentState: raise NotImplementedError @@ -2352,6 +2353,7 @@ class LocalClient(AbstractClient): initial_message_sequence: Optional[List[Message]] = None, tags: Optional[List[str]] = None, message_buffer_autoclear: bool = False, + response_format: Optional[ResponseFormatUnion] = None, ) -> AgentState: """Create an agent @@ -2405,6 +2407,7 @@ class LocalClient(AbstractClient): "initial_message_sequence": initial_message_sequence, "tags": tags, "message_buffer_autoclear": message_buffer_autoclear, + "response_format": response_format, } # Only add name if it's not None diff --git a/letta/services/agent_manager.py b/letta/services/agent_manager.py index ce228c6c..aa94dae6 100644 --- a/letta/services/agent_manager.py +++ b/letta/services/agent_manager.py @@ -236,6 +236,7 @@ class AgentManager: base_template_id=agent_create.base_template_id, message_buffer_autoclear=agent_create.message_buffer_autoclear, enable_sleeptime=agent_create.enable_sleeptime, + response_format=agent_create.response_format, created_by_id=actor.id, last_updated_by_id=actor.id, )