Fix

Author: Nikos Papailiou

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

@@ -1,7 +1,6 @@
 import numpy as np
 import tiledb
 
-from tiledb.vector_search.ingestion import ingest
 from tiledb.vector_search.module import *
 from tiledb.vector_search.storage_formats import storage_formats
 from typing import Any, Mapping, Optional
@@ -140,6 +139,7 @@ def open_updates_array(self):
         return tiledb.open(self.update_arrays_uri, mode="w")
 
     def consolidate_updates(self):
+        from tiledb.vector_search.ingestion import ingest
         return ingest(
             index_type=self.index_type,
             index_uri=self.uri,

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

@@ -2,6 +2,9 @@
 from functools import partial
 
 from tiledb.cloud.dag import Mode
+from tiledb.vector_search.index import Index
+from tiledb.vector_search.flat_index import FlatIndex
+from tiledb.vector_search.ivf_flat_index import IVFFlatIndex
 from tiledb.vector_search._tiledbvspy import *
 import numpy as np
 
@@ -28,7 +31,7 @@ def ingest(
     verbose: bool = False,
     trace_id: Optional[str] = None,
     mode: Mode = Mode.LOCAL,
-):
+) -> Index:
     """
     Ingest vectors into TileDB.
 
@@ -97,8 +100,6 @@ def ingest(
     from tiledb.cloud.utilities import get_logger
     from tiledb.cloud.utilities import set_aws_context
     from tiledb.vector_search.storage_formats import storage_formats, STORAGE_VERSION
-    from tiledb.vector_search.flat_index import FlatIndex
-    from tiledb.vector_search.ivf_flat_index import IVFFlatIndex
 
     # use index_group_uri for internal clarity
     index_group_uri = index_uri