ML_DL/MUJAKJUNG (무작정 시리즈)

[OpenAI] OpenAI API 발급 및 request

swwho 2024. 8. 21. 23:52
728x90
반응형

OpenAI Key

  • GPT, DALL-E 등 openai가 제공하는 모델들을 API 형태로 사용하기 위해 필요한 키
  • 개인에게 발급되기 때문에, 외부 공유 없이 보안에 신경써야 한다. (해킹을 통한 과금 방지)
  • 기존 user key가 아닌 project 별 관리를 위한 project key 사용을 권장하고 있다.
  • 처음 팝업에서 복사하지 못하거나 key를 잊어버릴 경우는 삭제하고 재발급해야 한다.

API Key 환경변수 등록

  • 하드코딩으로 API Key를 사용할 수 있으나, github나 코드 공유 시 문제가 될 수 있다.
  • 환경변수에 원하는 변수 이름, 변수 값에는 발급받은 OpenAI Key를 저장하고 등록한다.
  • os를 통해 저장한 변수를 가져올 수 있다.
import os
api_key = os.environ.get('등록한 변수 이름')

 


Billing

  • add payment method를 통해 결제 카드를 등록해야 한다. (해외 결제 가능 카드)
  • 카드 등록 후, 원하는 금액의 credit을 충전해야 API 사용이 가능하다. (credit이 없다면 429 에러 발생)

ChatGPT 모델 사용

  • 발급받은 키와 정해진 코드를 통해 chatgpt 온라인과 같은 결과를 받을 수 있다.
  • openai의 모델을 사용하기 위해서는 openai를 설치해야 한다.
  • from openai 구문이 필요하기 때문에, 파일 이름이 openai.py는 피해야 한다.
# pip install openai

from openai import OpenAI
client = OpenAI()

completion = client.chat.completions.create(
    model="gpt-4o-mini",
    messages=[
        {"role": "system", "content": "프롬프트"},
        {
            "role": "user",
            "content": "원하는 내용 입력"
        }
    ]
)

print(completion.choices[0].message.content)