From 021981fce03aa3a69226709e951d36b6a4546645 Mon Sep 17 00:00:00 2001 From: kingbri Date: Fri, 17 Nov 2023 00:47:51 -0500 Subject: [PATCH] API: Re-add depends endpoints Mistakenly removed API key authentication for the models endpoints in testing. Signed-off-by: kingbri --- main.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/main.py b/main.py index 9b37840..4227409 100644 --- a/main.py +++ b/main.py @@ -47,8 +47,8 @@ app.add_middleware( ) # Model list endpoint -@app.get("/v1/models") -@app.get("/v1/model/list") +@app.get("/v1/models", dependencies=[Depends(check_api_key)]) +@app.get("/v1/model/list", dependencies=[Depends(check_api_key)]) async def list_models(): model_config = config.get("model", {}) if "model_dir" in model_config: