Python

파이썬에서 ChatGPT API 사용하기

0

파이썬 ChatGPT API

최근에 가장 핫한 대화형 언어 모델인 ChatGPT를 사용하기 위해 OPENAI의 API를 사용하여 파이썬으로 구현해보겠습니다. 파이썬 답게 openai라는 파이썬 모듈이 제공되고 있어 해당 모듈을 설치한 후 OpenAI의 웹사이트에서 Api Key를 발급받아 연동만 해주면 되기 때문에 구현 자체는 매우 간단합니다.

platform.openai.com

OpenAI의 웹사이트에서 Api Key를 발급받았다면 ChatGPT의 텍스트 생성 모델인 text-davinci-003를 사용하여 주제와 세부요구사항에 맞게 자동으로 글을 생성하도록 구현할 수 있습니다.

API KEY 발급받기

OpenAI의 API Key는 다음과 같이 웹사이트에서 간단한 절차를 거쳐 발급받을 수 있습니다.

  • platform.openai.com에 접속하여 회원가입 후 로그인
  • https://platform.openai.com/account/api-keys 페이지에서 create new key 버튼을 눌러 Api Key 발급
  • 생성된 Api Key 복사
create new key 버튼을 눌러 api key 발급
create new key 버튼을 눌러 api key 발급platform.openai.com

모듈 설치 및 ChatGPT 사용하기

Api Key를 발급받았다면 프롬프트에서 다음의 명령어로 필요한 모듈을 설치합니다.

openai 모듈 설치하기
pip install openai

설치한 모듈을 임포트하고 다음과 같이 스크립트를 작성하여 ChatGPT를 사용할 수 있습니다.

ChatGPT 사용하기
import openai

# OPENAI API 키 입력
openai.api_key = "api_key"

def gpt_text(gpt_prompt):
  response = openai.Completion.create(
    engine="text-davinci-003",
    prompt=gpt_prompt,
    temperature=0.5,
    max_tokens=256,
    top_p=1.0,
    frequency_penalty=0.0,
    presence_penalty=0.0
  )
  return response['choices'][0]['text']

temperature는 무작위성을 제어하는데 이 값을 낮추면 무작위성이 줄어들고, max_tokens는 응답을 생성 할 최대 토큰 수를 설정합니다. 이 외에 더 자세한 설명은 OpenAI의 API 참고 문서를 통해 확인할 수 있습니다.

참고로 한글은 지원되지 않고, 물음표 등의 기호도 넣으면 안된다고 합니다.

print(gpt_text('Will artificial intelligence create a new gold rush era'))

여기서는 “text-davinci-003” 엔진을 사용했는데, GPT-3 문서에서 다양한 GPT-3엔진에 대하여 확인할 수 있습니다.

답글 남기기