From b907421285d17e3df4f67f8918b6e368eb6ae897 Mon Sep 17 00:00:00 2001 From: kingbri Date: Mon, 8 Jul 2024 13:41:44 -0400 Subject: [PATCH] Main: Fix launch if EXPORT_OPENAPI is unset A default needs to be provided with getenv. Fix that with an empty string. Signed-off-by: kingbri --- common/model.py | 2 +- main.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/common/model.py b/common/model.py index f37c765..dc65ed7 100644 --- a/common/model.py +++ b/common/model.py @@ -13,7 +13,7 @@ from common import config from common.logger import get_loading_progress_bar from common.utils import unwrap -if not os.getenv("EXPORT_OPENAPI"): +if not os.getenv("EXPORT_OPENAPI", "").lower() in ("true", "1"): from backends.exllamav2.model import ExllamaV2Container # Global model container diff --git a/main.py b/main.py index 03f6be7..efb153f 100644 --- a/main.py +++ b/main.py @@ -29,7 +29,7 @@ async def entrypoint(args: Optional[dict] = None): signal.signal(signal.SIGINT, signal_handler) signal.signal(signal.SIGTERM, signal_handler) - if os.getenv("EXPORT_OPENAPI").lower() in ("true", "1"): + if os.getenv("EXPORT_OPENAPI", "").lower() in ("true", "1"): openapi_json = export_openapi() async with aiofiles.open("openapi.json", "w") as f: