test: fix tests with milvus

Sinaptik-AI · Oct 16, 2024 · 24ad441 · 24ad441
1 parent 1a9d86b
commit 24ad441
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 4 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -56,6 +56,7 @@ lancedb = {version = "^0.11.0", optional = true}
 sentence-transformers = {version = "^2.3.0", optional = true}
 sqlglot = {extras = ["rs"], version = "^25.0.3"}
 pymilvus = {version = "^2.4.6", optional = true}
+milvus-model = {version = "^0.2.7", optional = true}
 
 [tool.poetry.group.dev]
 optional = true
@@ -98,7 +99,7 @@ flask = ["flask"]
 qdrant = ["qdrant-client"]
 pinecone = ["pinecone-client"]
 ibm-watsonx-ai = ["ibm-watsonx-ai"]
-milvus = ["pymilvus"]
+milvus = ["pymilvus", "milvus-model"]
 
 [tool.poetry.group.docs]
 optional = true