15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-29 07:52:50 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

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

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')]
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()