Merge pull request #2296 from langchain-ai/vb/remap

langgraph: handle messages-tuple stream mode in RemoteGraph
langchain-ai · Nov 1, 2024 · 47ea264 · 47ea264
2 parents d9743f6 + 84e023e
commit 47ea264
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/libs/langgraph/langgraph/pregel/remote.py b/libs/langgraph/langgraph/pregel/remote.py
@@ -538,6 +538,13 @@ def _get_stream_modes(
         if "messages" in updated_stream_modes:
             updated_stream_modes.remove("messages")
             updated_stream_modes.append("messages-tuple")
+
+        # if requested "messages-tuple",
+        # map to "messages" in requested_stream_modes
+        if "messages-tuple" in requested_stream_modes:
+            requested_stream_modes.remove("messages-tuple")
+            requested_stream_modes.append("messages")
+
         # add 'updates' mode if not present
         if "updates" not in updated_stream_modes:
             updated_stream_modes.append("updates")