File size: 857 Bytes
4b022af |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
import os
from dotenv import load_dotenv
# Load environment variables from .env file (if it exists)
load_dotenv()
# Get the Gemini API key from environment variables
GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
# Settings object for easy access
class Settings:
GEMINI_API_KEY: str = GEMINI_API_KEY
LOG_LEVEL: str = os.getenv("LOG_LEVEL", "INFO")
MAX_FILE_SIZE: int = int(os.getenv("MAX_FILE_SIZE", "10485760")) # 10MB default
def validate(self):
"""Validate required settings"""
if not self.GEMINI_API_KEY:
raise ValueError(
"GEMINI_API_KEY not found in environment variables. "
"Please set it in your environment or .env file."
)
settings = Settings()
# Validate settings on import (only if GEMINI_API_KEY is set)
if GEMINI_API_KEY:
settings.validate()
|