OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-06-26 18:26:48 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

refac: models api

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2024-11-19 18:45:26 -08:00
parent 7c5b845d16
commit 7966367107
2 changed files with 9 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
backend/open_webui/apps/webui/routers/models.py
View File

@@ -83,7 +83,8 @@ async def create_new_model(
###########################
@router.get("/id/{id}", response_model=Optional[ModelResponse])
# Note: We're not using the typical url path param here, but instead using a query parameter to allow '/' in the id
@router.get("/model", response_model=Optional[ModelResponse])
async def get_model_by_id(id: str, user=Depends(get_verified_user)):
model = Models.get_model_by_id(id)
if model:
@@ -105,7 +106,7 @@ async def get_model_by_id(id: str, user=Depends(get_verified_user)):
############################
@router.post("/id/{id}/toggle", response_model=Optional[ModelResponse])
@router.post("/model/toggle", response_model=Optional[ModelResponse])
async def toggle_model_by_id(id: str, user=Depends(get_verified_user)):
model = Models.get_model_by_id(id)
if model:
@@ -140,7 +141,7 @@ async def toggle_model_by_id(id: str, user=Depends(get_verified_user)):
############################
@router.post("/id/{id}/update", response_model=Optional[ModelModel])
@router.post("/model/update", response_model=Optional[ModelModel])
async def update_model_by_id(
id: str,
form_data: ModelForm,
@@ -163,7 +164,7 @@ async def update_model_by_id(
############################
@router.delete("/id/{id}/delete", response_model=bool)
@router.delete("/model/delete", response_model=bool)
async def delete_model_by_id(id: str, user=Depends(get_verified_user)):
model = Models.get_model_by_id(id)
if not model: