File size: 645 Bytes
0f166dc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
import os
from dataclasses import dataclass
@dataclass
class AppConfig:
hf_model_repo: str
sql_backend: str # "bigquery" or "motherduck"
gcp_project: str | None = None
motherduck_db: str | None = None
motherduck_token: str | None = None
@classmethod
def from_env(cls):
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")
) |