LCORE-642: Llama Stack configuration processing stub

Author: tisnik

src/lightspeed_stack.py

@@ -10,15 +10,17 @@
 
 from rich.logging import RichHandler
 
+from log import get_logger
 from configuration import configuration
+from llama_stack_configuration import generate_configuration
 from runners.uvicorn import start_uvicorn
 
 FORMAT = "%(message)s"
 logging.basicConfig(
     level="INFO", format=FORMAT, datefmt="[%X]", handlers=[RichHandler()]
 )
 
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 
 
 def create_argument_parser() -> ArgumentParser:
@@ -47,6 +49,28 @@ def create_argument_parser() -> ArgumentParser:
         help="path to configuration file (default: lightspeed-stack.yaml)",
         default="lightspeed-stack.yaml",
     )
+    parser.add_argument(
+        "-g",
+        "--generate-llama-stack-configuration",
+        dest="generate_llama_stack_configuration",
+        help="generate Llama Stack configuration based on LCORE configuration",
+        action="store_true",
+        default=False,
+    )
+    parser.add_argument(
+        "-i",
+        "--input-config-file",
+        dest="input_config_file",
+        help="Llama Stack input configuration file",
+        default="run.yaml",
+    )
+    parser.add_argument(
+        "-o",
+        "--output-config-file",
+        dest="output_config_file",
+        help="Llama Stack output configuration file",
+        default="run_.yaml",
+    )
 
     return parser
 
@@ -74,6 +98,24 @@ def main() -> None:
             raise SystemExit(1) from e
         return
 
+    # -g or --generate-llama-stack-configuration CLI flags are used to (re)generate
+    # configuration for Llama Stack
+    if args.generate_llama_stack_configuration:
+        try:
+            generate_configuration(
+                args.input_config_file,
+                args.output_config_file,
+                configuration.configuration,
+            )
+            logger.info(
+                "Llama Stack configuration generated and stored into %s",
+                args.output_config_file,
+            )
+        except Exception as e:
+            logger.error("Failed to generate Llama Stack configuration: %s", e)
+            raise SystemExit(1) from e
+        return
+
     # Store config path in env so each uvicorn worker can load it
     # (step is needed because process context isn’t shared).
     os.environ["LIGHTSPEED_STACK_CONFIG_PATH"] = args.config_file

src/llama_stack_configuration.py

@@ -0,0 +1,37 @@
+"""Llama Stack configuration handling."""
+
+import yaml
+
+from log import get_logger
+
+from models.config import Configuration
+
+logger = get_logger(__name__)
+
+
+# pylint: disable=too-many-ancestors
+class YamlDumper(yaml.Dumper):
+    """Custom YAML dumper with proper indentation levels."""
+
+    def increase_indent(self, flow: bool = False, indentless: bool = False) -> None:
+        """Control the indentation level of formatted YAML output."""
+        _ = indentless
+        return super().increase_indent(flow, False)
+
+
+def generate_configuration(
+    input_file: str, output_file: str, config: Configuration
+) -> None:
+    """Generate new Llama Stack configuration."""
+    logger.info("Reading Llama Stack configuration from file %s", input_file)
+
+    with open(input_file, "r", encoding="utf-8") as file:
+        ls_config = yaml.safe_load(file)
+
+    logger.info("Processing Llama Stack configuration")
+    _ = config
+
+    logger.info("Writing Llama Stack configuration into file %s", output_file)
+
+    with open(output_file, "w", encoding="utf-8") as file:
+        yaml.dump(ls_config, file, Dumper=YamlDumper, default_flow_style=False)