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={