Austin/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2024-11-16 11:42:29 +08:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

fix: segment resort in dataset retrieve by index_node_id_to_position (#721)

Browse Source
This commit is contained in:
takatost 2023-08-02 21:31:54 +08:00 committed by GitHub
parent 0d791839e6
commit 654985177f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/core/tool/dataset_retriever_tool.py
View File

@ -109,7 +109,11 @@ class DatasetRetrieverTool(BaseTool):
).all()
if segments:
for segment in segments:
index_node_id_to_position = {id: position for position, id in enumerate(index_node_ids)}
sorted_segments = sorted(segments,
key=lambda segment: index_node_id_to_position.get(segment.index_node_id,
float('inf')))
for segment in sorted_segments:
if segment.answer:
document_context_list.append(f'question:{segment.content} \nanswer:{segment.answer}')
else: