fix: create response_format for creation (#1870)

Co-authored-by: Sarah Wooders <sarahwooders@gmail.com>
This commit is contained in:
Andy Li
2025-04-23 21:48:34 -07:00
committed by GitHub
parent d48fcde351
commit 9a69afb32b
2 changed files with 4 additions and 0 deletions

View File

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

View File

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