파이썬 환경에서 제미나이 (gemini) API 활용하기

반응형

 

제미나이 API Key 발급받기

Google AI Studio에 가입하고 API 키를 발급받는다.

 

Google AI Studio

Google AI Studio is the fastest way to start building with Gemini, our next generation family of multimodal generative AI models.

aistudio.google.com

발급받은 Gemini API 키를 안전하게 보관하고, 환경 변수로 설정한다. 

이때, .env 파일을 사용하여 API 키를 관리하는 것이 좋다.

 

 

파이썬 코드 작성

import google.generativeai as genai
import os
from dotenv import load_dotenv

load_dotenv()

GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
genai.configure(api_key=GOOGLE_API_KEY)
model = genai.GenerativeModel('gemini-pro')

def generate_content(prompt):
    response = model.generate_content(prompt)
    return response.text

prompt = f"""
제미나이에 요청할 프롬프트 내용
"""

content = generate_content(prompt)
print(content)

 

 

반응형