From 374ffbbc81bb72823e966ed1fcd845b251dcea85 Mon Sep 17 00:00:00 2001
From: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
Date: Mon, 14 Oct 2024 16:08:38 +0530
Subject: [PATCH] Update llama_index.py

---
 ragged/dataset/llama_index.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/ragged/dataset/llama_index.py b/ragged/dataset/llama_index.py
index 8ead803..a679834 100644
--- 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