Update custom_utils.py

custom_utils.py

@@ -1,17 +1,16 @@
 import openai, os, time
 
 from datasets import load_dataset
-from pymongo.collection import Collection
-from pymongo.errors import OperationFailure
+#from pymongo.collection import Collection
+#from pymongo.errors import OperationFailure
 from pymongo.mongo_client import MongoClient
-from pymongo.operations import SearchIndexModel
+#from pymongo.operations import SearchIndexModel
 
 DB_NAME = "airbnb_dataset"
 COLLECTION_NAME = "listings_reviews"
 
 def connect_to_database():
     MONGODB_ATLAS_CLUSTER_URI = os.environ["MONGODB_ATLAS_CLUSTER_URI"]
-
     mongo_client = MongoClient(MONGODB_ATLAS_CLUSTER_URI, appname="advanced-rag")
     db = mongo_client.get_database(DB_NAME)
     collection = db.get_collection(COLLECTION_NAME)