Merge pull request #2263 from tisnik/lcore-3141-models-group-schema-export

LCORE-3141: models group schema export ability

Author: tisnik

README.md

@@ -860,6 +860,8 @@ options:
                         dump actual configuration into JSON file and quit
   -s, --dump-schema     dump configuration schema into OpenAPI-compatible file and quit
   -m, --dump-models     dump schemas for all models into OpenAPI-compatible file and quit
+  -gr, --dump-models-group DUMP_MODELS_GROUP
+                        dump schemas for selected models group into OpenAPI-compatible file and quit
   -c, --config CONFIG_FILE
                         path to configuration file (default: lightspeed-stack.yaml)
   --synthesized-config-output SYNTHESIZED_CONFIG_OUTPUT

src/lightspeed_stack.py

@@ -27,6 +27,9 @@ def create_argument_parser() -> ArgumentParser:
     - -v / --verbose: enable verbose output
     - -d / --dump-configuration: dump the loaded configuration to JSON and exit
     - -s / --dump-schema: dump the configuration schema to OpenAPI JSON and exit
+    - -m / --dump-models: dump schemas for all models into OpenAPI-compatible file and quit
+    - -gr / --dump-models-group DUMP_MODELS_GROUP
+                        dump schemas for selected models group into OpenAPI-compatible file and quit
     - -c / --config: path to the configuration file (default "lightspeed-stack.yaml")
     - -g / --generate-llama-stack-configuration: generate a Llama Stack
                                                  configuration from the service configuration
@@ -69,6 +72,14 @@ def create_argument_parser() -> ArgumentParser:
         action="store_true",
         default=False,
     )
+    parser.add_argument(
+        "-gr",
+        "--dump-models-group",
+        dest="dump_models_group",
+        help="dump schemas for selected models group into OpenAPI-compatible file and quit",
+        action="store",
+        default=None,
+    )
     parser.add_argument(
         "-c",
         "--config",
@@ -213,6 +224,18 @@ def main() -> None:
             raise SystemExit(1) from e
         return
 
+    # -gr or --dump-models-group CLI parameter is used to dump schema for
+    # selected models into a JSON file that is compatible with OpenAPI schema
+    # specification
+    if args.dump_models_group is not None:
+        try:
+            models_dumper.dump_models_group(args.dump_models_group)
+            logger.info("Schema for group %s of models dumped", args.dump_models_group)
+        except Exception as e:
+            logger.error("Failed to dump schema for models: %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[constants.CONFIG_PATH_ENV_VAR] = args.config_file