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

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

Browse Source
This commit is contained in:
shaohuzhang1 2024-10-29 18:49:41 +08:00 committed by shaohuzhang1
parent 6ad9bd83a2
commit 32b32959d9
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
apps/dataset/serializers/document_serializers.py
View File

@ -500,6 +500,8 @@ class DocumentSerializers(ApiMixin, serializers.Serializer):
# 创建工作簿对象 # 创建工作簿对象
workbook = openpyxl.Workbook() workbook = openpyxl.Workbook()
workbook.remove_sheet(workbook.active) workbook.remove_sheet(workbook.active)
if len(data_dict.keys()) == 0:
data_dict['sheet'] = []
for sheet_id in data_dict: for sheet_id in data_dict:
# 添加工作表 # 添加工作表
worksheet = workbook.create_sheet(document_dict.get(sheet_id)) worksheet = workbook.create_sheet(document_dict.get(sheet_id))
@ -982,8 +984,6 @@ class DocumentSerializers(ApiMixin, serializers.Serializer):
except AlreadyQueued as e: except AlreadyQueued as e:
raise AppApiException(500, "任务正在执行中,请勿重复下发") raise AppApiException(500, "任务正在执行中,请勿重复下发")
class BatchGenerateRelated(ApiMixin, serializers.Serializer): class BatchGenerateRelated(ApiMixin, serializers.Serializer):
dataset_id = serializers.UUIDField(required=True, error_messages=ErrMessage.uuid("知识库id")) 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") model_id = instance.get("model_id")
prompt = instance.get("prompt") prompt = instance.get("prompt")
for document_id in document_id_list: 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: class FileBufferHandle: