Update rag.py

rag.py

@@ -19,7 +19,7 @@ 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=qdd2GZ0DVgc"
+YOUTUBE_URL = "https://www.youtube.com/watch?v=--khbXchTeE"
 
 YOUTUBE_DIR = "/data/youtube"
 CHROMA_DIR  = "/data/chroma"
@@ -71,12 +71,10 @@ def document_storage_mongodb(chunks):
                                             index_name = MONGODB_INDEX_NAME)
 
 def document_retrieval_chroma():
-    print("### Retrieve Chroma")
     return Chroma(embedding_function = OpenAIEmbeddings(disallowed_special = ()),
                   persist_directory = CHROMA_DIR)
 
 def document_retrieval_mongodb():
-    print("### Retrieve MongoDB")
     return MongoDBAtlasVectorSearch.from_connection_string(MONGODB_ATLAS_CLUSTER_URI,
                                                            MONGODB_DB_NAME + "." + MONGODB_COLLECTION_NAME,
                                                            OpenAIEmbeddings(disallowed_special = ()),
@@ -119,6 +117,6 @@ def rag_chain(config, openai_api_key, rag_option, prompt):
                                             verbose = False)
     
     completion = rag_chain({"query": prompt})
-    print(completion)
+    print("###" + completion)
     
     return completion, rag_chain