WEBDIENSTE/langgraph
1
0
Fork 0
mirror of https://github.com/langchain-ai/langgraph.git synced 2026-09-01 04:39:01 +02:00
Code Issues Packages Projects Releases Wiki Activity

langgraph: handle messages-tuple stream mode in RemoteGraph

Browse Source
This commit is contained in:
vbarda
2024-11-01 12:43:59 -04:00
parent d9743f684e
commit 84e023e10b
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libs/langgraph/langgraph/pregel/remote.py
+7
View File
@@ -538,6 +538,13 @@ class RemoteGraph(PregelProtocol):
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")