Skip to content

Commit c0b770f

Browse files
committed
refactor: update dataset_id to knowledge_id in zip_split_handle.py and tools.py
1 parent 7dc3cb9 commit c0b770f

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

apps/common/handle/impl/text/zip_split_handle.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -45,8 +45,15 @@ def save_image(image_list):
4545
QuerySet(File).filter(id__in=[i.id for i in image_list]).values('id')]
4646
save_image_list = [image for image in image_list if not exist_image_list.__contains__(str(image.id))]
4747
save_image_list = list({img.id: img for img in save_image_list}.values())
48-
if len(save_image_list) > 0:
49-
QuerySet(File).bulk_create(save_image_list)
48+
print('Saving images:', len(save_image_list))
49+
# for file in save_image_list:
50+
# file_bytes = file.meta.pop('content')
51+
# file.meta['knowledge_id'] = self.data.get('knowledge_id')
52+
# file.source_type = FileSourceType.KNOWLEDGE
53+
# file.source_id = self.data.get('knowledge_id')
54+
# file.save(file_bytes)
55+
# if len(save_image_list) > 0:
56+
# QuerySet(File).bulk_create(save_image_list)
5057

5158

5259
default_split_handle = TextSplitHandle()

0 commit comments

Comments
 (0)