NexQuant/rdagent/components/runner/conf.py

from __future__ import annotations

from pathlib import Path

from dotenv import load_dotenv
from pydantic_settings import BaseSettings

# make sure that env variable is loaded while calling Config()
load_dotenv(verbose=True, override=True)

from pydantic_settings import BaseSettings


class RunnerSettings(BaseSettings):
    class Config:
        env_prefix = "RUNNER_"  # Use MODEL_CODER_ as prefix for environment variables

    cache_result: bool = True  # whether to cache the result of the docker execution
    cache_path: str = str(Path.cwd() / "runner_cache/")  # the path to store the cache


RUNNER_SETTINGS = RunnerSettings()