diff --git a/ai21/clients/common/beta/assistant/threads.py b/ai21/clients/common/beta/assistant/threads.py index 68cc9460..883b75c0 100644 --- a/ai21/clients/common/beta/assistant/threads.py +++ b/ai21/clients/common/beta/assistant/threads.py @@ -25,8 +25,7 @@ def create( def _create_body(self, messages: List[Message] | NotGiven, **kwargs) -> Optional[dict]: body = remove_not_given({"messages": messages, **kwargs}) - if "messages" in body: - body["messages"] = [modify_message_content(message) for message in body["messages"]] + body["messages"] = [modify_message_content(message) for message in body.get("messages", [])] return body