Only look at chain runs which are nodes

This commit is contained in:
Nuno Campos
2024-09-17 17:32:14 -07:00
parent 01bb6cbd4d
commit c0d8493db5
+1 -1
View File
@@ -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,