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

feat: 分段接口添加认证

Browse Source
This commit is contained in:
shaohuzhang1 2024-08-29 10:40:01 +08:00 committed by shaohuzhang1
parent 57a4f564e7
commit 83c91da117
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
apps/dataset/views/document.py
View File

@ -293,23 +293,24 @@ class Document(APIView):
return result.success(operate.delete()) return result.success(operate.delete())
class SplitPattern(APIView): class SplitPattern(APIView):
authentication_classes = [TokenAuth]
@action(methods=['GET'], detail=False) @action(methods=['GET'], detail=False)
@swagger_auto_schema(operation_summary="获取分段标识列表", @swagger_auto_schema(operation_summary="获取分段标识列表",
operation_id="获取分段标识列表", operation_id="获取分段标识列表",
tags=["知识库/文档"], tags=["知识库/文档"])
security=[])
def get(self, request: Request): def get(self, request: Request):
return result.success(DocumentSerializers.SplitPattern.list()) return result.success(DocumentSerializers.SplitPattern.list())
class Split(APIView): class Split(APIView):
authentication_classes = [TokenAuth]
parser_classes = [MultiPartParser] parser_classes = [MultiPartParser]
@action(methods=['POST'], detail=False) @action(methods=['POST'], detail=False)
@swagger_auto_schema(operation_summary="分段文档", @swagger_auto_schema(operation_summary="分段文档",
operation_id="分段文档", operation_id="分段文档",
manual_parameters=DocumentSerializers.Split.get_request_params_api(), manual_parameters=DocumentSerializers.Split.get_request_params_api(),
tags=["知识库/文档"], tags=["知识库/文档"])
security=[])
def post(self, request: Request): def post(self, request: Request):
split_data = {'file': request.FILES.getlist('file')} split_data = {'file': request.FILES.getlist('file')}
request_data = request.data request_data = request.data