add query bundle to node postprocessor call (#6780)

run-llama · Jul 8, 2023 · 6c6c8ef · 6c6c8ef
1 parent c829422
commit 6c6c8ef
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,10 @@
 # ChangeLog
 
+## Unreleased
+
+### Bug Fixes / Nits
+- fixed passing in query bundle to node postprocessors (#6780)
+
 ## [v0.7.3] - 2023-07-07
 
 ### New Features

diff --git a/llama_index/query_engine/retriever_query_engine.py b/llama_index/query_engine/retriever_query_engine.py
@@ -107,7 +107,9 @@ def retrieve(self, query_bundle: QueryBundle) -> List[NodeWithScore]:
         nodes = self._retriever.retrieve(query_bundle)
 
         for node_postprocessor in self._node_postprocessors:
-            nodes = node_postprocessor.postprocess_nodes(nodes)
+            nodes = node_postprocessor.postprocess_nodes(
+                nodes, query_bundle=query_bundle
+            )
 
         return nodes