From 83c91da1172a7c9d408a9924f4610e107c383997 Mon Sep 17 00:00:00 2001 From: shaohuzhang1 Date: Thu, 29 Aug 2024 10:40:01 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E5=88=86=E6=AE=B5=E6=8E=A5=E5=8F=A3?= =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E8=AE=A4=E8=AF=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/dataset/views/document.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/apps/dataset/views/document.py b/apps/dataset/views/document.py index dc76655d4..f522d01ce 100644 --- a/apps/dataset/views/document.py +++ b/apps/dataset/views/document.py @@ -293,23 +293,24 @@ class Document(APIView): return result.success(operate.delete()) class SplitPattern(APIView): + authentication_classes = [TokenAuth] + @action(methods=['GET'], detail=False) @swagger_auto_schema(operation_summary="获取分段标识列表", operation_id="获取分段标识列表", - tags=["知识库/文档"], - security=[]) + tags=["知识库/文档"]) def get(self, request: Request): return result.success(DocumentSerializers.SplitPattern.list()) class Split(APIView): + authentication_classes = [TokenAuth] parser_classes = [MultiPartParser] @action(methods=['POST'], detail=False) @swagger_auto_schema(operation_summary="分段文档", operation_id="分段文档", manual_parameters=DocumentSerializers.Split.get_request_params_api(), - tags=["知识库/文档"], - security=[]) + tags=["知识库/文档"]) def post(self, request: Request): split_data = {'file': request.FILES.getlist('file')} request_data = request.data