diff --git a/scripts/generate_openapi_schema.py b/scripts/generate_openapi_schema.py index 2d97738a..11681ceb 100644 --- a/scripts/generate_openapi_schema.py +++ b/scripts/generate_openapi_schema.py @@ -10,9 +10,14 @@ # it is needed to read proper configuration in order to start the app to generate schema from configuration import configuration +from client import LlamaStackClientHolder + cfg_file = "lightspeed-stack.yaml" configuration.load_configuration(cfg_file) +# Llama Stack client needs to be loaded before REST API is fully initialized +LlamaStackClientHolder().load(configuration.configuration.llama_stack) + from app.main import app # noqa: E402 pylint: disable=C0413