From ca69566d67685b31cc43fdb3318406404fb1e9d7 Mon Sep 17 00:00:00 2001 From: vbarda Date: Thu, 31 Oct 2024 10:13:24 -0400 Subject: [PATCH] remove events --- libs/langgraph/langgraph/pregel/remote.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/libs/langgraph/langgraph/pregel/remote.py b/libs/langgraph/langgraph/pregel/remote.py index bf7a76bd5..35eee1679 100644 --- a/libs/langgraph/langgraph/pregel/remote.py +++ b/libs/langgraph/langgraph/pregel/remote.py @@ -541,6 +541,10 @@ class RemoteGraph(PregelProtocol): # add 'updates' mode if not present if "updates" not in updated_stream_modes: updated_stream_modes.append("updates") + + # remove 'events', as it's not supported in Pregel + if "events" in updated_stream_modes: + updated_stream_modes.remove("events") return (updated_stream_modes, requested_stream_modes, req_single, stream) def stream(