From 68a5c3f4c7b79d21ae4fa6b76a2f48fc721e8e84 Mon Sep 17 00:00:00 2001 From: Tat Dat Duong Date: Fri, 7 Mar 2025 20:13:18 +0100 Subject: [PATCH] Implement batch events for RemotePregel --- libs/langgraph/langgraph/pregel/remote.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/libs/langgraph/langgraph/pregel/remote.py b/libs/langgraph/langgraph/pregel/remote.py index 6ba547b14..64fb4edb6 100644 --- a/libs/langgraph/langgraph/pregel/remote.py +++ b/libs/langgraph/langgraph/pregel/remote.py @@ -457,6 +457,20 @@ class RemoteGraph(PregelProtocol): for state in states: yield self._create_state_snapshot(state) + def bulk_update_state( + self, + config: RunnableConfig, + updates: list[tuple[Optional[dict[str, Any]], Optional[str]]], + ) -> RunnableConfig: + raise NotImplementedError + + async def abulk_update_state( + self, + config: RunnableConfig, + updates: list[tuple[Optional[dict[str, Any]], Optional[str]]], + ) -> RunnableConfig: + raise NotImplementedError + def update_state( self, config: RunnableConfig,