jalr/tabbyAPI-ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Main: Fix launch if EXPORT_OPENAPI is unset

Browse source 
A default needs to be provided with getenv. Fix that with an empty
string.

Signed-off-by: kingbri <bdashore3@proton.me>
This commit is contained in:
kingbri 2024-07-08 13:41:44 -04:00
parent a59e8ef9e7
commit b907421285
2 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
common/model.py
View file

@ -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

2
main.py
View file

@ -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: