Openai
import os
import openai
openai.api_key = os.getenv("OPENAI_API_KEY")
Completion
model = "text-davinci-003"
response = openai.Completion.create(
model=model,
prompt="Make a list of astronomical observatories:",
)
Chat Completion
model = "gpt-3.5-turbo"
model = "gpt-4"
response = openai.ChatCompletion.create(
model=model,
messages=[
{"role": "system", "content": "You are a helpful assistant."},
{"role": "user", "content": "Who won the world series in 2020?"},
{"role": "assistant", "content": "The Los Angeles Dodgers won the World Series in 2020."},
{"role": "user", "content": "Where was it played?"}
]
)
response = openai.ChatCompletion.create(
model="gpt-4",
messages=[
{"role": "system", "content": "You are a helpful assistant."},
{"role": "user", "content": "Hello!"}
]
)
print(response.choices[0].message)