test.py

#!/usr/bin/env python3
import openai
import sys

import core

MODEL = "gpt-4o-2024-08-06"
MODEL = "o1-preview"
MODEL = "chatgpt-4o-latest"

core.load_key()
client = openai.OpenAI()

prompt = " ".join(sys.argv[1:])

# System message can provide further control over tone and task
# There is a way how to send more advanced discussion too
messages = [
    # {"role": "system", "content": "You are a helpful assistant."},
    {"role": "user", "content": prompt},
    # {"role": "user", "content": "Knock knock."},
    # {"role": "assistant", "content": "Who's there?"},
    # {"role": "user", "content": "Orange."},
    # ... And model would proceed with "Orange who?"
]

response = client.chat.completions.create(
    model=MODEL,
    messages=messages,
)

print(response.model)
print(response.choices[0].message.content.strip())
print(response.usage)