milvus: always set self._collection (#16306)

* milvus: always set self._collection fix based on discussion from #16165 (comment) * lint + version bump
run-llama · Oct 1, 2024 · e32e8ec · e32e8ec
1 parent 1024574
commit e32e8ec
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
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
@@ -353,6 +353,10 @@ def __init__(
                         "Hybrid retrieval requires Milvus 2.4.0 or later."
                     ) from e
                 self._create_hybrid_index(collection_name)
+        else:
+            self._collection = Collection(
+                collection_name, using=self._milvusclient._using
+            )
 
         # Set properties
         if collection_properties:

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.2.5"
+version = "0.2.6"
 
 [tool.poetry.dependencies]
 python = ">=3.8.1,<4.0"