[Colab+GPT]Colab을 이용하여 생성형 AI 환경 준비

Colab에서 OpenAI 모델 사용하기

이번 글에서는 Google Colab 환경에서 OpenAI의 GPT 모델을 사용하는 방법을 소개합니다.
OpenAI API 키 발급 방법은 이전 글에서 설명했으니, 이 글에서는 Google Drive에 발급받은 키를 안전하게 사용하는 방법과, 간단한 API 연결 테스트 코드를 중심으로 설명합니다. Key 발급을 참고하는 경우 다음 글에서 Key 발급 부분까지만 참고하고 로컬 환경에 대한 설명 부분을 현재 문서에서 Colab 기반으로 환경이 변경되었다고 생각하면 됩니다.

[ChatGPT API]ChatGPT API Key 생성 방법을 알아보자.

---

📌 준비물

• Google 계정 (Colab 사용용)
• 발급받은 OpenAI API 키
• Google Colab 환경

---

1. 🔐 API 키를 .env 파일로 안전하게 저장하기

API 키는 외부에 노출되면 남용될 수 있으므로 노출되지 않도록 관리가 필요합니다. 만약 깃허브 등을 사용한다면 노출의 위험이 매우 크기 때문에 그대로 올리면 일은 없어야 합니다.
따라서 .env 파일에 저장해두고 개발 코드에서는 .env 파일의 내용을 읽어 오는 방법을 이용하겠습니다. 

📁 로컬에서 .env 파일 만들기

colabd에서 .env 파일 생성하는 번거로운 과정 대신 로컬에서 텍스트 에디터(메모장 등)로 만들어서 아래처럼 작성해 주세요.

OPENAI_API_KEY=sk-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx

 

만약 메모장으로 만들었다면 저장할 때 반드시 .txt가 붙지 않도록 확인합니다. 

만들어진 .env를 구글 드라이브를 마운트해서 접근할 수 있도록 하겠습니다. 사용하는 위치는 각자의 상황에 맞도록 생각합니다.

---

2. 📂 Colab에서 .env 파일 읽기 – 구글 드라이브 마운트 방식

아래 코드를 Colab에 복사해서 실행하세요

from google.colab import drive 
drive.mount('/content/drive')

이제 구글 드라이브에서 .env 파일이 있는 경로를 확인하고, 해당 경로를 사용해 환경변수를 로드합니다.

예를 들어 .env 파일을 MyDrive/secret/ 폴더에 저장했다면 다음과 같이 코드 내에서 폴더 위치로 옮겨 가고 .env 파일을 읽을 수 있습니다. 

만약 dotenv가 설치되지 않았다면 아래 명령어로 설치할 수 있어요
!pip install python-dotenv

!pip install python-dotenv

 

다음 코드를 통해 .env 파일에 저장된 OPEN_API_KEY 값을 읽고 오며 확인을 위해 print() 함수를 통해 본인의 키값이 출력되는 것을 확인할 수 있습니다. 

from dotenv import load_dotenv
import os

project_path = "/content/drive/My Drive/Secret"
os.chdir(project_path)

load_dotenv(dotenv_path=".env")

api_key = os.getenv("OPENAI_API_KEY")
print("Loaded key:", api_key)

Loaded key: sk-proj-xxxxxx

---

3. 💬 OpenAI  GPT 모델에 연결 테스트 – chat 함수 사용

이제 API 키를 이용해서 GPT 모델에 요청을 보내보겠습니다. 간단한 질문으로 연결이 잘 되는지 확인하는 용도 입니다. 

✨ 설치 및 설정

먼저 openai 설치를 위해 다음 명령어를 실행합니다. 

!pip install openai 

💡 테스트 프롬프트 실행

모델은 가능한 어느 것이든 변경할 수 있으며 정상적으로 키를 통해 결과를 얻을 수 있는지를 확인합니다.

import os
import openai


# 환경변수에서 API 키 설정
client = openai.OpenAI(api_key=os.environ["OPENAI_API_KEY"])

prompt = "생성형 AI에 대해 설명 부탁해"

# ChatCompletion 호출 (원하는 모델 지정 가능("gpt-4o", "gpt-4-turbo", "gpt-3.5-turbo"))
response = client.chat.completions.create(
    model="gpt-4o",
    messages=[{"role":"user", "content":prompt}],
    max_tokens=150,    
)

# 응답 결과 출력
print("응답:", response.choices[0].message.content.strip())

---

실행 결과가 정상적으로 출력된다면 OpenAI 모델을 연결할 준비가 되었습니다. 여기까지는 Google Drive와 Colab을 이용하여 생성형 AI를 이용하는 가장 기본인 API를 통하여 모델을 연동하는 방법이 준비가 되었다고 생각하면 됩니다. 기타 다양한 응용을 할 수 있습니다.