WEBDIENSTE/langgraph
1
0
Fork 0
mirror of https://github.com/langchain-ai/langgraph.git synced 2026-08-22 23:52:23 +02:00
Code Issues Packages Projects Releases Wiki Activity

Default state value to None if channel is empty.

Browse Source
This commit is contained in:
Andrew Nguonly
2024-04-17 15:38:50 -07:00
parent 8a39eb41cd
commit 084ab53252
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
langgraph/pregel/__init__.py
+4 -4
View File
@@ -316,7 +316,7 @@ class Pregel(
)
values = read_channels(channels, self.stream_channels_list)
return StateSnapshot(
values[self.stream_channels]
values.get(self.stream_channels, None)
if isinstance(self.stream_channels, str)
else values,
tuple(name for name, _ in next_tasks),
@@ -337,7 +337,7 @@ class Pregel(
)
values = read_channels(channels, self.stream_channels_list)
return StateSnapshot(
values[self.stream_channels]
values.get(self.stream_channels, None)
if isinstance(self.stream_channels, str)
else values,
tuple(name for name, _ in next_tasks),
@@ -356,7 +356,7 @@ class Pregel(
)
values = read_channels(channels, self.stream_channels_list)
yield StateSnapshot(
values[self.stream_channels]
values.get(self.stream_channels, None)
if isinstance(self.stream_channels, str)
else values,
tuple(name for name, _ in next_tasks),
@@ -378,7 +378,7 @@ class Pregel(
)
values = read_channels(channels, self.stream_channels_list)
yield StateSnapshot(
values[self.stream_channels]
values.get(self.stream_channels, None)
if isinstance(self.stream_channels, str)
else values,
tuple(name for name, _ in next_tasks),