From 99a798e117690caa8fc4199d9a2c33c7ad481640 Mon Sep 17 00:00:00 2001 From: kingbri Date: Thu, 21 Dec 2023 23:14:04 -0500 Subject: [PATCH] API: Add auth enforcement to draft list This didn't have an API key gate. Signed-off-by: kingbri --- main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/main.py b/main.py index b0b6a06..61a8f51 100644 --- a/main.py +++ b/main.py @@ -101,7 +101,7 @@ async def get_current_model(): return model_card -@app.get("/v1/model/draft/list") +@app.get("/v1/model/draft/list", dependencies=[Depends(check_api_key)]) async def list_draft_models(): model_config = unwrap(config.get("model"), {}) draft_config = unwrap(model_config.get("draft"), {})