Update custom_utils.py

custom_utils.py

@@ -323,7 +323,7 @@ def get_stage_sorting():
 def invoke_search(db, collection, pipeline):
     results = collection.aggregate(pipeline)
     
-    #print(f"Vector search millis elapsed: {get_millis_elapsed(db, collection, pipeline)}")
+    print(f"Vector search millis elapsed: {get_millis_elapsed(db, collection, pipeline)}")
     
     return list(results)
 
@@ -338,6 +338,8 @@ def get_millis_elapsed(db, collection, pipeline):
 
     explain_vector_search = explain_query_execution["stages"][0]["$vectorSearch"]
 
+    print(explain_vector_search)
+    
     return explain_vector_search["explain"]["collectStats"]["allCollectorStats"]["millisElapsed"]
 
 def get_text_embedding(openai_api_key, text):