[Python] OpenAI API 활용해 ChatGPT 구현하기

python

by 거니 노트 2023. 12. 26. 15:16

OpenAI API를 사용하는 방법 중 특정 URL로 요청을 보내도 되지만 python에서는 openai 라는 라이브러리가 있어 이를 활용하기로 했습니다.

openai 라이브러리를 사용해서 ChatGPT에게 질문을 보내려면 API key가 필요하기 때문에 발급을 먼저 해주겠습니다.

https://platform.openai.com/

openai 공식 사이트에 들어가 로그인 후 API keys 탭으로 들어갑니다.

Create new secret key를 클릭하여 새로운 key를 발급해줍니다.

생성을 하게되면 key값이 나오게 되고 복사를 할 수 있는 버튼이 나오는데 그 시점에 복사를 하여 따로 저장을 해두셔야 합니다. Done을 누르게 되면 내 key값의 일부가 가려져 알 수가 없게 되기 때문에 따로 메모장에 저장하도록 합시다.

key를 생성하였으니 python으로 넘어가 key를 활용하여 ChatGPT를 구현해보겠습니다.

Version Info

python 3.9.17 / openai 1.6.1

라이브러리를 install 하기 전 python 가상환경을 먼저 받아줍니다.

# 가상황경 설치
python -m venv venv

# 가상환경 접속
source venv/bin/activate

# 가상환경 나가기
deactivate

가상황경에 접속 후 pip를 이용하여 openai를 install 합니다.

pip install openai

맨 처음 발급한 key를 my-key부분에 넣고 실행하면 content에 작성한 질문의 대한 대답을 반환하여 출력합니다.

import openai

openai_key = "my-key"

openai.api_key = openai_key

model = "gpt-3.5-turbo-1106"

message = [
    {
        "role": "system",
        "content": "안녕 너는 누구니?"
    }
]

tokens = 100

response = openai.chat.completions.create(
    model=model,
    messages=message,
    max_tokens=tokens
)

answer = response.choices[0].message.content

print(answer)

출력 예시