Python

[ Python / chatGPT ] chatGPT open API를 사용해보자

menus 2024. 2. 29. 17:52
728x90

 

 

안녕하세요 !   (●'◡'●)

 

이번 글에서는 챗봇을 만드는 간단한 사이드 프로젝트를 진행하면서

 

chatGPT API를 사용할 기회가 생겨서 간단히 정리해보려고 합니다 !

 

 


 

 

 

https://platform.openai.com/docs/overview

 

 

위 링크에 나와있는 공식 페이지 튜토리얼을 참고하여 진행했습니다.

 

저는 flask로 진행할거여서 python으로 작업을 했습니다 !

 

작업 폴더를 생성해주고 api 폴더를 만들어주었습니다.

 

api 폴더로 이동한 뒤 가상 환경을 생성하기 위해 콘솔에 아래와 같이 입력했습니다.

 

python -m venv openai-env

 

 

이제 API 키를 설정해주겠습니다.

 

OpenAI Key를 아직 발급받지 않았다면 위에 첨부된 링크에서 발급받으실 수 있습니다 !

 

 

API keys에서 Create new secret key를 누르면 키를 발급받을 수 있습니다.

 

키는 처음에만 확인할 수 있고 이후에는 확인하지 못하니 잘 보관하시는게 좋습니다.

 

이제 발급받은 키로 API를 요청해보도록 하겠습니다.

 

이때, git에 키가 올라가면 안되니까 .env파일을 만들어서 키를 관리해주겠습니다.

 

.env파일을 만들어 주신 후

 

API_KEY="발급받은 키"

 

 

이렇게 입력해주시면 됩니다.

 

.gitignore에서 커밋을 할때 .env파일이 올라가지 않도록 설정해주면 되겠죠?

 

이제 chatGPT 공식문서에 나와있는 것처럼 gpt와 대화하는 형식으로 content에 넣어서 파일을 실행시키면 답변을 확인할 수 있습니다 !

 

import os
from openai import OpenAI
from dotenv import load_dotenv
 
load_dotenv()
api_key = os.getenv("API_KEY")

client = OpenAI(
    api_key=api_key)
completion = client.chat.completions.create(
    model="gpt-3.5-turbo",
    messages=[
        {"role": "user", "content": "인사해봐"}
    ],
    temperature=0.8,
)
print("챗봇 : "+completion.choices[0].message.content)

 

api key를 .env 파일에서 따로 관리하고 있기 때문에 

 

python_dotenv 패키지를 설치해주었습니다.

 

pip install python_dotenv

 

 

completion.choices[0].message만 출력했을 때는 content이외에 role, function_call 등이 함께 출력되어서 챗봇과 대화하는 형식으로 받아오기 위해 content만 가져오도록 했습니다.

 

오늘은 이렇게 chatGPT api를 활용하는 방법을 알아보았습니다 !

 

다음에도 유용한 내용으로 찾아오도록 하겠습니다.

 

감사합니다.