langchain-ai/langgraph
1
0
Fork 0
mirror of https://github.com/langchain-ai/langgraph.git synced 2026-07-19 14:33:37 -04:00
Code Issues 314 Packages Projects Releases 17 Wiki Activity

[PR #6878] feat(langgraph): add messages_key filtering for message streams #5369

New Issue
Open
opened 2026-02-20 17:51:36 -05:00 by yindo · 0 comments
yindo commented 2026-02-20 17:51:36 -05:00
Owner
Copy Link
Copy Source

Original Pull Request: https://github.com/langchain-ai/langgraph/pull/6878

State: open
Merged: No

Description

Adds configurable key filtering for stream_mode="messages" so users can restrict message streaming to specific state key(s), preventing unintended leakage from internal message fields.

What changed

  • Added messages_key support to runtime streaming APIs:
    • Pregel.stream(..., messages_key=...)
    • Pregel.astream(..., messages_key=...)
  • Added compile-time default:
    • StateGraph.compile(messages_key=...)
  • Updated StreamMessagesHandler to filter emitted state messages by key when configured.
  • Preserved existing behavior by default (messages_key=None still streams all message-like fields).
  • Added sync + async regression tests and a standalone repro script:
    • tests/test_pregel.py::test_stream_mode_messages_with_messages_key_filter
    • tests/test_pregel_async.py::test_stream_mode_messages_with_messages_key_filter
    • reproduce_issue_6798.py

Issue

Closes #6798

Dependencies

None

Twitter handle

**Original Pull Request:** https://github.com/langchain-ai/langgraph/pull/6878 **State:** open **Merged:** No --- ## Description Adds configurable key filtering for `stream_mode="messages"` so users can restrict message streaming to specific state key(s), preventing unintended leakage from internal message fields. ### What changed - Added `messages_key` support to runtime streaming APIs: - `Pregel.stream(..., messages_key=...)` - `Pregel.astream(..., messages_key=...)` - Added compile-time default: - `StateGraph.compile(messages_key=...)` - Updated `StreamMessagesHandler` to filter emitted state messages by key when configured. - Preserved existing behavior by default (`messages_key=None` still streams all message-like fields). - Added sync + async regression tests and a standalone repro script: - `tests/test_pregel.py::test_stream_mode_messages_with_messages_key_filter` - `tests/test_pregel_async.py::test_stream_mode_messages_with_messages_key_filter` - `reproduce_issue_6798.py` ## Issue Closes #6798 ## Dependencies None ## Twitter handle
yindo added the pull-request label 2026-02-20 17:51:36 -05:00
Sign in to join this conversation.
No Branch/Tag Specified
Branches Tags
Labels
Clear labels
agents
awaiting author revision
bug
change
chat models
checkpointer
ci
code quality
code quality
dependencies
documentation
drawing
duplicate
enhancement
good first issue
help wanted
invalid
maintainer
meta
missing information
not maintained by langchain
open-swe-auto
open-swe-max
open-swe-max-auto
pending
performance
pull-request
Mirrored from GitHub Pull Request
 question
requires design
sdk
server
stale
studio
tests
type checking
usage question
v0.5
v1.1
No Label pull-request
Milestone
No items
No Milestone
Projects
Clear projects
No project
Assignees
Clear assignees
yindo
No Assignees
1 Participants
Notifications
Due Date
No due date set.
Dependencies

No dependencies set.

Reference: langchain-ai/langgraph#5369
Delete Branch "%!s()"

Deleting a branch is permanent. Although the deleted branch may continue to exist for a short time before it actually gets removed, it CANNOT be undone in most cases. Continue?