인공지능, 데이터분석/[프로젝트] 간단한 실습 예제

[chatGPT API] chatGPT API 사용 실습

마법사 코딩공주 2023. 5. 2. 21:21
728x90
반응형

API keys 값 할당 받기

1. OpenAI 웹사이트 https://platform.openai.com

2. Sing up 버튼 선택 후 계정 생성, 계정 생성이 됐을 경우 로그인하여 자신의 계정을 선택 한 후, View API keys를 선택 -> 해당 페이지에서 API에서  keys값 받음

해당 keys값은 메모해둬야 함

 

chatGPT 실습하기

openai 라이브러리 설치

pip install openai
import openai

key1 = "api에서 받은 key값 복사하기"
openai.api_key = key1

실습 - 3개 이상 답변 받기

for i in range(3):
  text1 = input("my turn : ")
  msg = text1
  msg_input = [ ]
  msg_input.append( {"role":"user", "content":msg})
  response = openai.ChatCompletion.create(
         model="gpt-3.5-turbo",
         messages=msg_input
  )
  print(response.choices[0].message.content)

실습 대화 내용(my turn이 나의 질문)

실습 - 대화 내용 파일로 남기기

with open("test01.txt", "w") as f:
    for i in range(3):
        text1 = input("my turn : ")
        msg = text1
        msg_input = [ ]
        msg_input.append( {"role":"user", "content":msg})
        response = openai.ChatCompletion.create(
                model="gpt-3.5-turbo",
                messages=msg_input
        )
        print(response.choices[0].message.content)
        f.write(response.choices[0].message.content)

실습 - 함수화

msg_input = [ ]
def chatBot(msg):
  global msg_input
  msg_input.append( {"role":"user", "content":msg})
  
  response = openai.ChatCompletion.create(
       model="gpt-3.5-turbo",
       messages=msg_input
  )

  return response.choices[0].message.content
728x90
반응형