Update custom_utils.py

custom_utils.py

@@ -186,10 +186,6 @@ def vector_search_naive(openai_api_key,
         }
     }
 
-    #remove_embedding_stage = {
-    #    "$unset": "description_embedding"
-    #}
-    
     pipeline = [vector_search_stage, get_remove_embedding_stage()]
 
     return invoke_search(collection, pipeline)
@@ -222,11 +218,7 @@ def vector_search_advanced(openai_api_key,
             },
         }
     }
-
-    #remove_embedding_stage = {
-    #    "$unset": "description_embedding"
-    #}
-    
+  
     pipeline = [vector_search_stage, get_remove_embedding_stage()] + additional_stages
 
     return invoke_search(collection, pipeline)
@@ -238,8 +230,24 @@ def get_remove_embedding_stage():
 
 def invoke_search(collection, pipeline):
     results = collection.aggregate(pipeline)
+    
+    print(f"MongoDB execution time: {get_millis_elapsed()} millis")
+    
     return list(results)
 
+def get_millis_elapsed()
+    explain_query_execution = db.command(
+        "explain", {
+            "aggregate": collection.name,
+            "pipeline": pipeline,
+            "cursor": {}
+        }, 
+        verbosity='executionStats')
+
+    explain_vector_search = explain_query_execution["stages"][0]["$vectorSearch"]
+    
+    return explain_vector_search["explain"]["collectStats"]["millisElapsed"]
+
 def get_text_embedding(openai_api_key, text):
     if not text or not isinstance(text, str):
         return None