From 32e3c1f6f3acd28ec48b868568d3337b783e56d7 Mon Sep 17 00:00:00 2001
From: Logan <logan.markewich@live.com>
Date: Wed, 10 Apr 2024 10:31:09 -0600
Subject: [PATCH] small milvus bug fix (#12704)
---
.../llama_index/vector_stores/milvus/base.py | 4 +++-
.../llama-index-vector-stores-milvus/pyproject.toml | 2 +-
2 files changed, 4 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 645230b34f..4ae7ed6355 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
@@ -148,7 +148,7 @@ class MilvusVectorStore(BasePydanticVectorStore):
consistency_level=consistency_level,
overwrite=overwrite,
text_key=text_key,
- output_fields=output_fields,
+ output_fields=output_fields or [],
index_config=index_config if index_config else {},
search_config=search_config if search_config else {},
)
@@ -297,6 +297,8 @@ class MilvusVectorStore(BasePydanticVectorStore):
# Limit output fields
if query.output_fields is not None:
output_fields = query.output_fields
+ elif len(self.output_fields) > 0:
+ output_fields = self.output_fields
# Convert to string expression
string_expr = ""
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 427c364340..d5fc147b1d 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.8"
+version = "0.1.9"
[tool.poetry.dependencies]
python = ">=3.8.1,<4.0"
--
GitLab