diff --git a/bbconf/_version.py b/bbconf/_version.py index 9e78220..f075dd3 100644 --- a/bbconf/_version.py +++ b/bbconf/_version.py @@ -1 +1 @@ -__version__ = "0.14.0" +__version__ = "0.14.1" diff --git a/bbconf/modules/bedfiles.py b/bbconf/modules/bedfiles.py index 1db3a24..1792725 100644 --- a/bbconf/modules/bedfiles.py +++ b/bbconf/modules/bedfiles.py @@ -2114,7 +2114,6 @@ def reindex_hybrid_search(self, batch: int = 1000, purge: bool = False) -> None: select(Bed) .join(BedMetadata, Bed.id == BedMetadata.id) .where(Bed.indexed == False) - .limit(batch) ) with Session(self._sa_engine) as session: @@ -2196,11 +2195,16 @@ def reindex_hybrid_search(self, batch: int = 1000, purge: bool = False) -> None: operation_info = self.config.qdrant_client.upsert( collection_name=self.config.config.qdrant.hybrid_collection, points=points, + wait=False, ) - session.commit() pbar.write("Uploaded batch to qdrant.") points = [] - assert operation_info.status == "completed" + print(operation_info.status) + assert ( + operation_info.status == "completed" + or operation_info.status == "acknowledged" + ) + session.commit() pbar.write(f"File: {result.id} successfully indexed.") pbar.update(1) @@ -2210,7 +2214,10 @@ def reindex_hybrid_search(self, batch: int = 1000, purge: bool = False) -> None: collection_name=self.config.config.qdrant.hybrid_collection, points=points, ) - assert operation_info.status == "completed" + assert ( + operation_info.status == "completed" + or operation_info.status == "acknowledged" + ) session.commit() return None