본문 바로가기
머신러닝/머신러닝 개발

파이썬을 이용한 ChatGPT API 호출

by CHML 2023. 12. 3.
1. Open AI API 키 발급

먼저 ChatGPT를 개발한 OpenAI에 접속하여 회원가입을 한다. OpenAI 홈페이지에서 [그림 1]과 같이 왼쪽 사이드바에서 API Keys 메뉴를 클릭한다. 그 다음 API keys 메뉴에서 Create new secret key 버튼을 클릭하여 API 키를 발급 받는다. 발급된 API 키는 유출되지 않도록 주의해야한다.

[그림 1] OpenAI API 키 발급.

 

2. 파이썬을 이용한 ChatGPT API 호출

OpenAI에서는 OpenAI에서 개발한 인공지능 모델을 쉽게 사용할 수 있도록 파이썬 OpenAI 패키지를 제공하고 있다. OpenAI 패키지는 아래의 pip 명령어를 이용하여 다운받을 수 있다.

pip install openai​

파이썬 OpenAI 패키지를 다운 받은 후에 아래와 같이 파이썬 코드를 작성한다. OpenAI 객체의 `api_key`에는 발급 받은 API 키를 문자열로 입력한다. 아래의 예제 코드에서는 "ChatGPT는 어디에 활용될 수 있나요?"를 ChatGPT에게 질문한다.

from openai import OpenAI


client = OpenAI(api_key='발급 받은 API 키')
query = 'ChatGPT는 어디에 활용될 수 있나요?'
message = [{'role': 'user', 'content': query}]
completion = client.chat.completions.create(model='gpt-3.5-turbo', messages=message)
response_text = completion.choices[0].message.content

print(response_text)

우리가 질문한 내용에 대한 ChatGPT의 답변은 `response_text`라는 변수에 문자열의 형태로 저장되며, 파이썬 코드에서는 `print(response_text)`를 통해 답변의 내용을 확인할 수 있다.