From c0b770f41ea75614b339cac2686a43767b33f450 Mon Sep 17 00:00:00 2001 From: CaptainB Date: Wed, 18 Jun 2025 21:15:53 +0800 Subject: [PATCH] refactor: update dataset_id to knowledge_id in zip_split_handle.py and tools.py --- apps/common/handle/impl/text/zip_split_handle.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/apps/common/handle/impl/text/zip_split_handle.py b/apps/common/handle/impl/text/zip_split_handle.py index b7d8857ea..c01f435c3 100644 --- a/apps/common/handle/impl/text/zip_split_handle.py +++ b/apps/common/handle/impl/text/zip_split_handle.py @@ -45,8 +45,15 @@ def save_image(image_list): QuerySet(File).filter(id__in=[i.id for i in image_list]).values('id')] save_image_list = [image for image in image_list if not exist_image_list.__contains__(str(image.id))] save_image_list = list({img.id: img for img in save_image_list}.values()) - if len(save_image_list) > 0: - QuerySet(File).bulk_create(save_image_list) + print('Saving images:', len(save_image_list)) + # for file in save_image_list: + # file_bytes = file.meta.pop('content') + # file.meta['knowledge_id'] = self.data.get('knowledge_id') + # file.source_type = FileSourceType.KNOWLEDGE + # file.source_id = self.data.get('knowledge_id') + # file.save(file_bytes) + # if len(save_image_list) > 0: + # QuerySet(File).bulk_create(save_image_list) default_split_handle = TextSplitHandle()