From ed213b072a60661a0c6a4320e9413d7cdcb41646 Mon Sep 17 00:00:00 2001 From: Derek Higgins Date: Thu, 3 Jul 2025 16:33:19 +0100 Subject: [PATCH] fix: CI flakiness in vector IO tests by pinning pymilvus>=2.4.10 This occurred when marshmallow 4.0.0 was installed (which removed __version_info__) By pinning pymilvus to >=2.4.10, we ensure marshmallow doesn't get installed. Also set the dependency in InlineProviderSpec as this is the one that takes effect when using the "inline::milvus" provider. Fixes #2588 Signed-off-by: Derek Higgins --- llama_stack/providers/registry/vector_io.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/llama_stack/providers/registry/vector_io.py b/llama_stack/providers/registry/vector_io.py index de7e08445..c13e65bbc 100644 --- a/llama_stack/providers/registry/vector_io.py +++ b/llama_stack/providers/registry/vector_io.py @@ -520,7 +520,7 @@ Please refer to the inline provider documentation. Api.vector_io, AdapterSpec( adapter_type="milvus", - pip_packages=["pymilvus[marshmallow<3.13.0]"], + pip_packages=["pymilvus>=2.4.10"], module="llama_stack.providers.remote.vector_io.milvus", config_class="llama_stack.providers.remote.vector_io.milvus.MilvusVectorIOConfig", description=""" @@ -633,7 +633,7 @@ For more details on TLS configuration, refer to the [TLS setup guide](https://mi InlineProviderSpec( api=Api.vector_io, provider_type="inline::milvus", - pip_packages=["pymilvus"], + pip_packages=["pymilvus>=2.4.10"], module="llama_stack.providers.inline.vector_io.milvus", config_class="llama_stack.providers.inline.vector_io.milvus.MilvusVectorIOConfig", api_dependencies=[Api.inference],