From 33964233a5cfa9779a3a75d626b01ab6223f6f3b Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Thu, 21 Mar 2024 17:15:18 -0700 Subject: [PATCH] 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 --- litellm/proxy/proxy_server.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 0d653935832..738fd2d633d 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -162,8 +162,11 @@ ui_link = f"/ui/" ui_message = ( 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( - docs_url="/", + docs_url=_docs_url, title="LiteLLM API", description=f"Proxy Server to call 100+ LLMs in the OpenAI format\n\n{ui_message}", version=version,