natbutter/functiongemma-tuning-lab

1import os2from pathlib import Path3from typing import Final, Optional, List4from dataclasses import dataclass, field5 6@dataclass7class AppConfig:8    """9    Central configuration class.10    """11    # Directory Setup12    ARTIFACTS_DIR: Final[Path] = Path("artifacts")13    OUTPUT_DIR: Final[Path] = ARTIFACTS_DIR.joinpath("functiongemma-tuning-lab-demo")14    15    # Model & Data16    HF_TOKEN: Final[Optional[str]] = os.getenv('HF_TOKEN')17    18    # Model Configuration19    # Mutable: User can change this in the UI20    MODEL_NAME: str = 'google/functiongemma-270m-it'21 22    AVAILABLE_MODELS: List[str] = field(default_factory=lambda: [23        'google/functiongemma-270m-it'24    ])25 26    DEFAULT_DATASET: Final[str] = 'bebechien/SimpleToolCalling'27    28    def __post_init__(self):29        self.ARTIFACTS_DIR.mkdir(parents=True, exist_ok=True)30