Examples for how to use the ChatGPT API from python

chat_gpt_api_example.py

import os
from openai import OpenAI

# 1. Must create an OpenAI account
# 2. Must generate a personalized API key at https://platform.openai.com/api-keys
# 3. Never save API keys in code, instead save as environment variable with `export OPEN_API_KEY=your_key`


if os.environ['OPENAI_API_KEY'] is None:
    raise RuntimeError ("Must set environment variable \"OPENAI_API_KEY\"")

client = OpenAI(api_key=os.environ['OPENAI_API_KEY'])


completion = client.chat.completions.create(
  model="gpt-3.5-turbo",
  messages=[
    {"role": "system", "content": "You are an assistant that completes sentences."},
    {"role": "user", "content": "The weather will"},
    {"role": "assistant", "content": "be variable in New England."},
    {"role": "user", "content": "The sky will"}
    ]
)


print(completion.choices[0].message)

print(completion.usage)