Update rag.py

rag.py

@@ -47,9 +47,9 @@ def document_loading():
     docs.extend(loader.load())
     
     # YouTube
-    loader = GenericLoader(YoutubeAudioLoader([YOUTUBE_URL], YOUTUBE_DIR), 
-                           OpenAIWhisperParser())
-    docs.extend(loader.load())
+    #loader = GenericLoader(YoutubeAudioLoader([YOUTUBE_URL], YOUTUBE_DIR), 
+    #                       OpenAIWhisperParser())
+    #docs.extend(loader.load())
 
     return docs
 
@@ -116,6 +116,7 @@ def rag_chain(config, openai_api_key, rag_option, prompt):
                                             return_source_documents = True,
                                             verbose = False)
     
-    completion = rag_chain({"query": prompt})
+    completion = rag_chain({"query": prompt, "include_run_info": True})
+    print(completion)
     
     return completion, rag_chain