2024-08-22 09:41:14 +08:00
1 changed files with 2 additions and 2 deletions
@@ -224,8 +224,8 @@ class XinferenceRerank(BaseNodePostprocessor):

                if self.top_n is not None:
                    if len(new_nodes) > self.top_n:
-                        for index in new_nodes[5:-1]:
-                            new_nodes.remove(-1)
+                        for index in new_nodes[self.top_n:-1]:
+                            new_nodes.remove(index)

            event.on_end(payload={EventPayload.NODES: new_nodes})