openAI API 사용하기(feat. google colab)

openAI API 활용 계기

인턴생활 중 첫 업무는 OPENAI API, 이 외의 GPT의 상업적 활용이 어느수준으로 가능한가를 확인하는것이었다.

이 외에도 주어진 업무가 있지만 기업에서 주도하는 프로젝트이니 만큼 자세히는 말하지 않는것이 맞다고 생각하여, openAI API를 연동하는 것에 대한 글만 작성하게 되었다.

 

우선 왜 COLAB인가? 왜 PYTHON인가?

openAI API 공식 문서에는 node.js, python 이 외에도 여러 언어로 설명이 되어있다.

이 중 python, colab을 고른것은 두가지 이유가 있다.

첫째로 같은 팀 매니저님께서 python을 추천하셨다. node.js를 활용해도 큰 어려움은 없겠지만 인공지능 개발에는 python이 가장 최적화되어있기 때문에 API연동도 python을 사용하는것이 좋겠다고 하셨다.

두번째로 나는 인공지능 관련 프로그램을 로컬에서 돌려본 경험이 없다. 매번 google colab에서 돌렸기 때문에 사실 매니저님께 질문을 드릴때도 python으로 하라고 하시길 바라며 질문을 드렸다.

 

---

openAI API 활용하기

openAI api를 colab에 연동하기 위해서는 3가지 단계가 필요하다.

1. colab 접근

2. openAI API key 할당

3. colab에서 openAI API연동

---

1. colab 접근

 

구글 드라이브 > 새로 만들기 > 더보기 > google colaboratory

 

구글 메인페이지에서 [더보기]아이콘을 선택한다.

 

[드라이브]를 선택한다.

 

좌측 상단 [새로 만들기]를 선택한다.

 

[더보기]를 선택 후 [google colaboratory]를 선택한다.

 

이런 화면까지 들어온다면 colab환경 설정은 모두 완료된 것이다.

---

2. openAI API key 할당

 

https://openai.com/blog/openai-api

OpenAI API

위 링크에 접근 후 회원가입 및 로그인 수행

 

[view API keys]를선택한다. 이 후 페이지에서 [create new secretKey]를 선택한다.

본인이 원하는 key name을 입력하면 된다. 입력 후 아래 화면과 같이 key가 생성되면 (2)도 모두 완료된 것이다.

좌측 카테고리탭 [setting]에 접근하면 organization ID확인이 가능하다.

 

3. colab에서 openAI API연동

!pip install openai

import os
import openai
openai.organization = "org-"
openai.api_key = "sk-" # API Key
openai.Model.list()

openAI 라이브러리를 인스톨한 후 임포트한다.

이 후 organization id, api key를 입력한다. 이때 id, key는 (2)에서 알아낸 본인의 정보를 입력해야한다.

 

아래와 같이 모델의 정보가 출력되면 성공이다.

모델의 정보가 출력되지 않고 runtime관련 에러가 난다면 본인의 openAI API 사용량을 확인해보아야한다.

chatGPT-4(유료버전)을 결제한 유저는 기본적으로 GPT-3 openAI API 사용량을 어느정도 제공하는것으로 확인하였다.

유료버전 유저가 아닐경우 별도로 openAI API 결제를 수행한 후에 사용이 가능하다.

 

openAI API 비용은 token(단어)기준으로 청구되는데 하루동안 여러 테스트를 수행하였지만 1달러를 넘어간 경우는 없다!

비용에 대한 부담은 가지지 않고 테스트해도 문제 없을듯하다.

 

openAI 공식문서를 보며 단계를 밟아가면 큰 어려움 없이 도전해볼 수 있다.

https://platform.openai.com/docs/guides/gpt/faq