diff --git a/src/configuration.py b/src/configuration.py index 5c82620c..496f0e1c 100644 --- a/src/configuration.py +++ b/src/configuration.py @@ -4,7 +4,12 @@ from typing import Any, Optional import yaml -from models.config import Configuration, LLamaStackConfiguration, UserDataCollection +from models.config import ( + Configuration, + LLamaStackConfiguration, + UserDataCollection, + ServiceConfiguration, +) logger = logging.getLogger(__name__) @@ -43,6 +48,14 @@ def configuration(self) -> Configuration: ), "logic error: configuration is not loaded" return self._configuration + @property + def service_configuration(self) -> ServiceConfiguration: + """Return service configuration.""" + assert ( + self._configuration is not None + ), "logic error: configuration is not loaded" + return self._configuration.service + @property def llama_stack_configuration(self) -> LLamaStackConfiguration: """Return Llama stack configuration.""" diff --git a/tests/unit/test_configuration.py b/tests/unit/test_configuration.py index 46361b2f..43c09ffd 100644 --- a/tests/unit/test_configuration.py +++ b/tests/unit/test_configuration.py @@ -48,6 +48,7 @@ def test_init_from_dict() -> None: cfg.init_from_dict(config_dict) assert cfg.configuration is not None assert cfg.llama_stack_configuration is not None + assert cfg.service_configuration is not None def test_load_proper_configuration(tmpdir) -> None: @@ -76,3 +77,4 @@ def test_load_proper_configuration(tmpdir) -> None: cfg.load_configuration(cfg_filename) assert cfg.configuration is not None assert cfg.llama_stack_configuration is not None + assert cfg.service_configuration is not None