From 1ce8d628f248b0bf90c5d8b4f2c02602cfd189af Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Mon, 9 Sep 2024 09:17:36 -0700 Subject: [PATCH] In update state skip nodes not found in instance eg, because removed since --- libs/langgraph/langgraph/pregel/__init__.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/libs/langgraph/langgraph/pregel/__init__.py b/libs/langgraph/langgraph/pregel/__init__.py index 8c7a05e13..d1229d6e9 100644 --- a/libs/langgraph/langgraph/pregel/__init__.py +++ b/libs/langgraph/langgraph/pregel/__init__.py @@ -782,6 +782,7 @@ class Pregel(Runnable[Union[dict[str, Any], Any], Union[dict[str, Any], Any]]): last_seen_by_node = sorted( (v, n) for n, seen in checkpoint["versions_seen"].items() + if n in self.nodes for v in seen.values() ) # if two nodes updated the state at the same time, it's ambiguous @@ -939,6 +940,7 @@ class Pregel(Runnable[Union[dict[str, Any], Any], Union[dict[str, Any], Any]]): last_seen_by_node = sorted( (v, n) for n, seen in checkpoint["versions_seen"].items() + if n in self.nodes for v in seen.values() ) # if two nodes updated the state at the same time, it's ambiguous