Merge pull request #76 from tisnik/config-for-uvicorn

Config for uvicorn

Author: tisnik

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."""

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