Spaces:
Sleeping
Sleeping
File size: 1,915 Bytes
d0d2f42 | 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 | """Centralized environment-based configuration."""
from dataclasses import dataclass
from functools import lru_cache
import os
from dotenv import load_dotenv
load_dotenv()
@dataclass(frozen=True)
class Settings:
llm_provider: str
llm_model: str
llm_temperature: float
gemini_api_key: str
log_level: str
input_cost_per_1m_tokens: float
output_cost_per_1m_tokens: float
def _read_str(name: str, default: str) -> str:
return os.getenv(name, default).strip()
def _read_float(name: str, default: str) -> float:
raw_value = _read_str(name, default)
try:
return float(raw_value)
except ValueError as exc:
raise ValueError(
f"Invalid value for {name!r}: {raw_value!r}. Expected a float."
) from exc
@lru_cache(maxsize=1)
def get_settings() -> Settings:
"""Read .env values, apply defaults, and fail for missing critical config."""
llm_provider = _read_str("LLM_PROVIDER", "gemini").lower()
llm_model = _read_str(
"LLM_MODEL",
_read_str("GEMINI_MODEL", "gemini-3-flash-preview"),
)
llm_temperature = _read_float("LLM_TEMPERATURE", "0.2")
gemini_api_key = _read_str("GEMINI_API_KEY", "")
log_level = _read_str("LOG_LEVEL", "INFO").upper()
input_cost_per_1m_tokens = _read_float("INPUT_COST_PER_1M_TOKENS", "0.0")
output_cost_per_1m_tokens = _read_float("OUTPUT_COST_PER_1M_TOKENS", "0.0")
if llm_provider == "gemini" and not gemini_api_key:
raise ValueError(
"Missing GEMINI_API_KEY. Set it in your environment or .env file."
)
return Settings(
llm_provider=llm_provider,
llm_model=llm_model,
llm_temperature=llm_temperature,
gemini_api_key=gemini_api_key,
log_level=log_level,
input_cost_per_1m_tokens=input_cost_per_1m_tokens,
output_cost_per_1m_tokens=output_cost_per_1m_tokens,
)
|