Update rag_llamaindex.py

rag_llamaindex.py

@@ -23,7 +23,7 @@ logging.basicConfig(stream = sys.stdout, level = logging.INFO)
 logging.getLogger().addHandler(logging.StreamHandler(stream = sys.stdout))
 
 class LlamaIndexRAG:
-    def load_documents():
+    def load_documents(self):
         docs = []
     
         # PDF
@@ -58,7 +58,7 @@ class LlamaIndexRAG:
     
         return docs
 
-    def store_documents(config, docs):
+    def store_documents(self, config, docs):
         storage_context = StorageContext.from_defaults(
             vector_store = get_vector_store())
     
@@ -67,7 +67,7 @@ class LlamaIndexRAG:
             storage_context = storage_context
         )
 
-    def get_vector_store():
+    def get_vector_store(self):
         return MongoDBAtlasVectorSearch(
             MongoClient(MONGODB_ATLAS_CLUSTER_URI),
             db_name = MONGODB_DB_NAME,
@@ -75,12 +75,12 @@ class LlamaIndexRAG:
             index_name = MONGODB_INDEX_NAME
         )
 
-    def rag_ingestion_llamaindex(config):
+    def ingestion(self, config):
         docs = load_documents()
     
         store_documents(config, docs)
 
-    def rag_retrieval(config, prompt):
+    def retrieval(self, config, prompt):
         index = VectorStoreIndex.from_vector_store(
             vector_store = get_vector_store())