fix(proxy_server.py): allow user to disable swagger ui docs via env

user can disable swagger ui docs by setting 'NO_DOCS="True"' in their env
This commit is contained in:
Krrish Dholakia 2024-03-21 17:15:18 -07:00
parent 44a91fe43a
commit 33964233a5

View file

@ -162,8 +162,11 @@ ui_link = f"/ui/"
ui_message = ( ui_message = (
f"👉 [```LiteLLM Admin Panel on /ui```]({ui_link}). Create, Edit Keys with SSO" f"👉 [```LiteLLM Admin Panel on /ui```]({ui_link}). Create, Edit Keys with SSO"
) )
_docs_url = None if os.getenv("NO_DOCS", "False") == "True" else "/"
app = FastAPI( app = FastAPI(
docs_url="/", docs_url=_docs_url,
title="LiteLLM API", title="LiteLLM API",
description=f"Proxy Server to call 100+ LLMs in the OpenAI format\n\n{ui_message}", description=f"Proxy Server to call 100+ LLMs in the OpenAI format\n\n{ui_message}",
version=version, version=version,