15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-26 10:12:51 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

feat: add model setting

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 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 django.utils.translation import gettext_lazy as _
@ -72,3 +72,9 @@ class ModelEditApi(APIMixin):
@staticmethod
def get_response():
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.authentication import has_permissions
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 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.serializers.model_serializer import ModelSerializer
@ -82,6 +82,7 @@ class Model(APIView):
description=_('Delete model'),
operation_id=_('Delete model'),
parameters=GetModelApi.get_query_params_api(),
responses=DefaultModelResponse.get_response(),
tags=[_('Model')])
@has_permissions(PermissionConstants.MODEL_DELETE.get_workspace_permission())
def delete(self, request: Request, workspace_id: str, model_id: str):
@ -147,6 +148,7 @@ class Model(APIView):
description=_('Pause model download'),
operation_id=_('Pause model download'),
parameters=GetModelApi.get_query_params_api(),
responses=DefaultModelResponse.get_response(),
tags=[_('Model')])
@has_permissions(PermissionConstants.MODEL_CREATE.get_workspace_permission())
def put(self, request: Request, workspace_id: str, model_id: str):