fix multiple retrieval in knowledge node (#8942)

2024-09-30 23:07:04 +08:00
parent 7850a28ec8
commit 625e4c4c72
2 changed files with 5 additions and 3 deletions
--- a/api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py
+++ b/api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py
@@ -229,7 +229,9 @@ class KnowledgeRetrievalNode(BaseNode):
                            source["content"] = segment.get_sign_content()
                        retrieval_resource_list.append(source)
        if retrieval_resource_list:
-            retrieval_resource_list = sorted(retrieval_resource_list, key=lambda x: x.get("score"), reverse=True)
+            retrieval_resource_list = sorted(
+                retrieval_resource_list, key=lambda x: x.get("metadata").get("score"), reverse=True
+            )
            position = 1
            for item in retrieval_resource_list:
                item["metadata"]["position"] = position