Skip to content

Commit

Permalink
chore(examples): fix static types in assistants example (#852)
Browse files Browse the repository at this point in the history
  • Loading branch information
stainless-bot committed Nov 20, 2023
1 parent f6c2aed commit 5b47b2c
Showing 1 changed file with 12 additions and 18 deletions.
30 changes: 12 additions & 18 deletions examples/assistant.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
import openai
import time

import openai

# gets API Key from environment variable OPENAI_API_KEY
client = openai.OpenAI()

Expand All @@ -16,38 +17,31 @@
message = client.beta.threads.messages.create(
thread_id=thread.id,
role="user",
content="I need to solve the equation `3x + 11 = 14`. Can you help me?"
content="I need to solve the equation `3x + 11 = 14`. Can you help me?",
)

run = client.beta.threads.runs.create(
thread_id=thread.id,
assistant_id=assistant.id,
instructions="Please address the user as Jane Doe. The user has a premium account."
thread_id=thread.id,
assistant_id=assistant.id,
instructions="Please address the user as Jane Doe. The user has a premium account.",
)

print("checking assistant status. ")
while True:
run = client.beta.threads.runs.retrieve(
thread_id=thread.id,
run_id=run.id
)
run = client.beta.threads.runs.retrieve(thread_id=thread.id, run_id=run.id)

if run.status == "completed":
print("done!")
messages = client.beta.threads.messages.list(
thread_id=thread.id
)
messages = client.beta.threads.messages.list(thread_id=thread.id)

print("messages: ")
for message in messages:
print({
"role": message.role,
"message": message.content[0].text.value
})
assert message.content[0].type == "text"
print({"role": message.role, "message": message.content[0].text.value})

client.beta.assistants.delete(assistant.id)

break
else:
print("in progress...")
time.sleep(5)
time.sleep(5)

0 comments on commit 5b47b2c

Please sign in to comment.