36 lines
1.3 KiB
Python
36 lines
1.3 KiB
Python
|
# coding=utf-8
|
|||
|
|
"""
|
|||
|
|
@project: MaxKB
|
|||
|
|
@Author：虎虎
|
|||
|
|
@file： user.py
|
|||
|
|
@date：2025/4/14 19:25
|
|||
|
|
@desc:
|
|||
|
|
"""
|
|||
|
|
from drf_spectacular.utils import extend_schema
|
|||
|
|
from rest_framework.views import APIView
|
|||
|
|
from django.utils.translation import gettext_lazy as _
|
|||
|
|
from rest_framework.request import Request
|
|||
|
|
|
|||
|
|
from common.auth import TokenAuth
|
|||
|
|
from common.auth.authentication import has_permissions
|
|||
|
|
from common.constants.permission_constants import PermissionConstants
|
|||
|
|
from common.result import result
|
|||
|
|
from models_provider.api.model import ModelCreateAPI
|
|||
|
|
from models_provider.serializers.model import ModelSerializer
|
|||
|
|
|
|||
|
|
|
|||
|
|
class Model(APIView):
|
|||
|
|
authentication_classes = [TokenAuth]
|
|||
|
|
|
|||
|
|
@extend_schema(methods=['POST'],
|
|||
|
|
description=_("Create model"),
|
|||
|
|
operation_id=_("Create model"),
|
|||
|
|
tags=[_("Model")],
|
|||
|
|
request=ModelCreateAPI.get_request(),
|
|||
|
|
responses=ModelCreateAPI.get_response())
|
|||
|
|
@has_permissions(PermissionConstants.MODEL_CREATE)
|
|||
|
|
def post(self, request: Request, workspace_id: str):
|
|||
|
|
return result.success(
|
|||
|
|
ModelSerializer.Create(data={**request.data, 'user_id': request.user.id}).insert(workspace_id,
|
|||
|
|
with_valid=True))
|