From c145803fd22f6a9a798a753b668e320e7561144d Mon Sep 17 00:00:00 2001 From: A Vertex SDK engineer Date: Wed, 6 May 2026 09:18:26 -0700 Subject: [PATCH] chore: Update Langchain requirements to use `langchain-google-genai`. PiperOrigin-RevId: 911377302 --- setup.py | 9 +++++---- testing/constraints-langchain.txt | 4 ++-- .../test_agent_engine_templates_langchain.py | 8 +++++--- .../test_agent_engine_templates_langgraph.py | 4 ++-- .../test_reasoning_engine_templates_langchain.py | 8 +++++--- .../test_reasoning_engine_templates_langgraph.py | 4 ++-- 6 files changed, 21 insertions(+), 16 deletions(-) diff --git a/setup.py b/setup.py index b5ef39736a..de356dcdd4 100644 --- a/setup.py +++ b/setup.py @@ -190,10 +190,11 @@ ] langchain_extra_require = [ - "langchain >= 0.3, < 0.4", - "langchain-core >= 0.3, < 0.4", - "langchain-google-vertexai >= 2.0.22, < 3", - "langgraph >= 0.2.45, < 0.4", + "langchain >= 1.0.0, < 2.0.0", + "langchain-classic", + "langchain-core >= 1.0.0, < 2.0.0", + "langchain-google-genai >= 4.0.0, < 5.0.0", + "langgraph >= 1.0.0, < 2.0.0", "openinference-instrumentation-langchain >= 0.1.19, < 0.2", ] diff --git a/testing/constraints-langchain.txt b/testing/constraints-langchain.txt index 9768def9f3..79ebf5a59f 100644 --- a/testing/constraints-langchain.txt +++ b/testing/constraints-langchain.txt @@ -1,4 +1,4 @@ langchain langchain-core -langchain-google-vertexai -langgraph-checkpoint==2.0.1 # Pinned to unbreak unit tests. \ No newline at end of file +langchain-google-genai +langgraph-checkpoint \ No newline at end of file diff --git a/tests/unit/vertex_langchain/test_agent_engine_templates_langchain.py b/tests/unit/vertex_langchain/test_agent_engine_templates_langchain.py index ef3c99d4eb..46eafd1c34 100644 --- a/tests/unit/vertex_langchain/test_agent_engine_templates_langchain.py +++ b/tests/unit/vertex_langchain/test_agent_engine_templates_langchain.py @@ -27,9 +27,11 @@ from langchain_core import prompts -from langchain.load import dump as langchain_load_dump -from langchain.agents.format_scratchpad import format_to_openai_function_messages -from langchain.tools.base import StructuredTool +from langchain_core.load import dump as langchain_load_dump +from langchain_classic.agents.format_scratchpad import ( + format_to_openai_function_messages, +) +from langchain_core.tools import StructuredTool _DEFAULT_PLACE_TOOL_ACTIVITY = "museums" diff --git a/tests/unit/vertex_langchain/test_agent_engine_templates_langgraph.py b/tests/unit/vertex_langchain/test_agent_engine_templates_langgraph.py index 1048d8f842..8e9dea13e5 100644 --- a/tests/unit/vertex_langchain/test_agent_engine_templates_langgraph.py +++ b/tests/unit/vertex_langchain/test_agent_engine_templates_langgraph.py @@ -24,8 +24,8 @@ import pytest from langchain_core import runnables -from langchain.load import dump as langchain_load_dump -from langchain.tools.base import StructuredTool +from langchain_core.load import dump as langchain_load_dump +from langchain_core.tools import StructuredTool _DEFAULT_PLACE_TOOL_ACTIVITY = "museums" diff --git a/tests/unit/vertex_langchain/test_reasoning_engine_templates_langchain.py b/tests/unit/vertex_langchain/test_reasoning_engine_templates_langchain.py index 2f93b89b82..760b84b76b 100644 --- a/tests/unit/vertex_langchain/test_reasoning_engine_templates_langchain.py +++ b/tests/unit/vertex_langchain/test_reasoning_engine_templates_langchain.py @@ -27,9 +27,11 @@ from langchain_core import prompts -from langchain.load import dump as langchain_load_dump -from langchain.agents.format_scratchpad import format_to_openai_function_messages -from langchain.tools.base import StructuredTool +from langchain_core.load import dump as langchain_load_dump +from langchain_classic.agents.format_scratchpad import ( + format_to_openai_function_messages, +) +from langchain_core.tools import StructuredTool _DEFAULT_PLACE_TOOL_ACTIVITY = "museums" diff --git a/tests/unit/vertex_langchain/test_reasoning_engine_templates_langgraph.py b/tests/unit/vertex_langchain/test_reasoning_engine_templates_langgraph.py index d49d206a15..bd5865d2f4 100644 --- a/tests/unit/vertex_langchain/test_reasoning_engine_templates_langgraph.py +++ b/tests/unit/vertex_langchain/test_reasoning_engine_templates_langgraph.py @@ -24,8 +24,8 @@ import pytest from langchain_core import runnables -from langchain.load import dump as langchain_load_dump -from langchain.tools.base import StructuredTool +from langchain_core.load import dump as langchain_load_dump +from langchain_core.tools import StructuredTool _DEFAULT_PLACE_TOOL_ACTIVITY = "museums"