jalr/tabbyAPI-ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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
@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"), {})