Skip to content

Loader

ConfigLoader

Config loader

Source code in utu/config/loader.py 
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
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)

load_model_config classmethod 

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

Load model config

Source code in utu/config/loader.py 
35
36
37
38
39
@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

Source code in utu/config/loader.py 
41
42
43
44
45
@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

Source code in utu/config/loader.py 
47
48
49
50
51
52
53
@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

Source code in utu/config/loader.py 
55
56
57
58
59
60
61
@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)