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