Add util functions for checking updates existence

Author: Nikos Papailiou

apis/python/src/tiledb/vector_search/index.py

@@ -136,7 +136,7 @@ def query(self, queries: np.ndarray, k, **kwargs):
             raise TypeError(f"A query in queries has {query_dimensions} dimensions, but the indexed data had {self.dimensions} dimensions")
 
         with tiledb.scope_ctx(ctx_or_config=self.config):
-            if not self.group.meta["has_updates"]:
+            if not self.has_updates():
                 if self.query_base_array:
                     return self.query_internal(queries, k, **kwargs)
                 else:
@@ -268,13 +268,19 @@ def get_dimensions(self):
     def query_internal(self, queries: np.ndarray, k, **kwargs):
         raise NotImplementedError
 
-    def update(self, vector: np.array, external_id: np.uint64, timestamp: int = None):
+    def has_updates(self):
+        return self.group.meta["has_updates"]
+
+    def set_has_updates(self, has_updates: bool = True):
         if not self.group.meta["has_updates"]:
             self.group.close()
             self.group = tiledb.Group(self.uri, "w", ctx=tiledb.Ctx(self.config))
-            self.group.meta["has_updates"] = True
+            self.group.meta["has_updates"] = has_updates
             self.group.close()
             self.group = tiledb.Group(self.uri, "r", ctx=tiledb.Ctx(self.config))
+
+    def update(self, vector: np.array, external_id: np.uint64, timestamp: int = None):
+        self.set_has_updates()
         updates_array = self.open_updates_array(timestamp=timestamp)
         vectors = np.empty((1), dtype="O")
         vectors[0] = vector
@@ -285,24 +291,14 @@ def update(self, vector: np.array, external_id: np.uint64, timestamp: int = None
     def update_batch(
         self, vectors: np.ndarray, external_ids: np.array, timestamp: int = None
     ):
-        if not self.group.meta["has_updates"]:
-            self.group.close()
-            self.group = tiledb.Group(self.uri, "w", ctx=tiledb.Ctx(self.config))
-            self.group.meta["has_updates"] = True
-            self.group.close()
-            self.group = tiledb.Group(self.uri, "r", ctx=tiledb.Ctx(self.config))
+        self.set_has_updates()
         updates_array = self.open_updates_array(timestamp=timestamp)
         updates_array[external_ids] = {"vector": vectors}
         updates_array.close()
         self.consolidate_update_fragments()
 
     def delete(self, external_id: np.uint64, timestamp: int = None):
-        if not self.group.meta["has_updates"]:
-            self.group.close()
-            self.group = tiledb.Group(self.uri, "w", ctx=tiledb.Ctx(self.config))
-            self.group.meta["has_updates"] = True
-            self.group.close()
-            self.group = tiledb.Group(self.uri, "r", ctx=tiledb.Ctx(self.config))
+        self.set_has_updates()
         updates_array = self.open_updates_array(timestamp=timestamp)
         deletes = np.empty((1), dtype="O")
         deletes[0] = np.array([], dtype=self.dtype)
@@ -311,12 +307,7 @@ def delete(self, external_id: np.uint64, timestamp: int = None):
         self.consolidate_update_fragments()
 
     def delete_batch(self, external_ids: np.array, timestamp: int = None):
-        if not self.group.meta["has_updates"]:
-            self.group.close()
-            self.group = tiledb.Group(self.uri, "w", ctx=tiledb.Ctx(self.config))
-            self.group.meta["has_updates"] = True
-            self.group.close()
-            self.group = tiledb.Group(self.uri, "r", ctx=tiledb.Ctx(self.config))
+        self.set_has_updates()
         updates_array = self.open_updates_array(timestamp=timestamp)
         deletes = np.empty((len(external_ids)), dtype="O")
         for i in range(len(external_ids)):