From 084ab53252ade2f1add8a4e001e59a920f9fc2b7 Mon Sep 17 00:00:00 2001 From: Andrew Nguonly Date: Wed, 17 Apr 2024 15:38:50 -0700 Subject: [PATCH] Default state value to None if channel is empty. --- langgraph/pregel/__init__.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/langgraph/pregel/__init__.py b/langgraph/pregel/__init__.py index 810538fc1..d216f86f6 100644 --- a/langgraph/pregel/__init__.py +++ b/langgraph/pregel/__init__.py @@ -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),