From 24ad44174cbe34b836b898040edd723de423eac2 Mon Sep 17 00:00:00 2001
From: Gabriele Venturi <lele.venturi@gmail.com>
Date: Wed, 16 Oct 2024 12:06:06 +0200
Subject: [PATCH] test: fix tests with milvus

---
 poetry.lock    | 25 ++++++++++++++++++++++---
 pyproject.toml |  3 ++-
 2 files changed, 24 insertions(+), 4 deletions(-)

diff --git a/poetry.lock b/poetry.lock
index ad9eec033..cc828313b 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -1,4 +1,4 @@
-# This file is automatically @generated by Poetry 1.8.3 and should not be changed by hand.
+# This file is automatically @generated by Poetry 1.8.2 and should not be changed by hand.
 
 [[package]]
 name = "aiohappyeyeballs"
@@ -3786,12 +3786,31 @@ python-versions = ">=3.7"
 files = [
     {file = "milvus_lite-2.4.10-py3-none-macosx_10_9_x86_64.whl", hash = "sha256:fc4246d3ed7d1910847afce0c9ba18212e93a6e9b8406048436940578dfad5cb"},
     {file = "milvus_lite-2.4.10-py3-none-macosx_11_0_arm64.whl", hash = "sha256:74a8e07c5e3b057df17fbb46913388e84df1dc403a200f4e423799a58184c800"},
+    {file = "milvus_lite-2.4.10-py3-none-manylinux2014_aarch64.whl", hash = "sha256:240c7386b747bad696ecb5bd1f58d491e86b9d4b92dccee3315ed7256256eddc"},
     {file = "milvus_lite-2.4.10-py3-none-manylinux2014_x86_64.whl", hash = "sha256:211d2e334a043f9282bdd9755f76b9b2d93b23bffa7af240919ffce6a8dfe325"},
 ]
 
 [package.dependencies]
 tqdm = "*"
 
+[[package]]
+name = "milvus-model"
+version = "0.2.7"
+description = "Model components for PyMilvus, the Python SDK for Milvus"
+optional = true
+python-versions = ">=3.8"
+files = [
+    {file = "milvus_model-0.2.7-py3-none-any.whl", hash = "sha256:c78d87fe9012ccc45218a26338a4d8086e96e938712ad4a983880f7d1163c8c6"},
+    {file = "milvus_model-0.2.7.tar.gz", hash = "sha256:e79f2adfb41b2eb32d09b7206de01453ff09468d939bbfa58ba439ed73b1b31d"},
+]
+
+[package.dependencies]
+numpy = "*"
+onnxruntime = "*"
+protobuf = "*"
+scipy = ">=1.10.0"
+transformers = ">=4.36.0"
+
 [[package]]
 name = "mkdocs"
 version = "1.5.3"
@@ -8681,7 +8700,7 @@ google-sheets = ["beautifulsoup4"]
 ibm-watsonx-ai = ["ibm-watsonx-ai"]
 lancedb = ["lancedb"]
 langchain = ["langchain"]
-milvus = ["pymilvus"]
+milvus = ["milvus-model", "pymilvus"]
 modin = ["modin"]
 numpy = ["numpy"]
 pinecone = ["pinecone-client"]
@@ -8699,4 +8718,4 @@ yfinance = ["yfinance"]
 [metadata]
 lock-version = "2.0"
 python-versions = ">=3.9,<3.9.7 || >3.9.7,<4.0"
-content-hash = "d21c51d44da069b6eca3acef9b565b552e37e8a784aad15515787a98b176a190"
+content-hash = "1b765afee999b83c0fc7f1c56c1a26656d91ca3e3455c9488becdd3e853fc31c"
diff --git a/pyproject.toml b/pyproject.toml
index 597efd400..c0bc971af 100644
--- 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