TextNode related typing

run-llama · Feb 15, 2025 · 2b2234c · 2b2234c
1 parent 4e9d83a
commit 2b2234c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/llama-index-core/llama_index/core/evaluation/retrieval/evaluator.py b/llama-index-core/llama_index/core/evaluation/retrieval/evaluator.py
@@ -45,7 +45,7 @@ async def _aget_retrieved_ids_and_texts(
 
         return (
             [node.node.node_id for node in retrieved_nodes],
-            [node.node.text for node in retrieved_nodes],
+            [node.text for node in retrieved_nodes],
         )
 
 
@@ -84,7 +84,7 @@ async def _aget_retrieved_ids_and_texts(
             node = scored_node.node
             if isinstance(node, ImageNode):
                 image_nodes.append(node)
-            if node.text:
+            if isinstance(node, TextNode):
                 text_nodes.append(node)
 
         if mode == "text":