修改GraphRAG缺陷

2024-09-19 11:38:35 +08:00
parent 03586a9b0d
commit 092f7230c1
3 changed files with 3 additions and 3 deletions
@@ -69,7 +69,7 @@ class CustomXinFerenceRerank(XinferenceRerank):
                    rerank_nodes.append(node)
            
            if len(rerank_nodes) > self.top_n:
-                rerank_nodes = sorted(rerank_nodes,key=lambda x:x.score)[:self.top_n]
+                rerank_nodes = sorted(rerank_nodes,key=lambda x:x.score,reverse = True)[:self.top_n]
            event.on_end(payload={EventPayload.NODES: rerank_nodes})
        dispatcher.event(ReRankEndEvent(nodes=rerank_nodes))
        return rerank_nodes