feat: Upgrade google-genai dependency required version to >=2.0.0

PiperOrigin-RevId: 908716648

Author: sararob

setup.py

@@ -326,8 +326,7 @@
         "google-cloud-storage >= 3.10.0, < 4.0.0; python_version>='3.13'",
         "google-cloud-bigquery >= 1.15.0, < 4.0.0, !=3.20.0",
         "google-cloud-resource-manager >= 1.3.3, < 3.0.0",
-        "google-genai >= 1.37.0, <2.0.0; python_version<'3.10'",
-        "google-genai >= 1.66.0, <2.0.0; python_version>='3.10'",
+        "google-genai >= 2.0.0, < 3.0.0",
     )
     + genai_requires,
     extras_require={

testing/constraints-3.10.txt

@@ -2,7 +2,7 @@
 # This constraints file is required for unit tests.
 # List all library dependencies and extras in this file.
 google-api-core==2.21.0 # Tests google-api-core with rest async support
-google-auth==2.47.0 # Tests google-auth with rest async support
+google-auth==2.49.0 # Updated to satisfy google-genai 2.0.0 requirement
 proto-plus==1.22.3
 protobuf
 mock==4.0.2

testing/constraints-3.11.txt

@@ -2,7 +2,7 @@
 # This constraints file is required for unit tests.
 # List all library dependencies and extras in this file.
 google-api-core==2.21.0 # Tests google-api-core with rest async support
-google-auth==2.47.0 # Tests google-auth with rest async support
+google-auth==2.49.0 # Updated to satisfy google-genai 2.0.0 requirement
 proto-plus
 protobuf
 mock==4.0.2

testing/constraints-3.12.txt

@@ -2,7 +2,7 @@
 # This constraints file is required for unit tests.
 # List all library dependencies and extras in this file.
 google-api-core==2.21.0 # Tests google-api-core with rest async support
-google-auth==2.47.0 # Tests google-auth with rest async support
+google-auth==2.49.0 # Updated to satisfy google-genai 2.0.0 requirement
 proto-plus
 mock==4.0.2
 google-cloud-storage==3.0.0 # Updated to v3.x, backward compatible with v2.x via wrapper

testing/constraints-3.13.txt

@@ -2,7 +2,7 @@
 # This constraints file is required for unit tests.
 # List all library dependencies and extras in this file.
 google-api-core==2.27.0 # Tests google-api-core with rest async support
-google-auth==2.47.0 # Tests google-auth with rest async support
+google-auth==2.49.0 # Updated to satisfy google-genai 2.0.0 requirement
 proto-plus
 mock==4.0.2
 google-cloud-storage==3.10.0 # Updated to v3.x, backward compatible with v2.x via wrapper

testing/constraints-3.14.txt

@@ -2,7 +2,7 @@
 # This constraints file is required for unit tests.
 # List all library dependencies and extras in this file.
 google-api-core==2.27.0
-google-auth==2.47.0 # Tests google-auth with rest async support
+google-auth==2.49.0 # Updated to satisfy google-genai 2.0.0 requirement
 proto-plus
 mock==4.0.2
 google-cloud-storage==3.10.0 # Updated for Python 3.14 compatibility

tests/unit/vertexai/genai/replays/conftest.py

@@ -17,6 +17,7 @@
 """Conftest for Vertex SDK GenAI tests."""
 
 import os
+import sys
 from unittest import mock
 
 from vertexai._genai import (
@@ -31,6 +32,7 @@
 from google.genai import _replay_api_client
 from google.genai import types as genai_types
 from google.genai import client as google_genai_client_module
+from google.genai import version as genai_version
 from vertexai._genai import _gcs_utils
 from vertexai._genai import prompt_optimizer
 import pytest
@@ -260,12 +262,20 @@ def client(use_vertex, replays_prefix, http_options, request):
         )
         os.environ["GOOGLE_GENAI_REPLAYS_DIRECTORY"] = replays_root_directory
 
+    genai_sdk_version = genai_version.__version__
+    module_version = aip_version.__version__
+    python_version = sys.version.split()[0]
+
+    tracking_label = (
+        f"google-genai-sdk/{genai_sdk_version}+{_GENAI_MODULES_TELEMETRY_HEADER}/"
+        f"{module_version} gl-python/{python_version}"
+    )
+
     if http_options is None:
         http_options = genai_types.HttpOptions()
     if http_options.headers is None:
         http_options.headers = {}
 
-    tracking_label = f"{_GENAI_MODULES_TELEMETRY_HEADER}/{aip_version.__version__}"
     http_options.headers["user-agent"] = tracking_label
     http_options.headers["x-goog-api-client"] = tracking_label
 

vertexai/_genai/client.py

@@ -15,6 +15,7 @@
 
 import asyncio
 import importlib
+import sys
 from typing import Optional, Union, TYPE_CHECKING
 from types import TracebackType, ModuleType
 
@@ -23,6 +24,7 @@
 from google.genai import _common
 from google.genai import client as genai_client
 from google.genai import types
+from google.genai import version as genai_version
 from . import live
 
 if TYPE_CHECKING:
@@ -219,21 +221,25 @@ def __init__(
         if http_options.headers is None:
             http_options.headers = {}
 
-        tracking_label = f"{_GENAI_MODULES_TELEMETRY_HEADER}/{aip_version.__version__}"
+        genai_sdk_version = genai_version.__version__
+        module_version = aip_version.__version__
+        python_version = sys.version.split()[0]
+
+        agent_platform_header = f"google-genai-sdk/{genai_sdk_version}+{_GENAI_MODULES_TELEMETRY_HEADER}/{module_version} gl-python/{python_version}"
 
         if "user-agent" in http_options.headers:
             http_options.headers["user-agent"] = (
-                f"{http_options.headers['user-agent']} {tracking_label}"
+                f"{agent_platform_header} " + http_options.headers["user-agent"]
             )
         else:
-            http_options.headers["user-agent"] = tracking_label
+            http_options.headers["user-agent"] = agent_platform_header
 
         if "x-goog-api-client" in http_options.headers:
             http_options.headers["x-goog-api-client"] = (
-                f"{http_options.headers['x-goog-api-client']} {tracking_label}"
+                f"{agent_platform_header} " + http_options.headers["x-goog-api-client"]
             )
         else:
-            http_options.headers["x-goog-api-client"] = tracking_label
+            http_options.headers["x-goog-api-client"] = agent_platform_header
 
         self._api_client = genai_client.Client._get_api_client(
             vertexai=True,