From 12756cccc6c8626e9f4567a4f2202fe32e2b074b Mon Sep 17 00:00:00 2001
From: Michael David <michael.david@yardi.com>
Date: Thu, 7 Mar 2024 10:58:35 -0800
Subject: [PATCH] Prevent forced flush() on document add (#11734)
---
.../llama_index/vector_stores/milvus/base.py | 3 ++-
.../llama-index-vector-stores-milvus/pyproject.toml | 2 +-
2 files changed, 3 insertions(+), 2 deletions(-)
diff --git a/llama-index-integrations/vector_stores/llama-index-vector-stores-milvus/llama_index/vector_stores/milvus/base.py b/llama-index-integrations/vector_stores/llama-index-vector-stores-milvus/llama_index/vector_stores/milvus/base.py
index c152cc89c7..1dad8078ab 100644
--- a/llama-index-integrations/vector_stores/llama-index-vector-stores-milvus/llama_index/vector_stores/milvus/base.py
+++ b/llama-index-integrations/vector_stores/llama-index-vector-stores-milvus/llama_index/vector_stores/milvus/base.py
@@ -199,7 +199,8 @@ class MilvusVectorStore(BasePydanticVectorStore):
# 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
index 0485e68826..79957eb2c7 100644
--- 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"
--
GitLab