pip install openai
import os
import openai
openai.api_key = os.getenv("OPENAI_API_KEY")
from openai import OpenAI
client = OpenAI()
model = "gpt-4o-mini"
model = "o3-mini"
completion = client.chat.completions.create(
model=model,
messages=[
{"role": "system", "content": "You are a helpful assistant."},
{
"role": "user",
"content": "Write a haiku about recursion in programming."
}
]
)
print(completion.choices[0].message)
print(completion.choices[0].message.content)
from openai import OpenAI
client = OpenAI()
prompt = """
Write a bash script that takes a matrix represented as a string with
format '[1,2],[3,4],[5,6]' and prints the transpose in the same format.
"""
# model = "o3-mini"
model = "o1-mini"
response = client.chat.completions.create(
model=model,
# reasoning_effort="medium", # o3-mini only
messages=[
{
"role": "user",
"content": prompt
}
]
)
print(response.choices[0].message.content)
See models: https://platform.openai.com/docs/models
See rate limits and usage tiers: https://platform.openai.com/docs/guides/rate-limits#usage-tiers
See spend: https://platform.openai.com/usage https://platform.openai.com/settings/organization/usage