Google Colab에서 OpenAI API Key 안전하게 저장하고 사용하는 방법

Google Colab에서 OpenAI API Key를 안전하게 저장하고 불러오는 방법을 소개합니다. 보안 비밀(Secrets) 메뉴를 활용하여 Key를 저장한 후, os.environ과 userdata.get_secret을 사용해 안전하게 호출하는 실전 코드 예제를 제공합니다.

 

 

 

Google Colab 사용시 각종 API Key 를 안전하게 호출하는 방법에 대해서 알아봅니다. 보안 비밀 메뉴(Secrets)에 OpenAI API Key를 저장한 후, 이를 코드에서 안전하게 호출하는 방법을 설명해드릴게요.

 

1. Google Colab의 보안 비밀 메뉴에 OpenAI API Key 저장

Google Colab에서는 Secrets 기능을 사용하여 다음과 같이 API Key를 보안 저장할 수 있습니다.

1) Google Colab 실행 후, 왼쪽 목차 아이콘 클릭
2) 키모양의 비밀(Secrets) 메뉴 클릭

3) 새 보안 비밀 추가 (+ ADD SECRET) 를 눌러서 다음과 같이 이름 및 값 입력
   - 이름 : OPENAI_API_KEY 와 같은 식으로 변수 처리
   - 값 : 실제 API 키 입력

4) 노트북 액세스 허용 여부 선택

 

 

 

2. 코드에서 OpenAI API Key 호출하는 방법

이제 OpenAI API Key를 안전하게 코드에서 불러와 사용할 수 있습니다.

1) os.environ 을 사용하여 호출하는 방법
Google Colab에서 저장한 Secret 값을 os.environ으로 불러올 수 있습니다. os.environ["openai_tts_env"]를 사용하여 Secret 값을 가져온 후, openai.api_key에 할당하면 API를 안전하게 호출할 수 있습니다.

import os
import openai

# Google Colab Secret에서 API Key 가져오기
openai.api_key = os.environ["openai_tts_env"]

# OpenAI API 호출 예제
response = openai.ChatCompletion.create(
    model="gpt-4",
    messages=[{"role": "user", "content": "Hello, how are you?"}]
)

print(response["choices"][0]["message"]["content"])

 

2) get_secret 함수를 사용하여 호출하는 방법 (Colab 전용)
Google Colab에서는 get_secret 함수를 사용하여 비밀 값을 불러올 수도 있습니다. userdata.get_secret("openai_tts_env")`를 사용하면, Secret 값을 간편하게 호출할 수 있습니다.

from google.colab import userdata
import openai

# Google Colab의 비밀값 가져오기
openai.api_key = userdata.get_secret("openai_tts_env")

# OpenAI API 호출 예제
response = openai.ChatCompletion.create(
    model="gpt-4",
    messages=[{"role": "user", "content": "Tell me a joke."}]
)

print(response["choices"][0]["message"]["content"])

 

결론적으로 Google Colab 전용 환경이라면 userdata.get_secret 이 적합하고, 로컬 또는 서버 등 다른 환경에서도 사용할 계획이라면 os.environ이 더 범용적이라고 할 수 있습니다.