Python에서 OpenAI의 GPT-4 API와 통신하는 간단한 예제 코드

OpenAI API 키 발급 (https://platform.openai.com/account/api-keys)
openai 라이브러리 설치:
bash


pip install openai
Python 예제 코드 (GPT-4 API 호출)
python


import openai

# OpenAI API 키 설정
openai.api_key = 'YOUR_API_KEY'

def chat_with_gpt4(prompt):
    try:
        response = openai.ChatCompletion.create(
            model="gpt-4",  # GPT-4 모델 지정
            messages=[
                {"role": "system", "content": "You are a helpful assistant."},
                {"role": "user", "content": prompt}
            ],
            max_tokens=500,
            temperature=0.7,
        )
        # 응답 텍스트 추출
        answer = response['choices'][0]['message']['content']
        return answer
    except Exception as e:
        return f"Error: {e}"

if __name__ == "__main__":
    user_input = "GPT-4와 통신하는 방법을 알려줘."
    result = chat_with_gpt4(user_input)
    print("GPT-4 응답:\n", result)
설명
model="gpt-4": GPT-4 모델 사용
messages: 대화 형식으로 역할(role)과 내용을 전달
temperature: 응답의 창의성 조절 (0~1 사이)
max_tokens: 최대 응답 길이 제한