refac: settings

This commit is contained in:
Timothy J. Baek 2024-02-21 19:06:27 -08:00
parent 797ed9ac7f
commit dc705c4984
2 changed files with 36 additions and 18 deletions

View file

@ -62,16 +62,16 @@ async def get_openai_url(user=Depends(get_admin_user)):
@app.post("/url/update")
async def update_openai_url(form_data: UrlUpdateForm, user=Depends(get_admin_user)):
try:
r = requests.head(form_data.url)
if r.ok:
app.state.AUTOMATIC1111_BASE_URL = form_data.url.strip("/")
return {
"AUTOMATIC1111_BASE_URL": app.state.AUTOMATIC1111_BASE_URL,
"status": True,
}
except Exception as e:
raise HTTPException(status_code=r.status_code, detail=ERROR_MESSAGES.DEFAULT(e))
if form_data.url == "":
app.state.AUTOMATIC1111_BASE_URL = AUTOMATIC1111_BASE_URL
else:
app.state.AUTOMATIC1111_BASE_URL = form_data.url.strip("/")
return {
"AUTOMATIC1111_BASE_URL": app.state.AUTOMATIC1111_BASE_URL,
"status": True,
}
@app.get("/models")