From 489f68c5f6b1b3c956a2aebbde50dfb35c920b86 Mon Sep 17 00:00:00 2001 From: DmytroPaliichuk Date: Thu, 25 Dec 2025 22:39:10 -0800 Subject: [PATCH] fix: update imported lib for query and stream_query The move of langchain.load to langchain_core.load was part of a major architectural --- vertexai/agent_engines/templates/langgraph.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/vertexai/agent_engines/templates/langgraph.py b/vertexai/agent_engines/templates/langgraph.py index 9f5295d81a..aff8b305b0 100644 --- a/vertexai/agent_engines/templates/langgraph.py +++ b/vertexai/agent_engines/templates/langgraph.py @@ -554,7 +554,7 @@ def query( Returns: The output of querying the Agent with the given input and config. """ - from langchain.load import dump as langchain_load_dump + from langchain_core.load import dump as langchain_load_dump if isinstance(input, str): input = {"input": input} @@ -587,7 +587,7 @@ def stream_query( Yields: The output of querying the Agent with the given input and config. """ - from langchain.load import dump as langchain_load_dump + from langchain_core.load import dump as langchain_load_dump if isinstance(input, str): input = {"input": input}