From 9e0c439dec44b060887db4c9ad9eec62d733bf5d Mon Sep 17 00:00:00 2001
From: Sara Robinson <sararob@google.com>
Date: Tue, 12 May 2026 12:57:45 -0700
Subject: [PATCH] feat: Update google-genai dep to >= 2.0

PiperOrigin-RevId: 914432747
---
 setup.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/setup.py b/setup.py
index e4c3e5e94b..2cca935fe3 100644
--- a/setup.py
+++ b/setup.py
@@ -197,7 +197,7 @@
     "langchain-google-genai >= 4.0.0, < 5.0.0",
     "langgraph >= 1.0.0, < 2.0.0",
     "openinference-instrumentation-langchain >= 0.1.19, < 0.2",
-    "google-genai < 2.0.0",
+    "google-genai >= 2.0.0, < 3.0.0",
 ]
 
 langchain_testing_extra_require = list(
@@ -328,8 +328,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, <3.0.0; python_version<'3.10'",
-        "google-genai >= 1.66.0, <3.0.0; python_version>='3.10'",
+        "google-genai >= 2.0.0, <3.0.0",
     )
     + genai_requires,
     extras_require={