feat: litellm proxy frontend integration

This commit is contained in:
Timothy J. Baek 2024-02-24 17:56:53 -08:00
parent a453be07e1
commit ec6f53e298
6 changed files with 823 additions and 360 deletions

View file

@ -109,6 +109,7 @@ def get_models(user=Depends(get_current_user)):
models = r.json()
return models
except Exception as e:
app.state.ENABLED = False
raise HTTPException(status_code=400, detail=ERROR_MESSAGES.DEFAULT(e))
@ -120,6 +121,7 @@ async def get_default_model(user=Depends(get_admin_user)):
return {"model": options["sd_model_checkpoint"]}
except Exception as e:
app.state.ENABLED = False
raise HTTPException(status_code=400, detail=ERROR_MESSAGES.DEFAULT(e))