mirror of
https://github.com/open-webui/open-webui
synced 2024-11-24 21:13:59 +00:00
127 lines
3.1 KiB
Python
127 lines
3.1 KiB
Python
from typing import Optional
|
|
|
|
from open_webui.apps.webui.models.models import (
|
|
ModelForm,
|
|
ModelModel,
|
|
ModelResponse,
|
|
Models,
|
|
)
|
|
from open_webui.constants import ERROR_MESSAGES
|
|
from fastapi import APIRouter, Depends, HTTPException, Request, status
|
|
|
|
|
|
from open_webui.utils.utils import get_admin_user, get_verified_user, has_access
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
###########################
|
|
# GetModels
|
|
###########################
|
|
|
|
|
|
@router.get("/", response_model=list[ModelResponse])
|
|
async def get_models(id: Optional[str] = None, user=Depends(get_verified_user)):
|
|
if user.role == "admin":
|
|
return Models.get_models()
|
|
else:
|
|
return Models.get_models_by_user_id(user.id)
|
|
|
|
|
|
############################
|
|
# CreateNewModel
|
|
############################
|
|
|
|
|
|
@router.post("/create", response_model=Optional[ModelModel])
|
|
async def create_new_model(
|
|
form_data: ModelForm,
|
|
user=Depends(get_verified_user),
|
|
):
|
|
|
|
model = Models.get_model_by_id(form_data.id)
|
|
if model:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_401_UNAUTHORIZED,
|
|
detail=ERROR_MESSAGES.MODEL_ID_TAKEN,
|
|
)
|
|
|
|
else:
|
|
model = Models.insert_new_model(form_data, user.id)
|
|
if model:
|
|
return model
|
|
else:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_401_UNAUTHORIZED,
|
|
detail=ERROR_MESSAGES.DEFAULT(),
|
|
)
|
|
|
|
|
|
###########################
|
|
# GetModelById
|
|
###########################
|
|
|
|
|
|
@router.get("/id/{id}", 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:
|
|
if (
|
|
user.role == "admin"
|
|
or model.user_id == user.id
|
|
or has_access(user.id, "read", model.access_control)
|
|
):
|
|
return model
|
|
else:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_401_UNAUTHORIZED,
|
|
detail=ERROR_MESSAGES.NOT_FOUND,
|
|
)
|
|
|
|
|
|
############################
|
|
# UpdateModelById
|
|
############################
|
|
|
|
|
|
@router.post("/id/{id}/update", response_model=Optional[ModelModel])
|
|
async def update_model_by_id(
|
|
id: str,
|
|
form_data: ModelForm,
|
|
user=Depends(get_verified_user),
|
|
):
|
|
model = Models.get_model_by_id(id)
|
|
|
|
if not model:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_401_UNAUTHORIZED,
|
|
detail=ERROR_MESSAGES.NOT_FOUND,
|
|
)
|
|
|
|
model = Models.update_model_by_id(id, form_data)
|
|
return model
|
|
|
|
|
|
############################
|
|
# DeleteModelById
|
|
############################
|
|
|
|
|
|
@router.delete("/id/{id}/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:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_401_UNAUTHORIZED,
|
|
detail=ERROR_MESSAGES.NOT_FOUND,
|
|
)
|
|
|
|
if model.user_id != user.id and user.role != "admin":
|
|
raise HTTPException(
|
|
status_code=status.HTTP_401_UNAUTHORIZED,
|
|
detail=ERROR_MESSAGES.UNAUTHORIZED,
|
|
)
|
|
|
|
result = Models.delete_model_by_id(id)
|
|
return result
|