forked from open-webui/open-webui
139 lines
3.8 KiB
Python
139 lines
3.8 KiB
Python
from fastapi import Depends, FastAPI, HTTPException, status
|
|
from datetime import datetime, timedelta
|
|
from typing import List, Union, Optional
|
|
|
|
from fastapi import APIRouter
|
|
from pydantic import BaseModel
|
|
import json
|
|
from apps.web.models.modelfiles import (
|
|
Modelfiles,
|
|
ModelfileForm,
|
|
ModelfileTagNameForm,
|
|
ModelfileUpdateForm,
|
|
ModelfileResponse,
|
|
)
|
|
|
|
from utils.utils import get_current_user
|
|
from constants import ERROR_MESSAGES
|
|
|
|
router = APIRouter()
|
|
|
|
############################
|
|
# GetModelfiles
|
|
############################
|
|
|
|
|
|
@router.get("/", response_model=List[ModelfileResponse])
|
|
async def get_modelfiles(skip: int = 0, limit: int = 50, user=Depends(get_current_user)):
|
|
return Modelfiles.get_modelfiles(skip, limit)
|
|
|
|
|
|
############################
|
|
# CreateNewModelfile
|
|
############################
|
|
|
|
|
|
@router.post("/create", response_model=Optional[ModelfileResponse])
|
|
async def create_new_modelfile(
|
|
form_data: ModelfileForm, user=Depends(get_current_user)
|
|
):
|
|
if user.role != "admin":
|
|
raise HTTPException(
|
|
status_code=status.HTTP_401_UNAUTHORIZED,
|
|
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
|
|
)
|
|
|
|
modelfile = Modelfiles.insert_new_modelfile(user.id, form_data)
|
|
|
|
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(),
|
|
)
|
|
|
|
|
|
############################
|
|
# GetModelfileByTagName
|
|
############################
|
|
|
|
|
|
@router.post("/", response_model=Optional[ModelfileResponse])
|
|
async def get_modelfile_by_tag_name(form_data: ModelfileTagNameForm, user=Depends(get_current_user)):
|
|
modelfile = Modelfiles.get_modelfile_by_tag_name(form_data.tag_name)
|
|
|
|
if modelfile:
|
|
return ModelfileResponse(
|
|
**{
|
|
**modelfile.model_dump(),
|
|
"modelfile": json.loads(modelfile.modelfile),
|
|
}
|
|
)
|
|
else:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_401_UNAUTHORIZED,
|
|
detail=ERROR_MESSAGES.NOT_FOUND,
|
|
)
|
|
|
|
|
|
############################
|
|
# UpdateModelfileByTagName
|
|
############################
|
|
|
|
|
|
@router.post("/update", response_model=Optional[ModelfileResponse])
|
|
async def update_modelfile_by_tag_name(
|
|
form_data: ModelfileUpdateForm, user=Depends(get_current_user)
|
|
):
|
|
if user.role != "admin":
|
|
raise HTTPException(
|
|
status_code=status.HTTP_401_UNAUTHORIZED,
|
|
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
|
|
)
|
|
modelfile = Modelfiles.get_modelfile_by_tag_name(form_data.tag_name)
|
|
if modelfile:
|
|
updated_modelfile = {
|
|
**json.loads(modelfile.modelfile),
|
|
**form_data.modelfile,
|
|
}
|
|
|
|
modelfile = Modelfiles.update_modelfile_by_tag_name(
|
|
form_data.tag_name, updated_modelfile
|
|
)
|
|
|
|
return ModelfileResponse(
|
|
**{
|
|
**modelfile.model_dump(),
|
|
"modelfile": json.loads(modelfile.modelfile),
|
|
}
|
|
)
|
|
else:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_401_UNAUTHORIZED,
|
|
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
|
|
)
|
|
|
|
|
|
############################
|
|
# DeleteModelfileByTagName
|
|
############################
|
|
|
|
|
|
@router.delete("/delete", response_model=bool)
|
|
async def delete_modelfile_by_tag_name(
|
|
form_data: ModelfileTagNameForm, user=Depends(get_current_user)
|
|
):
|
|
if user.role != "admin":
|
|
raise HTTPException(
|
|
status_code=status.HTTP_401_UNAUTHORIZED,
|
|
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
|
|
)
|
|
|
|
result = Modelfiles.delete_modelfile_by_tag_name(form_data.tag_name)
|
|
return result
|