ALM_LLM / utils /config.py
AshenH's picture
Update utils/config.py
81516f3 verified
raw
history blame
883 Bytes
import os
from dataclasses import dataclass
@dataclass
class AppConfig:
"""
Central configuration for the Tabular Agentic XAI app.
"""
# Common
hf_model_repo: str
sql_backend: str # "bigquery" or "motherduck"
# BigQuery
gcp_project: str | None = None
# MotherDuck
motherduck_db: str | None = None
motherduck_token: str | None = None
@classmethod
def from_env(cls):
"""
Reads env vars from .env (local) or Space Secrets (HF Spaces).
"""
return cls(
hf_model_repo=os.getenv("HF_MODEL_REPO", "your-username/your-private-tabular-model"),
sql_backend=os.getenv("SQL_BACKEND", "motherduck"),
gcp_project=os.getenv("GCP_PROJECT"),
motherduck_db=os.getenv("MOTHERDUCK_DB", "default"),
motherduck_token=os.getenv("MOTHERDUCK_TOKEN")
)