Fix typo in BasePGRetriever causing graph context to not be added (ru…

…n-llama#16357) fix
jzhao62 · Oct 4, 2024 · c613665 · c613665
1 parent b54b94f
commit c613665
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/llama-index-core/llama_index/core/indices/property_graph/sub_retrievers/base.py b/llama-index-core/llama_index/core/indices/property_graph/sub_retrievers/base.py
@@ -93,10 +93,10 @@ def _add_source_text(
             mapped_node = og_node_map.get(node_with_score.node.ref_doc_id or "", None)
 
             if mapped_node:
-                graph_content = graph_node_map.get(node.node_id, [])
+                graph_content = graph_node_map.get(mapped_node.node_id, [])
                 if len(graph_content) > 0:
                     graph_content_str = "\n".join(graph_content)
-                    cur_content = node.get_content()
+                    cur_content = mapped_node.get_content()
                     preamble_text = (
                         self._include_text_preamble
                         if self._include_text_preamble
@@ -105,7 +105,7 @@ def _add_source_text(
                     new_content = (
                         preamble_text + graph_content_str + "\n\n" + cur_content
                     )
-                    mapped_node = TextNode(**node.dict())
+                    mapped_node = TextNode(**mapped_node.dict())
                     mapped_node.text = new_content
                 result_nodes.append(
                     NodeWithScore(