OpenAI

# OpenAI API 사용 예제
# pip install openai

from openai import OpenAI

client = OpenAI()  # OPENAI_API_KEY 환경변수 사용

# ===== 1. 기본 채팅 완성 =====
response = client.chat.completions.create(
    model="gpt-4o",
    messages=[
        {"role": "system", "content": "당신은 유능한 AI 어시스턴트입니다."},
        {"role": "user", "content": "파이썬 데코레이터를 설명해주세요."}
    ],
    max_tokens=1000,
    temperature=0.7
)
print(response.choices[0].message.content)

# ===== 2. 스트리밍 응답 =====
stream = client.chat.completions.create(
    model="gpt-4o-mini",
    messages=[{"role": "user", "content": "인공지능의 미래는?"}],
    stream=True
)
for chunk in stream:
    if chunk.choices[0].delta.content:
        print(chunk.choices[0].delta.content, end="", flush=True)

# ===== 3. 이미지 분석 (Vision) =====
import base64

def encode_image(image_path):
    with open(image_path, "rb") as f:
        return base64.b64encode(f.read()).decode("utf-8")

response = client.chat.completions.create(
    model="gpt-4o",
    messages=[
        {
            "role": "user",
            "content": [
                {"type": "text", "text": "이 이미지를 설명해주세요."},
                {"type": "image_url", "image_url": {
                    "url": f"data:image/jpeg;base64,{encode_image('image.jpg')}"
                }}
            ]
        }
    ]
)

# ===== 4. Function Calling =====
tools = [{
    "type": "function",
    "function": {
        "name": "get_weather",
        "description": "특정 위치의 현재 날씨를 가져옵니다",
        "parameters": {
            "type": "object",
            "properties": {
                "location": {"type": "string", "description": "도시 이름"}
            },
            "required": ["location"]
        }
    }
}]

response = client.chat.completions.create(
    model="gpt-4o",
    messages=[{"role": "user", "content": "서울 날씨 어때?"}],
    tools=tools,
    tool_choice="auto"
)

# ===== 5. Embedding 생성 =====
response = client.embeddings.create(
    model="text-embedding-3-small",
    input="검색할 텍스트입니다"
)
embedding = response.data[0].embedding
print(f"임베딩 차원: {len(embedding)}")  # 1536