zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add model setting

Browse Source
This commit is contained in:
wxg0103 2025-04-21 10:24:13 +08:00
parent d3346deb62
commit 39841cb344
2 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
apps/models_provider/api/model.py
View File

@ -4,7 +4,7 @@ from drf_spectacular.utils import OpenApiParameter
from rest_framework import serializers from rest_framework import serializers
from common.mixins.api_mixin import APIMixin from common.mixins.api_mixin import APIMixin
from common.result import ResultSerializer from common.result import ResultSerializer, DefaultResultSerializer
from models_provider.serializers.model_serializer import ModelModelSerializer, ModelCreateRequest from models_provider.serializers.model_serializer import ModelModelSerializer, ModelCreateRequest
from django.utils.translation import gettext_lazy as _ from django.utils.translation import gettext_lazy as _
@ -72,3 +72,9 @@ class ModelEditApi(APIMixin):
@staticmethod @staticmethod
def get_response(): def get_response():
return ModelModelSerializer return ModelModelSerializer
class DefaultModelResponse(APIMixin):
@staticmethod
def get_response():
return DefaultResultSerializer()

6
apps/models_provider/views/model.py
View File

@ -14,9 +14,9 @@ from rest_framework.request import Request
from common.auth import TokenAuth from common.auth import TokenAuth
from common.auth.authentication import has_permissions from common.auth.authentication import has_permissions
from common.constants.permission_constants import PermissionConstants from common.constants.permission_constants import PermissionConstants
from common.result import result from common.result import result, DefaultResultSerializer
from common.utils.common import query_params_to_single_dict from common.utils.common import query_params_to_single_dict
from models_provider.api.model import ModelCreateAPI, GetModelApi, ModelEditApi, ModelListResponse from models_provider.api.model import ModelCreateAPI, GetModelApi, ModelEditApi, ModelListResponse, DefaultModelResponse
from models_provider.api.provide import ProvideApi from models_provider.api.provide import ProvideApi
from models_provider.serializers.model_serializer import ModelSerializer from models_provider.serializers.model_serializer import ModelSerializer
@ -82,6 +82,7 @@ class Model(APIView):
description=_('Delete model'), description=_('Delete model'),
operation_id=_('Delete model'), operation_id=_('Delete model'),
parameters=GetModelApi.get_query_params_api(), parameters=GetModelApi.get_query_params_api(),
responses=DefaultModelResponse.get_response(),
tags=[_('Model')]) tags=[_('Model')])
@has_permissions(PermissionConstants.MODEL_DELETE.get_workspace_permission()) @has_permissions(PermissionConstants.MODEL_DELETE.get_workspace_permission())
def delete(self, request: Request, workspace_id: str, model_id: str): def delete(self, request: Request, workspace_id: str, model_id: str):
@ -147,6 +148,7 @@ class Model(APIView):
description=_('Pause model download'), description=_('Pause model download'),
operation_id=_('Pause model download'), operation_id=_('Pause model download'),
parameters=GetModelApi.get_query_params_api(), parameters=GetModelApi.get_query_params_api(),
responses=DefaultModelResponse.get_response(),
tags=[_('Model')]) tags=[_('Model')])
@has_permissions(PermissionConstants.MODEL_CREATE.get_workspace_permission()) @has_permissions(PermissionConstants.MODEL_CREATE.get_workspace_permission())
def put(self, request: Request, workspace_id: str, model_id: str): def put(self, request: Request, workspace_id: str, model_id: str):