[Python] OpenAI API 사용해 ChatGPT 사용해보기

 

API Key가 세팅되었다면 API 연결을 해보겠습니다.

Python 3.6 이상이 설치되어있는지 확인해주세요.
설치되지 않았다면 설치하고 시작합니다.

 

1. OpenAi 설치

아래 명령어를 통해 openai 모듈을 설치합니다.
이 모듈을 통해 파이썬에서 api 연결이 가능합니다.

pip install openai

2. OpenAi 모듈 설치

이전 글에서 python-dotenv를 통해 API키를 세팅해뒀죠.

import os
from dotenv import load_dotenv

# load .env
load_dotenv()

#api_key test
api_key = os.environ.get("OPENAI_API_KEY")
print(api_key)

 

여기서 조금 더 수정해 API 호출을 해보겠습니다.

import os
from dotenv import load_dotenv
import openai


# load .env
load_dotenv()

api_key = os.environ.get("OPENAI_API_KEY")
openai.api_key = os.environ["OPENAI_API_KEY"]

# example
response = openai.chat.completions.create(
    model="gpt-3.5-turbo",
    messages=[
        {
            "role": "user",
            "content": "대한민국의 위도 경도를 알려줘.",
        },
    ],
    temperature=0.3,
    max_tokens=256,
)

# output
print(response);

# only content
output_text = print(response.choices[0].message.content)
print(output_text)

 

예제 코드입니다.

코드를 실행해봅시다.

 

response가 보일텐데요.

reponse 중 content만 뽑아내 보았습니다.

대한민국의 위도는 약 36도 00분 ~ 38도 15분, 경도는 약 125도 41분 ~ 129도 17분에 위치하고 있습니다.

 

정상적으로 답변이 오면 성공입니다.

3.Openai 오류

이 글은 최신 openai를 사용하는 글입니다.
혹시 pip install openai==0.28 을 요구하는 문제가 생긴다면 openai의 업그레이드가 필요합니다.