From c0d8493db5b5d248b8d1ee735fadf7594d5d6e8e Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Tue, 17 Sep 2024 17:32:14 -0700 Subject: [PATCH] Only look at chain runs which are nodes --- libs/langgraph/langgraph/pregel/messages.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/libs/langgraph/langgraph/pregel/messages.py b/libs/langgraph/langgraph/pregel/messages.py index 2960294e3..5ee839179 100644 --- a/libs/langgraph/langgraph/pregel/messages.py +++ b/libs/langgraph/langgraph/pregel/messages.py @@ -105,7 +105,7 @@ class StreamMessagesHandler(BaseCallbackHandler, _StreamingCallbackHandler): metadata: Optional[Dict[str, Any]] = None, **kwargs: Any, ) -> Any: - if metadata: + if metadata and kwargs.get("name") == metadata.get("langgraph_node"): self.metadata[run_id] = ( tuple(metadata["langgraph_checkpoint_ns"].split(NS_SEP)), metadata,