WEBDIENSTE/langgraph
1
0
Fork 0
mirror of https://github.com/langchain-ai/langgraph.git synced 2026-08-17 21:25:46 +02:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
langgraph: handle messages-tuple stream mode in RemoteGraph
This commit is contained in:
Nuno Campos
2024-11-01 09:50:51 -07:00
committed by GitHub
parent d9743f684e 84e023e10b
commit 47ea264b46
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")