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

fix(知识库): 修复知识库字符未更新的缺陷

Browse Source 
--bug=1049232 --user=王孝刚 【应用】对话日志-添加到知识库后，文档字符数未同步更新 https://www.tapd.cn/57709429/s/1614059
This commit is contained in:
wxg0103 2024-11-21 11:35:49 +08:00 committed by wxg
parent aba33eb00a
commit 4badd3ab6e
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
apps/application/serializers/chat_serializers.py
View File

@ -38,7 +38,7 @@ from common.util.field_message import ErrMessage
from common.util.file_util import get_file_content from common.util.file_util import get_file_content
from common.util.lock import try_lock, un_lock from common.util.lock import try_lock, un_lock
from dataset.models import Document, Problem, Paragraph, ProblemParagraphMapping from dataset.models import Document, Problem, Paragraph, ProblemParagraphMapping
from dataset.serializers.common_serializers import get_embedding_model_id_by_dataset_id from dataset.serializers.common_serializers import get_embedding_model_id_by_dataset_id, update_document_char_length
from dataset.serializers.paragraph_serializers import ParagraphSerializers from dataset.serializers.paragraph_serializers import ParagraphSerializers
from embedding.task import embedding_by_paragraph, embedding_by_paragraph_list from embedding.task import embedding_by_paragraph, embedding_by_paragraph_list
from setting.models import Model from setting.models import Model
@ -620,6 +620,7 @@ class ChatRecordSerializer(serializers.Serializer):
# 插入关联问题 # 插入关联问题
problem_paragraph_mapping.save() problem_paragraph_mapping.save()
chat_record.improve_paragraph_id_list.append(paragraph.id) chat_record.improve_paragraph_id_list.append(paragraph.id)
update_document_char_length(document_id)
# 添加标注 # 添加标注
chat_record.save() chat_record.save()
return ChatRecordSerializerModel(chat_record).data, paragraph.id, dataset_id return ChatRecordSerializerModel(chat_record).data, paragraph.id, dataset_id
@ -718,5 +719,6 @@ class ChatRecordSerializer(serializers.Serializer):
# 批量保存聊天记录 # 批量保存聊天记录
ChatRecord.objects.bulk_update(chat_record_list, ['improve_paragraph_id_list']) ChatRecord.objects.bulk_update(chat_record_list, ['improve_paragraph_id_list'])
update_document_char_length(document_id)
return paragraph_ids, dataset_id return paragraph_ids, dataset_id

2
ui/src/components/ai-chat/ExecutionDetailDialog.vue
View File

@ -175,7 +175,7 @@
</div> </div>
<div class="card-never border-r-4 mt-8"> <div class="card-never border-r-4 mt-8">
<h5 class="p-8-12"> <h5 class="p-8-12">
{{ item.type == WorkflowType.Application ? '应用回答' : 'AI 回答' }} {{ item.type == WorkflowType.Application ? '参数输出' : 'AI 回答' }}
</h5> </h5>
<div class="p-8-12 border-t-dashed lighter"> <div class="p-8-12 border-t-dashed lighter">
<MdPreview <MdPreview