From b3371a1d63c813fc6b192f90bb6cd3b3d43fe65b Mon Sep 17 00:00:00 2001 From: Tat Dat Duong Date: Wed, 30 Apr 2025 19:50:14 +0200 Subject: [PATCH] Add missing "nostream" support --- libs/langgraph/langgraph/constants.py | 4 +++- libs/langgraph/langgraph/pregel/messages.py | 6 ++++-- 2 files changed, 7 insertions(+), 3 deletions(-) diff --git a/libs/langgraph/langgraph/constants.py b/libs/langgraph/langgraph/constants.py index 59e72e9c9..167c340f1 100644 --- a/libs/langgraph/langgraph/constants.py +++ b/libs/langgraph/langgraph/constants.py @@ -14,8 +14,10 @@ EMPTY_SEQ: tuple[str, ...] = tuple() MISSING = object() # --- Public constants --- -TAG_NOSTREAM = sys.intern("langsmith:nostream") +TAG_NOSTREAM = sys.intern("nostream") """Tag to disable streaming for a chat model.""" +TAG_NOSTREAM_ALT = sys.intern("langsmith:nostream") +"""Tag to disable streaming for a chat model. (Deprecated in favour of "nostream")""" TAG_HIDDEN = sys.intern("langsmith:hidden") """Tag to hide a node/edge from certain tracing/streaming environments.""" START = sys.intern("__start__") diff --git a/libs/langgraph/langgraph/pregel/messages.py b/libs/langgraph/langgraph/pregel/messages.py index de6425309..ef82bd483 100644 --- a/libs/langgraph/langgraph/pregel/messages.py +++ b/libs/langgraph/langgraph/pregel/messages.py @@ -13,7 +13,7 @@ from langchain_core.callbacks import BaseCallbackHandler from langchain_core.messages import BaseMessage from langchain_core.outputs import ChatGenerationChunk, LLMResult -from langgraph.constants import NS_SEP, TAG_HIDDEN, TAG_NOSTREAM +from langgraph.constants import NS_SEP, TAG_HIDDEN, TAG_NOSTREAM, TAG_NOSTREAM_ALT from langgraph.types import Command, StreamChunk try: @@ -93,7 +93,9 @@ class StreamMessagesHandler(BaseCallbackHandler, _StreamingCallbackHandler): metadata: Optional[dict[str, Any]] = None, **kwargs: Any, ) -> Any: - if metadata and (not tags or TAG_NOSTREAM not in tags): + if metadata and ( + not tags or (TAG_NOSTREAM not in tags and TAG_NOSTREAM_ALT not in tags) + ): self.metadata[run_id] = ( tuple(cast(str, metadata["langgraph_checkpoint_ns"]).split(NS_SEP)), metadata,