15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-31 18:22:49 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

fix: Application The reference segmentation of the knowledge base retrieval node in the loop body is displayed as 0 (#4466)

This commit is contained in:
shaohuzhang1 2025-12-08 18:19:09 +08:00 committed by GitHub
parent 29e5f3c382
commit eb41ddf151
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
apps/application/serializers/application_chat_record.py
View File

@ -110,6 +110,16 @@ class ApplicationChatRecordQuerySerializers(serializers.Serializer):
return [ChatRecordSerializerModel(chat_record).data for chat_record in
QuerySet(ChatRecord).filter(chat_id=self.data.get('chat_id')).order_by(order_by)]
@staticmethod
def get_loop_workflow_node(details):
result = []
for item in details.values():
if item.get('type') == 'loop-node':
for loop_item in item.get('loop_node_data') or []:
for inner_item in loop_item.values():
result.append(inner_item)
return result
@staticmethod
def reset_chat_record(chat_record, show_source, show_exec):
knowledge_list = []
@ -119,7 +129,8 @@ class ApplicationChatRecordQuerySerializers(serializers.Serializer):
paragraph_list = chat_record.details.get('search_step').get(
'paragraph_list')
for item in chat_record.details.values():
for item in [*chat_record.details.values(),
*ApplicationChatRecordQuerySerializers.get_loop_workflow_node(chat_record.details)]:
if item.get('type') == 'search-knowledge-node' and item.get('show_knowledge', False):
paragraph_list = paragraph_list + (item.get(
'paragraph_list') or [])