Openai의 gpt api를 이용한 간단한 챗봇 구현

요즘 OpenAI가 개발한 GPT의 3세대 모델이 굉장히 핫하다. 

gpt3를 이용하여 간단한 챗봇을 구현해볼 것이다.

 

API Key를 등록하기 위하여 OpenAi의 사이트로 이동을 한다

https://openai.com/product

Product

 

 

Get Started를 선택한다.

 

 

간단한 챗봇 기능을 구현할 것이기 때문에 아래에서 Chat을 선택한다

 

 

 

API키를 발급받기 위하여 오른쪽 상단에서 View API keys를 선택한다.

 

 

 

새로운 시크릿키를 생성한다.

 

 

이제 openai에서 제공하는 text-davinci-003 모델을 이용하여 자연스러운 챗봇을 구현할 차례이다.

vs code를 이용하였다.

 

코드는 정말 간단하다.

• openai.api_key : 위에서 발급받은 api key를 넣어준다.
• model : 내가 사용할 모델명을 작성
• prompt : 유저의 대화 데이터
• max_tokens : gpt의 대화 토큰 수를 지정해준다.

 

openai.api_key = OPENAI_SECRET_KEY
response = openai.Completion.create(
model="text-davinci-003",
prompt=question,
temperature=0.9,
max_tokens=500,
top_p=1,
frequency_penalty=0,
presence_penalty=0.6
)

generated_text = response.choices[0].text
generated_text = generated_text.lstrip("\n")
print(generated_text)

 

위에서 작성한 코드로 결과값을 확인해보았다.

 

 

결과값은 정말 신기하였다. 

gpt 모델은 유저가 작성한 문장에 대한 자연스러운 대답을 나타내는 것을 확인할 수 있었다.