Update base.py

Empty nodes list trying to fetch all nodes from get_llama_nodes method
run-llama · Oct 9, 2024 · fcd067b · fcd067b
1 parent 65946eb
commit fcd067b
Showing 1 changed file with 2 additions and 2 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
@@ -140,13 +140,13 @@ async def async_add_source_text(
 
     def _retrieve(self, query_bundle: QueryBundle) -> List[NodeWithScore]:
         nodes = self.retrieve_from_graph(query_bundle)
-        if self.include_text:
+        if self.include_text and nodes:
             nodes = self.add_source_text(nodes)
         return nodes
 
     async def _aretrieve(self, query_bundle: QueryBundle) -> List[NodeWithScore]:
         nodes = await self.aretrieve_from_graph(query_bundle)
-        if self.include_text:
+        if self.include_text and nodes:
             nodes = await self.async_add_source_text(nodes)
         return nodes