From ffeeb436499d0e442c9bbb3b815c97bc5a8b953f Mon Sep 17 00:00:00 2001 From: Shunping Huang Date: Thu, 4 Dec 2025 10:24:08 -0500 Subject: [PATCH 1/3] Attempt to fix ml workflow failure on embedding tests. --- sdks/python/setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sdks/python/setup.py b/sdks/python/setup.py index 176c84c9966b..111489afd99c 100644 --- a/sdks/python/setup.py +++ b/sdks/python/setup.py @@ -491,7 +491,7 @@ def get_portability_package_data(): 'google-cloud-videointelligence>=2.0,<3', 'google-cloud-vision>=2,<4', 'google-cloud-recommendations-ai>=0.1.0,<0.11.0', - 'google-cloud-aiplatform>=1.26.0, < 2.0', + 'google-cloud-aiplatform>=1.26.0,<1.129.0', 'cloud-sql-python-connector>=1.18.2,<2.0.0', 'python-tds>=1.16.1', 'pg8000>=1.31.5', From 86bbb996a716a49bc5406e11d44cc4f98518bf9b Mon Sep 17 00:00:00 2001 From: Shunping Huang Date: Thu, 4 Dec 2025 14:23:35 -0500 Subject: [PATCH 2/3] Set google-genai upper bound --- sdks/python/setup.py | 1 + 1 file changed, 1 insertion(+) diff --git a/sdks/python/setup.py b/sdks/python/setup.py index 111489afd99c..b1bed3aeb627 100644 --- a/sdks/python/setup.py +++ b/sdks/python/setup.py @@ -492,6 +492,7 @@ def get_portability_package_data(): 'google-cloud-vision>=2,<4', 'google-cloud-recommendations-ai>=0.1.0,<0.11.0', 'google-cloud-aiplatform>=1.26.0,<1.129.0', + 'google-genai<1.53.0', 'cloud-sql-python-connector>=1.18.2,<2.0.0', 'python-tds>=1.16.1', 'pg8000>=1.31.5', From 47de1529481d60204f46adda5744f8e46bcf77f2 Mon Sep 17 00:00:00 2001 From: Shunping Huang Date: Thu, 4 Dec 2025 16:22:23 -0500 Subject: [PATCH 3/3] Pin fsspec. --- sdks/python/setup.py | 1 + 1 file changed, 1 insertion(+) diff --git a/sdks/python/setup.py b/sdks/python/setup.py index b1bed3aeb627..ad913d1784df 100644 --- a/sdks/python/setup.py +++ b/sdks/python/setup.py @@ -465,6 +465,7 @@ def get_portability_package_data(): ] + milvus_dependency, 'gcp': [ 'cachetools>=3.1.0,<7', + 'fsspec==2025.10.0', 'google-api-core>=2.0.0,<3', 'google-apitools>=0.5.31,<0.5.32; python_version < "3.13"', 'google-apitools>=0.5.35; python_version >= "3.13"',