modelfiles.py 3.78 KB
Newer Older
Timothy J. Baek's avatar
Timothy J. Baek committed
1
2
3
4
5
6
7
8
9
10
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,
11
12
    ModelfileTagNameForm,
    ModelfileUpdateForm,
Timothy J. Baek's avatar
Timothy J. Baek committed
13
14
15
    ModelfileResponse,
)

16
from utils.utils import get_current_user
Timothy J. Baek's avatar
Timothy J. Baek committed
17
18
19
20
21
22
23
24
25
26
from constants import ERROR_MESSAGES

router = APIRouter()

############################
# GetModelfiles
############################


@router.get("/", response_model=List[ModelfileResponse])
27
async def get_modelfiles(skip: int = 0, limit: int = 50, user=Depends(get_current_user)):
28
    return Modelfiles.get_modelfiles(skip, limit)
Timothy J. Baek's avatar
Timothy J. Baek committed
29
30
31
32
33
34
35
36


############################
# CreateNewModelfile
############################


@router.post("/create", response_model=Optional[ModelfileResponse])
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
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),
            }
        )
Timothy J. Baek's avatar
Timothy J. Baek committed
55
56
57
    else:
        raise HTTPException(
            status_code=status.HTTP_401_UNAUTHORIZED,
58
            detail=ERROR_MESSAGES.DEFAULT(),
Timothy J. Baek's avatar
Timothy J. Baek committed
59
60
61
62
63
64
65
66
        )


############################
# GetModelfileByTagName
############################


67
@router.post("/", response_model=Optional[ModelfileResponse])
68
async def get_modelfile_by_tag_name(form_data: ModelfileTagNameForm, user=Depends(get_current_user)):
69
70
71
72
73
74
75
76
77
    modelfile = Modelfiles.get_modelfile_by_tag_name(form_data.tag_name)

    if modelfile:
        return ModelfileResponse(
            **{
                **modelfile.model_dump(),
                "modelfile": json.loads(modelfile.modelfile),
            }
        )
Timothy J. Baek's avatar
Timothy J. Baek committed
78
79
80
    else:
        raise HTTPException(
            status_code=status.HTTP_401_UNAUTHORIZED,
81
            detail=ERROR_MESSAGES.NOT_FOUND,
Timothy J. Baek's avatar
Timothy J. Baek committed
82
83
84
85
86
87
88
89
        )


############################
# UpdateModelfileByTagName
############################


90
@router.post("/update", response_model=Optional[ModelfileResponse])
Timothy J. Baek's avatar
Timothy J. Baek committed
91
async def update_modelfile_by_tag_name(
92
    form_data: ModelfileUpdateForm, user=Depends(get_current_user)
Timothy J. Baek's avatar
Timothy J. Baek committed
93
):
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
    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),
            }
        )
Timothy J. Baek's avatar
Timothy J. Baek committed
116
117
118
    else:
        raise HTTPException(
            status_code=status.HTTP_401_UNAUTHORIZED,
119
            detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
Timothy J. Baek's avatar
Timothy J. Baek committed
120
121
122
123
124
125
126
127
        )


############################
# DeleteModelfileByTagName
############################


128
129
@router.delete("/delete", response_model=bool)
async def delete_modelfile_by_tag_name(
130
    form_data: ModelfileTagNameForm, user=Depends(get_current_user)
131
):
132
    if user.role != "admin":
Timothy J. Baek's avatar
Timothy J. Baek committed
133
134
        raise HTTPException(
            status_code=status.HTTP_401_UNAUTHORIZED,
135
            detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
Timothy J. Baek's avatar
Timothy J. Baek committed
136
        )
137
138
139

    result = Modelfiles.delete_modelfile_by_tag_name(form_data.tag_name)
    return result