fix: 修复知识库导出无文档知识库导出无法打开文件

Author: shaohuzhang1

apps/dataset/serializers/document_serializers.py

@@ -500,6 +500,8 @@ def get_workbook(data_dict, document_dict):
             # 创建工作簿对象
             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 @@ def generate_related(self, model_id, prompt, with_valid=True):
             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 @@ def batch_generate_related(self, instance: Dict, with_valid=True):
             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: