챗 GPT API 사용해보기

1. openai 패키지 설치하기

!pip install openai

2. 코랩의 환경변수에 API 키 등록

import os
from getpass import getpass

os.environ['OPENAI_API_KEY'] = getpass()

OS는 운영체제 인터페이스를 다룰 수 있는 모듈이다. OS에는 환경 변수를 불러오는 os.environ 코드가 있고 이는 dict 타입의 구조처럼 key와 value 쌍으로 이루어져 있으므로 dict 처럼 key와 value 값을 지정하고 삭제할 수 있다.

getpass는 입력한 값을 감추는 모듈이다. getpass() 함수는 실행 시 input() 함수처럼 사용자의 입력을 받을 때까지 기다렸다가 사용자의 입력이 완료되면 그 값을 변수에 대입하는 기능을 가졌다. 하지만 input()과 달리 작성 중인 입력값을 볼 수 없도록 감추며 출력창에도 표기되지 않는다.

위 코드를 실행 후 등록된 API 키를 확인하고 싶다면 아래 코드를 실행 후 코드를 삭제한다.

os.environ['OPENAI_API_KEY'] or os.getenv('OPENAI_API_KEY')

3. API 키 설정하기

openai 모듈을 사용하기 위해서 다음과 같이 client 객체를 생성한다.

from openai import OpenAI

client = OpenAI(api_key=os.getenv('OPENAI_API_KEY'))

4. API 호출하기

다음은 챗GPT의 API를 호출하는 가장 기본적인 형태를 가진 코드이다.

completion = client.chat.completions.create(
    model="gpt-3.5-turbo",
    messages=[
        {"role": "system", "content": "당신은 헤이마트의 상담원입니다."},
        {"role": "user", "content": "헤이마트에 대해서 알려주세요."}
    ]
)

API를 호출하기 위해 client.chat.completions.create 함수를 작성해야 한다. create 함수의 매개변수로는  model에는 gpt-3.5-turbo, messages에는 리스트와 딕셔너리 형태로 값을 작성해야 한다. message에 값을 할당할 때는 반드시 딕셔너리를 담은 리스트를 작성해야 한다. 또한 딕셔너리를 작성할 때는 다음과 같이 "role"과 "content"로 이루어진 2개의 키가 포함되어 있어야 한다.

{"role" : "...", "content" : "..."}

"role"에 들어갈 수 있는 문자열은 "system", "user", "assistant", "tool" 네가지로 제한되어 있다. "content"에는 "role"에 들어간 규칙에 맞는 지침 또는 메시지를 작성해 주면 된다.

5. 출력값 확인

completion.choices[0].message

출력:

ChatCompletionMessage(content='헤이마트는 전자상거래 기업으로, 온라인에서 다양한 제품을 판매하는 회사입니다. 고객들은 헤이마트 웹사이트나 앱을 통해 여러 종류의 상품을 구매할 수 있습니다. 헤이마트는 의류, 가전제품, 가구, 식품 등 다양한 카테고리의 제품을 판매하고 있으며, 고객들에게 편리한 쇼핑 경험을 제공하기 위해 노력하고 있습니다. 또한 헤이마트는 고객 서비스와 배송 시스템을 지속적으로 개선하여 더 나은 서비스를 제공하고 있습니다.언제든지 더 궁금한 점이 있으시면 말씀해주세요.', refusal=None, role='assistant', audio=None, function_call=None, tool_calls=None, annotations=[])

completion.choices[0].message.content

출력:

헤이마트는 전자상거래 기업으로, 온라인에서 다양한 제품을 판매하는 회사입니다. 고객들은 헤이마트 웹사이트나 앱을 통해 여러 종류의 상품을 구매할 수 있습니다. 헤이마트는 의류, 가전제품, 가구, 식품 등 다양한 카테고리의 제품을 판매하고 있으며, 고객들에게 편리한 쇼핑 경험을 제공하기 위해 노력하고 있습니다. 또한 헤이마트는 고객 서비스와 배송 시스템을 지속적으로 개선하여 더 나은 서비스를 제공하고 있습니다.언제든지 더 궁금한 점이 있으시면 말씀해주세요.

 

[출처] Hey, 파이썬! 생성형 AI 활용 앱 만들어 줘

Hey, 파이썬! 생성형 AI 활용 앱 만들어 줘 - 예스24