From 021020632a5d4e1ef6c23c3e74f61fb9a3c9d41b Mon Sep 17 00:00:00 2001 From: zhanweidu Date: Sun, 27 Apr 2025 14:18:47 +0800 Subject: [PATCH] add switch to disable open api doc (#3744) Signed-off-by: congcongke --- python/sglang/srt/entrypoints/http_server.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/python/sglang/srt/entrypoints/http_server.py b/python/sglang/srt/entrypoints/http_server.py index 7f075f176..4754c84d0 100644 --- a/python/sglang/srt/entrypoints/http_server.py +++ b/python/sglang/srt/entrypoints/http_server.py @@ -84,6 +84,7 @@ from sglang.srt.utils import ( add_api_key_middleware, add_prometheus_middleware, delete_directory, + get_bool_env_var, kill_process_tree, set_uvicorn_logging_configs, ) @@ -126,7 +127,10 @@ async def lifespan(fast_api_app: FastAPI): # Fast API -app = FastAPI(lifespan=lifespan) +app = FastAPI( + lifespan=lifespan, + openapi_url=None if get_bool_env_var("DISABLE_OPENAPI_DOC") else "/openapi.json", +) app.add_middleware( CORSMiddleware, allow_origins=["*"],