chore: Update Langchain requirements to use langchain-google-genai.

PiperOrigin-RevId: 911377302

Author: vertex-sdk-bot

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",
 ]
 

testing/constraints-langchain.txt

@@ -1,4 +1,4 @@
 langchain
 langchain-core
-langchain-google-vertexai
-langgraph-checkpoint==2.0.1 # Pinned to unbreak unit tests.
+langchain-google-genai
+langgraph-checkpoint

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"

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"

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"

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"