Update rag_llamaindex.py

rag_llamaindex.py

@@ -60,7 +60,7 @@ class LlamaIndexRAG:
 
     def store_documents(self, config, docs):
         storage_context = StorageContext.from_defaults(
-            vector_store = get_vector_store())
+            vector_store = self.get_vector_store())
     
         VectorStoreIndex.from_documents(
             docs,
@@ -76,13 +76,13 @@ class LlamaIndexRAG:
         )
 
     def ingestion(self, config):
-        docs = load_documents()
+        docs = self.load_documents()
     
-        store_documents(config, docs)
+        self.store_documents(config, docs)
 
     def retrieval(self, config, prompt):
         index = VectorStoreIndex.from_vector_store(
-            vector_store = get_vector_store(self))
+            vector_store = self.get_vector_store())
 
         query_engine = index.as_query_engine(
             similarity_top_k = config["k"]