Update custom_utils.py

custom_utils.py

@@ -229,11 +229,11 @@ def get_remove_embedding_stage():
 def invoke_search(db, collection, pipeline):
     results = collection.aggregate(pipeline)
     
-    print(f"Vector search millis elapsed: {get_millis_elapsed(db)}")
+    print(f"Vector search millis elapsed: {get_millis_elapsed(db, collection, pipeline)}")
     
     return list(results)
 
-def get_millis_elapsed(db):
+def get_millis_elapsed(db, collection, pipeline):
     explain_query_execution = db.command(
         "explain", {
             "aggregate": collection.name,