Prevent forced flush() on document add (#11734)

run-llama · Mar 7, 2024 · 12756cc · 12756cc
1 parent 4947327
commit 12756cc
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/...s/vector_stores/llama-index-vector-stores-milvus/llama_index/vector_stores/milvus/base.py b/...s/vector_stores/llama-index-vector-stores-milvus/llama_index/vector_stores/milvus/base.py
@@ -199,7 +199,8 @@ def add(self, nodes: List[BaseNode], **add_kwargs: Any) -> List[str]:
 
         # Insert the data into milvus
         self._collection.insert(insert_list)
-        self._collection.flush()
+        if add_kwargs.get("force_flush", False):
+            self._collection.flush()
         self._create_index_if_required()
         logger.debug(
             f"Successfully inserted embeddings into: {self.collection_name} "

diff --git a/llama-index-integrations/vector_stores/llama-index-vector-stores-milvus/pyproject.toml b/llama-index-integrations/vector_stores/llama-index-vector-stores-milvus/pyproject.toml
@@ -27,7 +27,7 @@ exclude = ["**/BUILD"]
 license = "MIT"
 name = "llama-index-vector-stores-milvus"
 readme = "README.md"
-version = "0.1.5"
+version = "0.1.6"
 
 [tool.poetry.dependencies]
 python = ">=3.8.1,<4.0"