From 576fa6db20d8997df0c6c9f356a556eb31109e16 Mon Sep 17 00:00:00 2001 From: Sydney Runkle Date: Thu, 12 Mar 2026 11:17:20 -0400 Subject: [PATCH] switch generics --- libs/langgraph/langgraph/pregel/main.py | 8 ++++---- libs/langgraph/langgraph/pregel/protocol.py | 4 ++-- libs/langgraph/langgraph/pregel/remote.py | 4 ++-- libs/langgraph/langgraph/types.py | 2 +- 4 files changed, 9 insertions(+), 9 deletions(-) diff --git a/libs/langgraph/langgraph/pregel/main.py b/libs/langgraph/langgraph/pregel/main.py index d4190a0e6..3ecefc1dc 100644 --- a/libs/langgraph/langgraph/pregel/main.py +++ b/libs/langgraph/langgraph/pregel/main.py @@ -2456,7 +2456,7 @@ class Pregel( debug: bool | None = None, version: Literal["v2"], **kwargs: Unpack[DeprecatedKwargs], - ) -> Iterator[StreamPart[OutputT, StateT]]: ... + ) -> Iterator[StreamPart[StateT, OutputT]]: ... @overload def stream( @@ -2787,7 +2787,7 @@ class Pregel( debug: bool | None = None, version: Literal["v2"], **kwargs: Unpack[DeprecatedKwargs], - ) -> AsyncIterator[StreamPart[OutputT, StateT]]: ... + ) -> AsyncIterator[StreamPart[StateT, OutputT]]: ... @overload def astream( @@ -3194,7 +3194,7 @@ class Pregel( durability: Durability | None = None, version: Literal["v2"], **kwargs: Any, - ) -> list[StreamPart[OutputT, StateT]]: ... + ) -> list[StreamPart[StateT, OutputT]]: ... @overload def invoke( @@ -3364,7 +3364,7 @@ class Pregel( durability: Durability | None = None, version: Literal["v2"], **kwargs: Any, - ) -> list[StreamPart[OutputT, StateT]]: ... + ) -> list[StreamPart[StateT, OutputT]]: ... @overload async def ainvoke( diff --git a/libs/langgraph/langgraph/pregel/protocol.py b/libs/langgraph/langgraph/pregel/protocol.py index e5e957f50..259554ddb 100644 --- a/libs/langgraph/langgraph/pregel/protocol.py +++ b/libs/langgraph/langgraph/pregel/protocol.py @@ -117,7 +117,7 @@ class PregelProtocol(Runnable[InputT, Any], Generic[StateT, ContextT, InputT, Ou interrupt_after: All | Sequence[str] | None = None, subgraphs: bool = False, version: Literal["v2"], - ) -> Iterator[StreamPart[OutputT, StateT]]: ... + ) -> Iterator[StreamPart[StateT, OutputT]]: ... @overload @abstractmethod @@ -161,7 +161,7 @@ class PregelProtocol(Runnable[InputT, Any], Generic[StateT, ContextT, InputT, Ou interrupt_after: All | Sequence[str] | None = None, subgraphs: bool = False, version: Literal["v2"], - ) -> AsyncIterator[StreamPart[OutputT, StateT]]: ... + ) -> AsyncIterator[StreamPart[StateT, OutputT]]: ... @overload @abstractmethod diff --git a/libs/langgraph/langgraph/pregel/remote.py b/libs/langgraph/langgraph/pregel/remote.py index db2860ab6..16241a7a7 100644 --- a/libs/langgraph/langgraph/pregel/remote.py +++ b/libs/langgraph/langgraph/pregel/remote.py @@ -705,7 +705,7 @@ class RemoteGraph( params: QueryParamTypes | None = None, version: Literal["v2"], **kwargs: Any, - ) -> Iterator[StreamPart[OutputT, StateT]]: ... + ) -> Iterator[StreamPart[StateT, OutputT]]: ... @overload def stream( @@ -860,7 +860,7 @@ class RemoteGraph( params: QueryParamTypes | None = None, version: Literal["v2"], **kwargs: Any, - ) -> AsyncIterator[StreamPart[OutputT, StateT]]: ... + ) -> AsyncIterator[StreamPart[StateT, OutputT]]: ... @overload def astream( diff --git a/libs/langgraph/langgraph/types.py b/libs/langgraph/langgraph/types.py index 547513498..d04d82da7 100644 --- a/libs/langgraph/langgraph/types.py +++ b/libs/langgraph/langgraph/types.py @@ -335,7 +335,7 @@ StreamPart = TypeAliasType( | CheckpointStreamPart[StateT] | TasksStreamPart | DebugStreamPart[StateT], - type_params=(OutputT, StateT), + type_params=(StateT, OutputT), ) """A discriminated union of all v2 stream part types.