Update llama_index.py

lancedb · Oct 14, 2024 · 374ffbb · 374ffbb
1 parent c3b875f
commit 374ffbb
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ragged/dataset/llama_index.py b/ragged/dataset/llama_index.py
@@ -40,6 +40,7 @@ def __init__(self, dataset_name: Optional[str] = None, path: Optional[str] = Non
 
         parser = SentenceSplitter()
         nodes = parser.get_nodes_from_documents(documents)
+        self.nodes = nodes
         self.documents = [TextNode(id=node.id_, text=node.text) for node in nodes]
         self.context_column = context_column_name
         self.query_column = query_column_name