From 32b32959d9b4d4c749b554f109f898dfa458f2d6 Mon Sep 17 00:00:00 2001 From: shaohuzhang1 Date: Tue, 29 Oct 2024 18:49:41 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8D=E7=9F=A5=E8=AF=86?= =?UTF-8?q?=E5=BA=93=E5=AF=BC=E5=87=BA=E6=97=A0=E6=96=87=E6=A1=A3=E7=9F=A5?= =?UTF-8?q?=E8=AF=86=E5=BA=93=E5=AF=BC=E5=87=BA=E6=97=A0=E6=B3=95=E6=89=93?= =?UTF-8?q?=E5=BC=80=E6=96=87=E4=BB=B6?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/dataset/serializers/document_serializers.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/apps/dataset/serializers/document_serializers.py b/apps/dataset/serializers/document_serializers.py index aa61b9469..61a6b02c4 100644 --- a/apps/dataset/serializers/document_serializers.py +++ b/apps/dataset/serializers/document_serializers.py @@ -500,6 +500,8 @@ class DocumentSerializers(ApiMixin, serializers.Serializer): # 创建工作簿对象 workbook = openpyxl.Workbook() workbook.remove_sheet(workbook.active) + if len(data_dict.keys()) == 0: + data_dict['sheet'] = [] for sheet_id in data_dict: # 添加工作表 worksheet = workbook.create_sheet(document_dict.get(sheet_id)) @@ -982,8 +984,6 @@ class DocumentSerializers(ApiMixin, serializers.Serializer): except AlreadyQueued as e: raise AppApiException(500, "任务正在执行中,请勿重复下发") - - class BatchGenerateRelated(ApiMixin, serializers.Serializer): dataset_id = serializers.UUIDField(required=True, error_messages=ErrMessage.uuid("知识库id")) @@ -995,7 +995,8 @@ class DocumentSerializers(ApiMixin, serializers.Serializer): model_id = instance.get("model_id") prompt = instance.get("prompt") for document_id in document_id_list: - DocumentSerializers.GenerateRelated(data={'document_id': document_id}).generate_related(model_id, prompt) + DocumentSerializers.GenerateRelated(data={'document_id': document_id}).generate_related(model_id, + prompt) class FileBufferHandle: