Update rag_llamaindex.py

rag_llamaindex.py

@@ -40,7 +40,7 @@ class LlamaIndexRAG(BaseRAG):
         out_path = out_dir / "gpt-4.pdf"
     
         if not out_path.exists():
-            r = requests.get(PDF_URL)
+            r = requests.get(self.PDF_URL)
             with open(out_path, "wb") as f:
                 f.write(r.content)
 
@@ -50,13 +50,13 @@ class LlamaIndexRAG(BaseRAG):
         # Web
         SimpleWebPageReader = download_loader("SimpleWebPageReader")
         loader = SimpleWebPageReader()
-        docs.extend(loader.load_data(urls = [WEB_URL]))
+        docs.extend(loader.load_data(urls = [self.WEB_URL]))
         #print("docs = " + str(len(docs)))
 
         # YouTube
         loader = YoutubeTranscriptReader()
-        docs.extend(loader.load_data(ytlinks = [YOUTUBE_URL_1,
-                                                YOUTUBE_URL_2]))
+        docs.extend(loader.load_data(ytlinks = [self.YOUTUBE_URL_1,
+                                                self.YOUTUBE_URL_2]))
         #print("docs = " + str(len(docs)))
     
         return docs
@@ -72,10 +72,10 @@ class LlamaIndexRAG(BaseRAG):
 
     def get_vector_store(self):
         return MongoDBAtlasVectorSearch(
-            MongoClient(MONGODB_ATLAS_CLUSTER_URI),
+            MongoClient(self.MONGODB_ATLAS_CLUSTER_URI),
             db_name = MONGODB_DB_NAME,
-            collection_name = MONGODB_COLLECTION_NAME,
-            index_name = MONGODB_INDEX_NAME
+            collection_name = self.MONGODB_COLLECTION_NAME,
+            index_name = self.MONGODB_INDEX_NAME
         )
 
     def ingestion(self, config):