From 187c5b1871d2a39b1beda0843ab2b92368eb288b Mon Sep 17 00:00:00 2001 From: benshuk Date: Mon, 20 Jan 2025 17:20:03 +0200 Subject: [PATCH 1/2] fix: :bug: create thread without messages plz --- ai21/clients/common/beta/assistant/threads.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) 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 From ba0d7d6549c5a7baa2f9db2e4f8ff9a29f4f0525 Mon Sep 17 00:00:00 2001 From: benshuk Date: Mon, 20 Jan 2025 17:24:42 +0200 Subject: [PATCH 2/2] docs: :memo: update assistant example --- examples/studio/assistant/assistant.py | 11 +++-------- examples/studio/assistant/async_assistant.py | 11 +++-------- 2 files changed, 6 insertions(+), 16 deletions(-) diff --git a/examples/studio/assistant/assistant.py b/examples/studio/assistant/assistant.py index a3ba465c..e0628f29 100644 --- a/examples/studio/assistant/assistant.py +++ b/examples/studio/assistant/assistant.py @@ -6,14 +6,9 @@ def main(): assistant = ai21_client.beta.assistants.create(name="My Assistant") - thread = ai21_client.beta.threads.create( - messages=[ - { - "role": "user", - "content": "Hello", - }, - ] - ) + thread = ai21_client.beta.threads.create() + + ai21_client.beta.threads.messages.create(thread_id=thread.id, role="user", content="Hello") run = ai21_client.beta.threads.runs.create_and_poll( thread_id=thread.id, diff --git a/examples/studio/assistant/async_assistant.py b/examples/studio/assistant/async_assistant.py index f0cf4b90..cc4d1ae7 100644 --- a/examples/studio/assistant/async_assistant.py +++ b/examples/studio/assistant/async_assistant.py @@ -8,14 +8,9 @@ async def main(): assistant = await ai21_client.beta.assistants.create(name="My Assistant") - thread = await ai21_client.beta.threads.create( - messages=[ - { - "role": "user", - "content": "Hello", - }, - ] - ) + thread = await ai21_client.beta.threads.create() + + await ai21_client.beta.threads.messages.create(thread_id=thread.id, role="user", content="Hello") run = await ai21_client.beta.threads.runs.create_and_poll( thread_id=thread.id,