[chatGPT] chatGPT API 사용하기 🫧

캡스톤 중에 챗봇 API를 사용할 일이 생겨서 사용해보았다!
 
아래 사이트에서 회원가입을 해준다

OpenAI API

 
오른쪽 상단 프로필에서 View API keys를 click !

 
create new secret key !

이때 키는 발급받고 처음이자 마지막으로 확인 가능하기 때문에 어딘가에 잘 저장해두어야 한다
 
Usage에서 사용량 확인 가능합니다아 ~ 🍡

 
그리고 코드 작성 !
매우 간단해요 ^ㅁ^
 
아래는 전체 코드

import openai

API_KEY = '발급받은 key 입력'

def callChatGPT(prompt, API_KEY=API_KEY):
    
    messages = []

    #get api key
    openai.api_key = API_KEY

    messages.append({"role":"user", "content":prompt})

    completion = openai.ChatCompletion.create(
        model="gpt-3.5-turbo",
        messages=messages
    )
    chat_response = completion.choices[0].message.content
    messages.append({"role":"assitant", "content":chat_response})

    return messages[1]["content"]

prompt = input("Inser a prompt: ")
print(callChatGPT(prompt))

 

먼저 모듈 임포트 !

import openai

그리고 openai를 사용하기 위해 설치가 필요합니다

pip install openai

 
다음으로는 prompt 입력받고!
callChatGPT 함수를 부르고 출력!
 

정말 간단하게 chatGPT API를 사용하는 방법! 이었습니다..~

---