From 1f745ca01715db8e0db779e55d9dc7962f9455fe Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Thu, 5 Dec 2024 13:50:42 -0800 Subject: [PATCH] Lint --- libs/langgraph/langgraph/graph/state.py | 4 ++-- libs/langgraph/langgraph/pregel/write.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/libs/langgraph/langgraph/graph/state.py b/libs/langgraph/langgraph/graph/state.py index 0462008c8..6e37c6150 100644 --- a/libs/langgraph/langgraph/graph/state.py +++ b/libs/langgraph/langgraph/graph/state.py @@ -667,7 +667,7 @@ class CompiledStateGraph(CompiledGraph): raise InvalidUpdateError(msg) # state updaters - write_entries = [ + write_entries: list[Union[ChannelWriteEntry, ChannelWriteTupleEntry]] = [ ChannelWriteTupleEntry( mapper=_get_root if output_keys == ["__root__"] else _get_updates ) @@ -706,7 +706,7 @@ class CompiledStateGraph(CompiledGraph): writers=[ # publish to this channel and state keys ChannelWrite( - [ChannelWriteEntry(key, key)] + write_entries, + write_entries + [ChannelWriteEntry(key, key)], tags=[TAG_HIDDEN], ), ], diff --git a/libs/langgraph/langgraph/pregel/write.py b/libs/langgraph/langgraph/pregel/write.py index baee23ab2..8e9cf35b3 100644 --- a/libs/langgraph/langgraph/pregel/write.py +++ b/libs/langgraph/langgraph/pregel/write.py @@ -37,7 +37,7 @@ class ChannelWriteEntry(NamedTuple): class ChannelWriteTupleEntry(NamedTuple): - mapper: Callable[[Any], Sequence[tuple[str, Any]]] + mapper: Callable[[Any], Optional[Sequence[tuple[str, Any]]]] """Function to extract tuples from value.""" value: Any = PASSTHROUGH """Value to write, or PASSTHROUGH to use the input."""