From c8a605cbc816dbbf40280344f448401725e87eae Mon Sep 17 00:00:00 2001 From: Tat Dat Duong Date: Fri, 14 Mar 2025 01:56:37 +0100 Subject: [PATCH] Update PregelProtocol --- libs/langgraph/langgraph/pregel/protocol.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/libs/langgraph/langgraph/pregel/protocol.py b/libs/langgraph/langgraph/pregel/protocol.py index 9a5f56b76..5a27f417a 100644 --- a/libs/langgraph/langgraph/pregel/protocol.py +++ b/libs/langgraph/langgraph/pregel/protocol.py @@ -73,14 +73,14 @@ class PregelProtocol( def bulk_update_state( self, config: RunnableConfig, - updates: Sequence[StateUpdate], + updates: Sequence[Sequence[StateUpdate]], ) -> RunnableConfig: ... @abstractmethod async def abulk_update_state( self, config: RunnableConfig, - updates: Sequence[StateUpdate], + updates: Sequence[Sequence[StateUpdate]], ) -> RunnableConfig: ... @abstractmethod