반응형
1. Gemini 패키지 설치
pip install google-genai2. Python 예제 코드
main.py에 아래 코드를 입력하세요.
from google import genai
# Gemini 홈페이지에서 Key를 받아서 key.txt 파일에 넣고 같은 폴더에 저장합니다.
with open('./key.txt', 'r', encoding='utf-8') as f:
key = f.read().rstrip()
print(key)
# Key값으로 Gemini Client를 생성합니다.
client = genai.Client(api_key=key)
response = client.models.generate_content(
model="gemini-2.0-flash", contents="Explain how AI works in a few words"
)
result = response.text.rstrip()
print(result)3. API KEY 설정하기
예제에서 Client() parameter로 key를 넘겨도 사용이 가능합니다.
하지만 매번 코드 작성때 마다 넣어 주어야 해서 이를 피하고자 한다면
아래 처럼 환경 변수에 설정해서 사용할 수 있습니다.
1. Shell Script 작성하기
예를 들어, key.txt라는 파일에 OpenAI 키가 한 줄에 저장되어 있다고 할 때 사용하실 수 있습니다.
#!/bin/bash
# key.txt 파일에서 첫 줄을 읽어 GEMINI_API_KEY로 export
export GEMINI_API_KEY=$(<key.txt)
# 확인 (선택)
echo "GEMINI_API_KEY has been set."2. 사용 방법
- 위 스크립트를 예를 들어 set_api_key.sh라는 파일로 저장합니다.
- key.txt 파일에 키를 저장해 둡니다. 예:
AIzaS...- 실행 전 권한 부여:
chmod +x set_api_key.sh- 실행 방법
set_api_key.sh를 일반적으로 실행하지 않고 항상 source 나 . 으로 실행해야 설정된 값이 현재 session에 남아 있습니다.
source ./set_api_key.sh
python main.py
반응형