OpenAI API로 시작하는 o3 모델 Python 퀵스타트

%env OPENAI_API_KEY=KEY

!pip install openai weave

import os
import weave
from openai import OpenAI
import re

# Initialize Weave and W&B
weave.init('o3-mini')

# Initialize OpenAI client
client = OpenAI(api_key=os.getenv('OPENAI_API_KEY'))

o3_assistant_prompt = "You are a " + input("Who should I be, as I answer your prompt?")
o3_user_prompt = input("What prompt do you want me to complete?")
o3_prompt = o3_assistant_prompt, o3_user_prompt
print(o3_prompt)

@weave.op()
def generate_content(o3_assistant_prompt: str, o3_user_prompt: str) -> str:
    prompt = f"{o3_assistant_prompt}\nUser request: {o3_user_prompt}"
    messages = [
        {"role": "assistant", "content": o3_assistant_prompt},
        {"role": "user", "content": o3_user_prompt}
    ]
    response = client.chat.completions.create(
        model="o3-mini",
        messages=messages,
        reasoning_effort="high"  # This can be set to low, medium or high.
    )
    return response.choices[0].message.content

# Generate content using the provided assistant and user prompts
content = generate_content(o3_assistant_prompt, o3_user_prompt)

# Print the generated response
print(content)