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
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) -> DictConfig:
        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 cfg

    # @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_agent_config(cls, name: str = "default") -> AgentConfig:
        """Load agent config from /configs/agents"""
        cfg = cls._load_config_to_dict(name, config_path="../../configs/agents")
        return AgentConfig(**cfg)

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

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

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

load_agent_config classmethod 

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

Load agent config from /configs/agents

Source code in utu/config/loader.py 
34
35
36
37
38
@classmethod
def load_agent_config(cls, name: str = "default") -> AgentConfig:
    """Load agent config from /configs/agents"""
    cfg = cls._load_config_to_dict(name, config_path="../../configs/agents")
    return AgentConfig(**cfg)

load_toolkit_config classmethod 

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

Load toolkit config from /configs/agents/tools

Source code in utu/config/loader.py 
40
41
42
43
44
@classmethod
def load_toolkit_config(cls, name: str = "search") -> ToolkitConfig:
    """Load toolkit config from /configs/agents/tools"""
    cfg = cls._load_config_to_dict(name, config_path="../../configs/agents/tools")
    return ToolkitConfig(**cfg)

load_model_config classmethod 

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

Load model config from /configs/agents/model

Source code in utu/config/loader.py 
46
47
48
49
50
@classmethod
def load_model_config(cls, name: str = "base") -> ModelConfigs:
    """Load model config from /configs/agents/model"""
    cfg = cls._load_config_to_dict(name, config_path="../../configs/agents/model")
    return ModelConfigs(**cfg)

load_eval_config classmethod 

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

Load eval config from /configs

Source code in utu/config/loader.py 
52
53
54
55
56
57
58
@classmethod
def load_eval_config(cls, name: str = "default") -> EvalConfig:
    """Load eval config from /configs"""
    if not name.startswith("eval/"):
        name = "eval/" + name
    cfg = cls._load_config_to_dict(name, config_path="../../configs")
    return EvalConfig(**cfg)