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