Update custom_utils.py

custom_utils.py

@@ -186,11 +186,11 @@ def vector_search_naive(openai_api_key,
         }
     }
 
-    remove_embedding_stage =     {
-        "$unset": "description_embedding"
-    }
+    #remove_embedding_stage = {
+    #    "$unset": "description_embedding"
+    #}
     
-    pipeline = [vector_search_stage, remove_embedding_stage]
+    pipeline = [vector_search_stage, get_remove_embedding_stage()]
 
     return invoke_search(collection, pipeline)
 
@@ -223,14 +223,19 @@ def vector_search_advanced(openai_api_key,
         }
     }
 
-    remove_embedding_stage =     {
-        "$unset": "description_embedding"
-    }
+    #remove_embedding_stage = {
+    #    "$unset": "description_embedding"
+    #}
     
-    pipeline = [vector_search_stage, remove_embedding_stage] + additional_stages
+    pipeline = [vector_search_stage, get_remove_embedding_stage()] + additional_stages
 
     return invoke_search(collection, pipeline)
 
+def get_remove_embedding_stage():
+    return {
+        "$unset": "description_embedding"
+    }
+
 def invoke_search(collection, pipeline):
     results = collection.aggregate(pipeline)
     return list(results)