From ddb29df6671b12eccf36d418bc4b7a3c7e1de18c Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Mon, 17 Mar 2025 09:39:23 -0700 Subject: [PATCH] Fix --- libs/langgraph/langgraph/graph/graph.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/libs/langgraph/langgraph/graph/graph.py b/libs/langgraph/langgraph/graph/graph.py index 9592dcfce..df81de4da 100644 --- a/libs/langgraph/langgraph/graph/graph.py +++ b/libs/langgraph/langgraph/graph/graph.py @@ -34,7 +34,6 @@ from langgraph.graph.branch import Branch from langgraph.pregel import Channel, Pregel from langgraph.pregel.protocol import PregelProtocol from langgraph.pregel.read import PregelNode -from langgraph.pregel.remote import RemoteGraph from langgraph.pregel.write import ChannelWrite, ChannelWriteEntry from langgraph.types import All, Checkpointer from langgraph.utils.runnable import RunnableLike, coerce_to_runnable @@ -422,6 +421,8 @@ class CompiledGraph(Pregel): xray: Union[int, bool] = False, ) -> DrawableGraph: """Returns a drawable representation of the computation graph.""" + from langgraph.pregel.remote import RemoteGraph + # gather subgraphs if xray: subpregels: dict[str, PregelProtocol] = { @@ -459,6 +460,8 @@ class CompiledGraph(Pregel): xray: Union[int, bool] = False, ) -> DrawableGraph: """Returns a drawable representation of the computation graph.""" + from langgraph.pregel.remote import RemoteGraph + # gather subgraphs if xray: subgraphs = {