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

728x90

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

1. `openai` 라이브러리 설치

pip install openai

pip install openai==0.28

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

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)

답변이 나온다!!

728x90

저작자표시

'집중 - 어제보다 나아가자 > [개발] 딥러닝 🍩' 카테고리의 다른 글

ChatGPT API 사용을 위한 카드 등록 방법 설명 (0)	2024.10.25
ChatGPT API 발급 받는 방법 (python에서 사용) (0)	2024.10.25
libtorch serialize (RT Convert) 오류 1 (0)	2024.09.20
libtorch classification inference , torch::jit::load fail (2)	2024.09.13