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

refactor: update dataset_id to knowledge_id in zip_split_handle.py and tools.py

Browse Source
This commit is contained in:
CaptainB 2025-06-18 21:15:53 +08:00
parent 7dc3cb98ef
commit c0b770f41e
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
apps/common/handle/impl/text/zip_split_handle.py
View File

@ -45,8 +45,15 @@ def save_image(image_list):
QuerySet(File).filter(id__in=[i.id for i in image_list]).values('id')] 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 = [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()) save_image_list = list({img.id: img for img in save_image_list}.values())
if len(save_image_list) > 0: print('Saving images:', len(save_image_list))
QuerySet(File).bulk_create(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() default_split_handle = TextSplitHandle()