Add missing "nostream" support

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__")

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,