Mirrors/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-16 11:57:51 +01:00
Code Issues Packages Projects Releases Wiki Activity

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,