File size: 2,783 Bytes
34967e4 |
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 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 |
import os
from functools import lru_cache
from typing import Optional
from pydantic_settings import BaseSettings
os.environ["WEAVE_CAPTURE_CODE"] = "false"
if os.getenv("HUGGINGFACE_DEMO"):
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
ANTHROPIC_API_KEY = os.getenv("ANTHROPIC_API_KEY")
REDIS_PASSWORD = ""
REDIS_USE_SSL = False
else:
from app.aws.secrets import get_secret
ENV = os.getenv("ENV", "LOCAL")
secrets = get_secret()
if ENV != "PROD":
os.environ["WANDB_API_KEY"] = secrets["WANDB_API_KEY"]
OPENAI_API_KEY = secrets["OPENAI_API_KEY"]
ANTHROPIC_API_KEY = secrets["ANTHROPIC_API_KEY"]
REDIS_PASSWORD = secrets["REDIS_PASSWORD"] if ENV == "PROD" else ""
REDIS_USE_SSL = True if ENV == "PROD" or ENV == "UAT" else False
os.environ["WANDB_BASE_URL"] = "https://api.wandb.ai"
class Settings(BaseSettings):
# Supported openai models
OPENAI_MODELS: list = [
"gpt-4o", # first model is the default of the vendor
"gpt-4o-2024-11-20",
"gpt-4o-mini",
"ft:gpt-4o-2024-08-06:dreem:selfridges-dresses-test:BVYmIA6e",
]
# Supported anthropic models
ANTHROPIC_MODELS: list = [
"claude-3-5-sonnet-latest" # first model is the default of the vendor
]
# Supprted AI Services
SUPPORTED_MODELS: list = OPENAI_MODELS + ANTHROPIC_MODELS
# API Keys
OPENAI_API_KEY: str
ANTHROPIC_API_KEY: str
DEFAULT_MAX_ATTEMPTS: int = 1
# AI Service Configuration
DEFAULT_MODEL: str = OPENAI_MODELS[0]
MAX_TOKENS: int = 2000
TEMPERATURE: float = 0.0
# CORS Configuration
CORS_ALLOW_ORIGINS: bool = True
# API Configuration
API_V1_PREFIX: str = "/api/v1"
PROJECT_NAME: str = "Dreem Attribution"
DEBUG: bool = False
# Rate Limiting
RATE_LIMIT_CALLS: int = 100
RATE_LIMIT_PERIOD: int = 60
# Cache Configuration
REDIS_PASSWORD: Optional[str] = REDIS_PASSWORD
REDIS_USE_SSL: Optional[bool] = REDIS_USE_SSL
# Logging
LOG_LEVEL: str = "INFO"
LOG_FORMAT: str = "json"
# API Keys
OPENAI_API_KEY: str = OPENAI_API_KEY
ANTHROPIC_API_KEY: str = ANTHROPIC_API_KEY
#
MAX_DOWNLOAD_RETRY: int = 10 # times
def validate_api_keys(self):
"""Validate that required API keys are present."""
if not self.OPENAI_API_KEY:
raise ValueError("OPENAI_API_KEY is required")
if not self.ANTHROPIC_API_KEY:
raise ValueError("ANTHROPIC_API_KEY is required")
# Create a cached instance of settings
@lru_cache
def get_settings() -> Settings:
"""
Create and cache a Settings instance.
Returns the same instance for subsequent calls.
"""
settings = Settings()
settings.validate_api_keys()
return settings
|