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

refactor: model

Browse Source
This commit is contained in:
wxg0103 2025-06-05 18:37:06 +08:00
parent fee5993c22
commit d05d82106e
3 changed files with 15 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
apps/models_provider/urls.py
View File

@ -11,11 +11,11 @@ urlpatterns = [
path('provider/model_list', views.Provide.ModelList.as_view()),
path('provider/model_params_form', views.Provide.ModelParamsForm.as_view()),
path('provider/model_form', views.Provide.ModelForm.as_view()),
path('workspace/<str:workspace_id>/model', views.Model.as_view()),
path('workspace/<str:workspace_id>/model/<str:model_id>/model_params_form', views.Model.ModelParamsForm.as_view()),
path('workspace/<str:workspace_id>/model/<str:model_id>', views.Model.Operate.as_view()),
path('workspace/<str:workspace_id>/model/<str:model_id>/pause_download', views.Model.PauseDownload.as_view()),
path('workspace/<str:workspace_id>/model/<str:model_id>/meta', views.Model.ModelMeta.as_view()),
path('workspace/<str:workspace_id>/model', views.ModelSetting.as_view()),
path('workspace/<str:workspace_id>/model/<str:model_id>/model_params_form', views.ModelSetting.ModelParamsForm.as_view()),
path('workspace/<str:workspace_id>/model/<str:model_id>', views.ModelSetting.Operate.as_view()),
path('workspace/<str:workspace_id>/model/<str:model_id>/pause_download', views.ModelSetting.PauseDownload.as_view()),
path('workspace/<str:workspace_id>/model/<str:model_id>/meta', views.ModelSetting.ModelMeta.as_view()),
]
if os.environ.get('SERVER_NAME', 'web') == 'local_model':

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

@ -31,7 +31,6 @@ def encryption_credential(credential):
return credential
def get_edit_model_details(request):
path = request.path
body = request.data
@ -44,6 +43,7 @@ def get_edit_model_details(request):
'query': query
}
def get_model_operation_object(model_id):
model_model = QuerySet(model=Model).filter(id=model_id).first()
if model_model is not None:
@ -53,7 +53,7 @@ def get_model_operation_object(model_id):
return {}
class Model(APIView):
class ModelSetting(APIView):
authentication_classes = [TokenAuth]
@extend_schema(methods=['POST'],

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

@ -28,7 +28,7 @@ class ModelApply(APIView):
responses=DefaultModelResponse.get_response(),
tags=[_('Model')] # type: ignore
)
def post(self, request: Request, model_id):
def post(self, request: Request, workspace_id, model_id):
return result.success(
ModelApplySerializers(data={'model_id': model_id}).embed_documents(request.data))
@ -40,7 +40,7 @@ class ModelApply(APIView):
responses=DefaultModelResponse.get_response(),
tags=[_('Model')] # type: ignore
)
def post(self, request: Request, model_id):
def post(self, request: Request, workspace_id, model_id):
return result.success(
ModelApplySerializers(data={'model_id': model_id}).embed_query(request.data))
@ -52,6 +52,6 @@ class ModelApply(APIView):
responses=DefaultModelResponse.get_response(),
tags=[_('Model')] # type: ignore
)
def post(self, request: Request, model_id):
def post(self, request: Request, workspace_id, model_id):
return result.success(
ModelApplySerializers(data={'model_id': model_id}).compress_documents(request.data))