docs: 📝 update assistant example

AI21Labs · Jan 20, 2025 · ba0d7d6 · ba0d7d6
1 parent 187c5b1
commit ba0d7d6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 16 deletions.
diff --git 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
@@ -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,