From 8a740537154aa0ef94b26b872f8073bdd6511cb8 Mon Sep 17 00:00:00 2001
From: Yeesian Ng <ysian@google.com>
Date: Fri, 26 Sep 2025 11:29:14 -0400
Subject: [PATCH 1/2] Update vertex_ai_session_service.py

---
 src/google/adk/sessions/vertex_ai_session_service.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/google/adk/sessions/vertex_ai_session_service.py b/src/google/adk/sessions/vertex_ai_session_service.py
index 29681622d2..57de710bd9 100644
--- a/src/google/adk/sessions/vertex_ai_session_service.py
+++ b/src/google/adk/sessions/vertex_ai_session_service.py
@@ -377,9 +377,9 @@ def _get_api_client(self):
     It needs to be instantiated inside each request so that the event loop
     management can be properly propagated.
     """
-    api_client = genai.Client(
+    api_client = genai.client.BaseApiClient(
         vertexai=True, project=self._project, location=self._location
-    )._api_client
+    )
 
     if new_options := self._api_client_http_options_override():
       api_client._http_options = new_options

From a1adbd70dc92b70ecacaecbd4fd16e6690356a1d Mon Sep 17 00:00:00 2001
From: Yeesian Ng <ysian@google.com>
Date: Fri, 26 Sep 2025 11:44:42 -0400
Subject: [PATCH 2/2] Update pyproject.toml

---
 pyproject.toml | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 2b0a329ace..bc582b5f28 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -38,8 +38,7 @@ dependencies = [
   "google-cloud-spanner>=3.56.0, <4.0.0",                   # For Spanner database
   "google-cloud-speech>=2.30.0, <3.0.0",                    # For Audio Transcription
   "google-cloud-storage>=2.18.0, <3.0.0",                   # For GCS Artifact service
-  # TODO: Modify genai version constraint after the connector/client session warning is fixed.
-  "google-genai>=1.21.1, <1.37.0",                           # Google GenAI SDK
+  "google-genai @ git+https://github.com/googleapis/python-genai.git@copybara/811800162", # For testing purposes
   "graphviz>=0.20.2, <1.0.0",                               # Graphviz for graph rendering
   "mcp>=1.8.0, <2.0.0;python_version>='3.10'",              # For MCP Toolset
   "opentelemetry-api>=1.37.0, <=1.37.0",                    # OpenTelemetry - limit upper version for sdk and api to not risk breaking changes from unstable _logs package.