tdpeuter/open-webui
tdpeuter/open-webui
1
Fork 0
forked from open-webui/open-webui
Code Issues Pull requests 1 Projects 1 Releases 2 Packages 1 Activity Actions

feat: modelfiles migration support

Browse source
This commit is contained in:
Timothy J. Baek 2023-12-26 23:58:40 -08:00
parent fbb64406d8
commit 21c7f50790
5 changed files with 124 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

13
backend/apps/web/models/modelfiles.py
View file

@ -74,11 +74,16 @@ class ModelfilesTable:
"timestamp": int(time.time()),
}
)
result = Modelfile.create(**modelfile.model_dump())
if result:
return modelfile
else:
try:
result = Modelfile.create(**modelfile.model_dump())
if result:
return modelfile
else:
return None
except:
return None
else:
return None

19
backend/apps/web/routers/modelfiles.py
View file

@ -56,12 +56,19 @@ async def create_new_modelfile(form_data: ModelfileForm, cred=Depends(bearer_sch
# Admin Only
if user.role == "admin":
modelfile = Modelfiles.insert_new_modelfile(user.id, form_data)
return ModelfileResponse(
**{
**modelfile.model_dump(),
"modelfile": json.loads(modelfile.modelfile),
}
)
if modelfile:
return ModelfileResponse(
**{
**modelfile.model_dump(),
"modelfile": json.loads(modelfile.modelfile),
}
)
else:
raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED,
detail=ERROR_MESSAGES.DEFAULT(),
)
else:
raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED,