Python에서 GPT-4o mini API 사용하는 방법 정리(ChatGPT)

ChatGPT 4o mini API를 파이썬에서 사용하는 방법은 다음과 같습니다. 이 과정에서 openai 패키지를 사용하고, API 키를 설정한 뒤 GPT-4o mini 모델을 호출하는 예시입니다.

1. openai 라이브러리 설치

pip install openai

 

pip install openai==0.28

저는 동작중에 에러 발생하여 0.28 버전으로 설치하였습니다. 

pycharm 터미널로 명령어 입력

설치 완료된 모습

2. API 키 설정

import openai
from getpass import getpass

# API 키 입력
OPENAI_YOUR_KEY = "YOUR API KEY"
openai.api_key = "OPENAI_YOUR_KEY"

3. GPT-4o mini를 사용하는 API 호출 함수 작성

다음 코드는 gpt-4o-mini 모델을 사용하여 메시지를 보내고 응답을 받는 함수입니다.

def call_gpt4o_mini(prompt):
    response = openai.ChatCompletion.create(
        model="gpt-4o-mini",  # 모델 선택
        messages=[
            {"role": "user", "content": prompt}
        ],
        max_tokens=150  # 출력 제한
    )
    return response['choices'][0]['message']['content']

# 예시 호출
result = call_gpt4o_mini("Explain how AI works.")
print(result)

4. 예시 호출

# 예시 호출
user_prompt = """
안녕 너는 누구니?
"""
result = call_gpt4o_mini(user_prompt)
print(result)

 

답변이 나온다!!