openai-python/examples/assistant_stream.py at main · djun/openai-python · GitHub

Repository navigation

import openai

# gets API Key from environment variable OPENAI_API_KEY

client = openai.OpenAI()

assistant = client.beta.assistants.create(

name="Math Tutor",

instructions="You are a personal math tutor. Write and run code to answer math questions.",

tools=[{"type": "code_interpreter"}],

model="gpt-4-1106-preview",

)

thread = client.beta.threads.create()

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?",

)

print("starting run stream")

stream = 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.",

stream=True,

)

for event in stream:

print(event.model_dump_json(indent=2, exclude_unset=True))

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