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

Merge pull request #321 from andrewnguonly/default-state-value

Browse Source 
Default state value (`StateGraph`) to `None` if channel is empty
This commit is contained in:
Nuno Campos
2024-04-17 15:54:38 -07:00
committed by GitHub
parent f8465bfd5e 084ab53252
commit 0d0bd8b430
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),