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

feat: 分段接口添加认证

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())
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