Update rag.py

rag.py

@@ -17,9 +17,10 @@ from pymongo import MongoClient
 RAG_CHROMA  = "Chroma"
 RAG_MONGODB = "MongoDB"
 
-PDF_URL     = "https://arxiv.org/pdf/2303.08774.pdf"
-WEB_URL     = "https://openai.com/research/gpt-4"
-YOUTUBE_URL = "https://www.youtube.com/watch?v=--khbXchTeE"
+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"
 
 YOUTUBE_DIR = "/data/youtube"
 CHROMA_DIR  = "/data/chroma"
@@ -47,7 +48,7 @@ def document_loading():
     docs.extend(loader.load())
     
     # YouTube
-    loader = GenericLoader(YoutubeAudioLoader([YOUTUBE_URL], YOUTUBE_DIR), 
+    loader = GenericLoader(YoutubeAudioLoader([YOUTUBE_URL_1, YOUTUBE_URL_2], YOUTUBE_DIR), 
                            OpenAIWhisperParser())
     docs.extend(loader.load())