diff --git a/src/configuration.py b/src/configuration.py index 1b2e87c1..573520d2 100644 --- a/src/configuration.py +++ b/src/configuration.py @@ -3,6 +3,10 @@ import logging from typing import Any, Optional +# We want to support environment variable replacement in the configuration +# similarly to how it is done in llama-stack, so we use their function directly +from llama_stack.core.stack import replace_env_vars + import yaml from models.config import ( Configuration, @@ -16,6 +20,7 @@ DatabaseConfiguration, ) + logger = logging.getLogger(__name__) @@ -38,6 +43,7 @@ def load_configuration(self, filename: str) -> None: """Load configuration from YAML file.""" with open(filename, encoding="utf-8") as fin: config_dict = yaml.safe_load(fin) + config_dict = replace_env_vars(config_dict) logger.info("Loaded configuration: %s", config_dict) self.init_from_dict(config_dict)