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

refactor: adjust paragraph positions during migration for consistency

Browse Source
This commit is contained in:
CaptainB 2025-06-27 18:38:44 +08:00
parent 65545a8f15
commit 22d86f046a
1 changed files with 26 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
apps/knowledge/serializers/paragraph.py
View File

@ -489,6 +489,19 @@ class ParagraphSerializers(serializers.Serializer):
target_document_id, target_knowledge_id, None) target_document_id, target_knowledge_id, None)
# 修改段落信息 # 修改段落信息
paragraph_list.update(document_id=target_document_id) paragraph_list.update(document_id=target_document_id)
# 将当前文档中所有段落的位置向下移动，为新段落腾出空间
Paragraph.objects.filter(document_id=target_document_id).exclude(
id__in=paragraph_id_list
).update(position=F('position') + len(paragraph_id_list))
# 重新查询迁移的段落
paragraph_list = Paragraph.objects.filter(
id__in=paragraph_id_list, document_id=target_document_id
)
# 将迁移的段落位置设置为从0开始的序号
for i, paragraph in enumerate(paragraph_list):
paragraph.position = i
paragraph.save()
# 不同数据集迁移 # 不同数据集迁移
else: else:
problem_list = QuerySet(Problem).filter( problem_list = QuerySet(Problem).filter(
@ -522,6 +535,19 @@ class ParagraphSerializers(serializers.Serializer):
pid_list = [paragraph.id for paragraph in paragraph_list] pid_list = [paragraph.id for paragraph in paragraph_list]
# 修改段落信息 # 修改段落信息
paragraph_list.update(knowledge_id=target_knowledge_id, document_id=target_document_id) paragraph_list.update(knowledge_id=target_knowledge_id, document_id=target_document_id)
# 将当前文档中所有段落的位置向下移动，为新段落腾出空间
Paragraph.objects.filter(document_id=target_document_id).exclude(
id__in=pid_list
).update(position=F('position') + len(pid_list))
# 重新查询迁移的段落
paragraph_list = Paragraph.objects.filter(
id__in=pid_list, document_id=target_document_id
)
# 将迁移的段落位置设置为从0开始的序号
for i, paragraph in enumerate(paragraph_list):
paragraph.position = i
paragraph.save()
# 修改向量段落信息 # 修改向量段落信息
update_embedding_document_id(pid_list, target_document_id, target_knowledge_id, embedding_model_id) update_embedding_document_id(pid_list, target_document_id, target_knowledge_id, embedding_model_id)