theroyallab/tabbyAPI
1
0
Fork 0
mirror of https://github.com/theroyallab/tabbyAPI.git synced 2026-04-29 02:31:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

API: Add auth enforcement to draft list

Browse Source 
This didn't have an API key gate.

Signed-off-by: kingbri <bdashore3@proton.me>
This commit is contained in:
kingbri
2023-12-21 23:14:04 -05:00
parent 5d80a049ae
commit 99a798e117
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
main.py
View File

@@ -101,7 +101,7 @@ async def get_current_model():
return model_card 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(): async def list_draft_models():
model_config = unwrap(config.get("model"), {}) model_config = unwrap(config.get("model"), {})
draft_config = unwrap(model_config.get("draft"), {}) draft_config = unwrap(model_config.get("draft"), {})