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

fix: 修复删除知识库的段落后，日志内容还是显示标注的缺陷

Browse Source
This commit is contained in:
wxg0103 2024-11-26 17:37:10 +08:00 committed by wxg
parent f1083d973e
commit de630b4595
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -395,7 +395,8 @@ class ChatRecordSerializerModel(serializers.ModelSerializer):
class Meta: class Meta:
model = ChatRecord model = ChatRecord
fields = ['id', 'chat_id', 'vote_status', 'problem_text', 'answer_text', fields = ['id', 'chat_id', 'vote_status', 'problem_text', 'answer_text',
'message_tokens', 'answer_tokens', 'const', 'improve_paragraph_id_list', 'run_time', 'index','answer_text_list', 'message_tokens', 'answer_tokens', 'const', 'improve_paragraph_id_list', 'run_time', 'index',
'answer_text_list',
'create_time', 'update_time'] 'create_time', 'update_time']
@ -457,6 +458,7 @@ class ChatRecordSerializer(serializers.Serializer):
def reset_chat_record(chat_record): def reset_chat_record(chat_record):
dataset_list = [] dataset_list = []
paragraph_list = [] paragraph_list = []
if 'search_step' in chat_record.details and chat_record.details.get('search_step').get( if 'search_step' in chat_record.details and chat_record.details.get('search_step').get(
'paragraph_list') is not None: 'paragraph_list') is not None:
paragraph_list = chat_record.details.get('search_step').get( paragraph_list = chat_record.details.get('search_step').get(
@ -468,6 +470,14 @@ class ChatRecordSerializer(serializers.Serializer):
row in row in
paragraph_list], paragraph_list],
{}).items()] {}).items()]
if len(chat_record.improve_paragraph_id_list) > 0:
paragraph_model_list = QuerySet(Paragraph).filter(id__in=chat_record.improve_paragraph_id_list)
if len(paragraph_model_list) < len(chat_record.improve_paragraph_id_list):
paragraph_model_id_list = [str(p.id) for p in paragraph_model_list]
chat_record.improve_paragraph_id_list = list(
filter(lambda p_id: paragraph_model_id_list.__contains__(p_id),
chat_record.improve_paragraph_id_list))
chat_record.save()
return { return {
**ChatRecordSerializerModel(chat_record).data, **ChatRecordSerializerModel(chat_record).data,