from typing import Literal

from screensuite.benchmarks.hubbasebenchmark import HubBaseBenchmarkConfig


class GaiaConfig(HubBaseBenchmarkConfig):
    hf_repo: Literal["GeekAgents/GAIA-web"] = "GeekAgents/GAIA-web"
    """HF repo name"""

    split: Literal["validation"] = "validation"
    """Split to use."""

    @classmethod
    def create(cls) -> "GaiaConfig":
        return cls()
