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 c152cc89c7f458df440036dae474eba78b9df36b..1dad8078ab442e90cad7bd84766ef28b14458278 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 0485e688265ca66bfec0ab245792457e725cd5a5..79957eb2c721ef321e26dc5897b0672410cedc37 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"