From 0422b4fc635f5e06e8f4db8e320ade5f851559f2 Mon Sep 17 00:00:00 2001 From: Derek Higgins Date: Fri, 4 Jul 2025 05:57:23 +0100 Subject: [PATCH] fix: CI flakiness in vector IO tests by pinning pymilvus>=2.4.10 (#2610) 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 https://github.com/meta-llama/llama-stack/issues/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],