TIL 2023.05.26

OpenAI API를 써보자

 

먼저 OpenAI에 회원가입을 하자.

https://openai.com

OpenAI

 

로그인 후 아래 페이지로 이동

https://platform.openai.com/

OpenAI API

 

자신의 프로필을 눌러 View API keys 에서

API key를 생성하자

key 는 잘 저장해놓자

 

openai 설치

pip install openai

 

임포트

import openai

openai.api_key = your_API_key

 

프롬프트를 담을 리스트 생성

gpt_prompt = []

 

역할 지정

gpt_prompt.append({
        "role": "system",
        "content": "Imagine the detail appearance of the input. Response shortly."
    })

 

내용 입력

gpt_prompt.append({
    "role": "user",
    "content": "귀여운 공룡"
    })

 

생성

prompt = openai.ChatCompletion.create(
            model="gpt-3.5-turbo",
            messages=gpt_prompt)

 

답변은

prompt = prompt["choices"][0]["message"]["content"]

이렇게 불러올 수 있다.