Update rag_llamaindex.py

rag_llamaindex.py

@@ -9,20 +9,22 @@ from llama_index.vector_stores.mongodb import MongoDBAtlasVectorSearch
 from pathlib import Path
 from pymongo import MongoClient
 
-PDF_URL       = "https://arxiv.org/pdf/2303.08774.pdf"
-WEB_URL       = "https://openai.com/research/gpt-4"
-YOUTUBE_URL_1 = "https://www.youtube.com/watch?v=--khbXchTeE"
-YOUTUBE_URL_2 = "https://www.youtube.com/watch?v=hdhZwyf24mE"
+#PDF_URL       = "https://arxiv.org/pdf/2303.08774.pdf"
+#WEB_URL       = "https://openai.com/research/gpt-4"
+#YOUTUBE_URL_1 = "https://www.youtube.com/watch?v=--khbXchTeE"
+#YOUTUBE_URL_2 = "https://www.youtube.com/watch?v=hdhZwyf24mE"
 
-MONGODB_ATLAS_CLUSTER_URI = os.environ["MONGODB_ATLAS_CLUSTER_URI"]
-MONGODB_DB_NAME           = "llamaindex_db"
-MONGODB_COLLECTION_NAME   = "gpt-4"
-MONGODB_INDEX_NAME        = "default"
+#MONGODB_ATLAS_CLUSTER_URI = os.environ["MONGODB_ATLAS_CLUSTER_URI"]
+#MONGODB_DB_NAME           = "llamaindex_db"
+#MONGODB_COLLECTION_NAME   = "gpt-4"
+#MONGODB_INDEX_NAME        = "default"
 
-logging.basicConfig(stream = sys.stdout, level = logging.INFO)
-logging.getLogger().addHandler(logging.StreamHandler(stream = sys.stdout))
+#logging.basicConfig(stream = sys.stdout, level = logging.INFO)
+#logging.getLogger().addHandler(logging.StreamHandler(stream = sys.stdout))
 
-class LlamaIndexRAG:
+class LlamaIndexRAG(BaseRAG):
+    MONGODB_DB_NAME = "llamaindex_db"
+    
     def load_documents(self):
         docs = []