From 5a22337d142a1e80c4d207e4d5ced83ec830821e Mon Sep 17 00:00:00 2001 From: "open-swe[bot]" Date: Tue, 29 Jul 2025 17:26:45 +0000 Subject: [PATCH] Apply patch --- libs/langgraph/langgraph/_internal/_runnable.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/libs/langgraph/langgraph/_internal/_runnable.py b/libs/langgraph/langgraph/_internal/_runnable.py index efa0dc825..c6c5ed239 100644 --- a/libs/langgraph/langgraph/_internal/_runnable.py +++ b/libs/langgraph/langgraph/_internal/_runnable.py @@ -61,6 +61,11 @@ try: except ImportError: _StreamingCallbackHandler = None # type: ignore +try: + from langgraph.pregel.protocol import PregelProtocol +except ImportError: + PregelProtocol = None # type: ignore + def _set_config_context( config: RunnableConfig, run: Any = None @@ -896,3 +901,4 @@ async def _consume_aiter(it: AsyncIterator[Any]) -> Any: else: output = chunk return output +