Loader

ConfigLoader

Config loader

源代码位于： utu/config/loader.py

class ConfigLoader:
    """Config loader"""

    config_path = "../../configs"
    version_base = "1.3"

    @classmethod
    def _load_config_to_dict(cls, name: str = "default", config_path: str = None) -> dict:
        config_path = config_path or cls.config_path
        with initialize(config_path=config_path, version_base=cls.version_base):
            cfg = compose(config_name=name)
            OmegaConf.resolve(cfg)
        # return dict instead of DictConfig -- avoid JSON serialization error
        return OmegaConf.to_container(cfg, resolve=True)

    # @classmethod
    # def _load_config_to_cls(cls, name: str, config_type: Type[TConfig] = None) -> TConfig:
    #     # TESTING
    #     cfg = cls._load_config_to_dict(name)
    #     return config_type(**cfg)

    @classmethod
    def load_model_config(cls, name: str = "base") -> ModelConfigs:
        """Load model config"""
        cfg = cls._load_config_to_dict(name, config_path="../../configs/model")
        return ModelConfigs(**cfg)

    @classmethod
    def load_toolkit_config(cls, name: str = "search") -> ToolkitConfig:
        """Load toolkit config"""
        cfg = cls._load_config_to_dict(name, config_path="../../configs/tools")
        return ToolkitConfig(**cfg)

    @classmethod
    def load_agent_config(cls, name: str = "default") -> AgentConfig:
        """Load agent config"""
        if not name.startswith("agents/"):
            name = "agents/" + name
        cfg = cls._load_config_to_dict(name, config_path="../../configs")
        return AgentConfig(**cfg)

    @classmethod
    def load_eval_config(cls, name: str = "default") -> EvalConfig:
        """Load eval config"""
        if not name.startswith("eval/"):
            name = "eval/" + name
        cfg = cls._load_config_to_dict(name, config_path="../../configs")
        return EvalConfig(**cfg)

    @classmethod
    def load_training_free_grpo_config(cls, name: str = "default") -> TrainingFreeGRPOConfig:
        """Load training-free GRPO config"""
        if not name.startswith("practice/"):
            name = "practice/" + name
        cfg = cls._load_config_to_dict(name, config_path="../../configs")
        return TrainingFreeGRPOConfig(**cfg)

load_model_config classmethod

load_model_config(name: str = 'base') -> ModelConfigs

Load model config

源代码位于： utu/config/loader.py

@classmethod
def load_model_config(cls, name: str = "base") -> ModelConfigs:
    """Load model config"""
    cfg = cls._load_config_to_dict(name, config_path="../../configs/model")
    return ModelConfigs(**cfg)

load_toolkit_config classmethod

load_toolkit_config(name: str = 'search') -> ToolkitConfig

Load toolkit config

源代码位于： utu/config/loader.py

@classmethod
def load_toolkit_config(cls, name: str = "search") -> ToolkitConfig:
    """Load toolkit config"""
    cfg = cls._load_config_to_dict(name, config_path="../../configs/tools")
    return ToolkitConfig(**cfg)

load_agent_config classmethod

load_agent_config(name: str = 'default') -> AgentConfig

Load agent config

源代码位于： utu/config/loader.py

@classmethod
def load_agent_config(cls, name: str = "default") -> AgentConfig:
    """Load agent config"""
    if not name.startswith("agents/"):
        name = "agents/" + name
    cfg = cls._load_config_to_dict(name, config_path="../../configs")
    return AgentConfig(**cfg)

load_eval_config classmethod

load_eval_config(name: str = 'default') -> EvalConfig

Load eval config

源代码位于： utu/config/loader.py

@classmethod
def load_eval_config(cls, name: str = "default") -> EvalConfig:
    """Load eval config"""
    if not name.startswith("eval/"):
        name = "eval/" + name
    cfg = cls._load_config_to_dict(name, config_path="../../configs")
    return EvalConfig(**cfg)

load_training_free_grpo_config classmethod

load_training_free_grpo_config(name: str = 'default') -> TrainingFreeGRPOConfig

Load training-free GRPO config

源代码位于： utu/config/loader.py

@classmethod
def load_training_free_grpo_config(cls, name: str = "default") -> TrainingFreeGRPOConfig:
    """Load training-free GRPO config"""
    if not name.startswith("practice/"):
        name = "practice/" + name
    cfg = cls._load_config_to_dict(name, config_path="../../configs")
    return TrainingFreeGRPOConfig(**cfg)