파이썬을 통한 챗GPT API 연동 간단하게 하는 방법

오늘은 핫한 챗GPT 를 파이썬을 통해 연동하는 방법에 대해 알아보려고 합니다.

물론 챗GPT에서 제공하는 무료 API를 사용할 수 있습니다!

 

다만, 실제 챗GPT에서 제공하는 수준보다는 조금 떨어져서 간단하게 연동하는 부분들에 대해서만 사용하는 걸 추천드립니다.

 

---

 

1. openai 설치

먼저 chatGPT에서 제공하는 오픈소스 openai 를 설치하여 import가 필요합니다.

 

pip install openai
import openai

 

---

2. chatGPT API 연동

먼저 chatGPT에 계정을 가입하셔야 합니다.

가입한 계정으로 로그인 한 후 세팅 > View API 를 눌러 신규로 API 키를 생성해주시면 됩니다.

혹시 못찾으시는 분들을 위해 로그인을 한 후 해당 URL로 접근하시면 아래 사진처럼 API 키 생성이 가능하니 도움되시길 바랍니다.

 

chatGPT API 키 생성 바로가기

---

chatGPT API keys 생성하기

 

해당 사이트에서 생성한 API 키값을 openai 모듈을 사용하여 연결만 해주시면 됩니다.

 

openai.api_key = "생성한 API Key값"

 

---

3. 챗gtp 모델 설정

다음으로는 한국어 전용 chatGPT 엔진 모델을 설정해주고 각 설정값을 통해 수준을 조절할 수 있습니다.

 

response = openai.Completion.create(
engine="text-davinci-003", # KoGPT-3 모델 선택
prompt=prompt,
temperature=0.7, # 문장의 창의성 정도를 조절
max_tokens=600, # 생성될 문장의 길이 제한
n=1, # 생성할 문장의 개수
stop=None
)

 

저는 text-davinci-003 모델을 선택하였고 창의성과 문장 길이 제한을 위 처럼 설정하여 간단하게 적용시켜 보았습니다.

prompt 를 통해 질문을 받고 해당 엔진을 통해 답변을 받는 구조입니다.

 

---

4. 답변 처리

우리가 질문을 하면 엔진에서 대답을 하고 해당 반응을 print 문으로 출력하여 결과값을 출력하는 샘플 코드를 작성했습니다. 

샘플 코드는 아래와 같습니다.

 

import openai

openai.api_key = "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"

prompt = "안녕 GPT야?"
response = openai.Completion.create(
engine="text-davinci-003", # KoGPT-3 모델 선택
prompt=prompt,
temperature=0.7, # 문장의 창의성 정도를 조절
max_tokens=10, # 생성될 문장의 길이 제한
n=3, # 생성할 문장의 개수
stop=None
)

answer = response.choices[0].text.strip()
print(answer)

 

시간 가는줄 모르고 계속 질문하며 chatGPT API랑 놀아보는 시간이었네요.

파이썬을 통해 간단하게 chatGPT API 를 연동해보는 방법에 대해 알아보았습니다.

궁금하신 부분이 있다면 댓글을 통해 알려주시길 바랍니다!

감사합니다.