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

test(langgraph): migrate streaming tests to stream_events(version='v3')

Browse Source 
Rename test files (test_stream_v2.py → test_stream_events_v3.py etc.),
rewrite all .stream_v2()/.astream_v2() call sites to
stream_events(input, version="v3") / astream_events(input, version="v3"),
and update test-file docstrings/comments and function names accordingly.
This commit is contained in:
Nick Hollon
2026-04-30 15:46:35 -04:00
parent fa36088a50
commit 6994703a8f
9 changed files with 162 additions and 203 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libs/langgraph/tests/test_interleave_arrival_order.py
+4 -4
View File
@@ -291,13 +291,13 @@ class TestInterleaveArrivalOrder:
# ---------------------------------------------------------------------------
# Integration test: interleave with stream_v2
# Integration test: interleave with stream_events(version="v3")
# ---------------------------------------------------------------------------
class TestInterleaveIntegration:
def test_interleave_values_and_messages(self) -> None:
run = _build_simple_graph().stream_v2({"value": "x", "items": []})
run = _build_simple_graph().stream_events({"value": "x", "items": []}, version="v3")
tagged = list(run.interleave("values", "messages"))
names = [name for name, _ in tagged]
assert set(names).issubset({"values", "messages"})
@@ -319,7 +319,7 @@ class TestInterleaveIntegration:
list(run.interleave("alpha"))
def test_interleave_releases_projections_on_completion(self) -> None:
run = _build_simple_graph().stream_v2({"value": "x", "items": []})
run = _build_simple_graph().stream_events({"value": "x", "items": []}, version="v3")
list(run.interleave("values", "messages"))
# Subscriptions should be released after the generator completes,
# so the channels can be re-iterated (they'll be empty / closed).
@@ -327,7 +327,7 @@ class TestInterleaveIntegration:
assert run.extensions["messages"]._subscribed is False
def test_interleave_releases_projections_on_early_break(self) -> None:
run = _build_simple_graph().stream_v2({"value": "x", "items": []})
run = _build_simple_graph().stream_events({"value": "x", "items": []}, version="v3")
gen = run.interleave("values", "messages")
next(gen)
gen.close()
libs/langgraph/tests/test_pregel_stream_v2.py → libs/langgraph/tests/test_pregel_stream_events_v3.py
+49 -66
View File
@@ -1,4 +1,4 @@
"""Tests for Pregel.stream_v2 / astream_v2 and the transformer pipeline."""
"""Tests for Pregel.stream_events(version="v3") / astream_events(version="v3") and the transformer pipeline."""
from __future__ import annotations
@@ -125,7 +125,7 @@ def _build_custom_stream_graph():
class _CustomPassthroughTransformer(StreamTransformer):
"""Opts a run into the `custom` stream mode without building a projection.
`stream_v2` requests only the modes that registered transformers
`stream_events(version="v3")` requests only the modes that registered transformers
declare via `required_stream_modes`. Custom events are raw user
emissions from `StreamWriter`, so tests that want them visible on
the main event log register this pass-through transformer.
@@ -390,25 +390,25 @@ class TestStreamChannelNamed:
# ---------------------------------------------------------------------------
# stream_v2 sync tests
# stream_events(version="v3") sync tests
# ---------------------------------------------------------------------------
class TestStreamV2Sync:
def test_values_projection(self) -> None:
run = _build_simple_graph().stream_v2({"value": "x", "items": []})
run = _build_simple_graph().stream_events({"value": "x", "items": []}, version="v3")
snapshots = list(run.values)
assert len(snapshots) >= 1
last = snapshots[-1]
assert "A" in last["value"] and "B" in last["value"]
def test_output(self) -> None:
run = _build_simple_graph().stream_v2({"value": "x", "items": []})
run = _build_simple_graph().stream_events({"value": "x", "items": []}, version="v3")
output = run.output
assert output == {"value": "xAB", "items": ["a", "b"]}
def test_raw_event_iteration(self) -> None:
run = _build_simple_graph().stream_v2({"value": "x", "items": []})
run = _build_simple_graph().stream_events({"value": "x", "items": []}, version="v3")
events = list(run)
assert len(events) > 0
for event in events:
@@ -418,23 +418,21 @@ class TestStreamV2Sync:
assert isinstance(event["params"]["timestamp"], int)
def test_extensions_has_native_keys(self) -> None:
run = _build_simple_graph().stream_v2({"value": "x", "items": []})
run = _build_simple_graph().stream_events({"value": "x", "items": []}, version="v3")
_ = run.output
assert "values" in run.extensions and "messages" in run.extensions
assert run.values is run.extensions["values"]
assert run.messages is run.extensions["messages"]
def test_extensions_is_read_only(self) -> None:
run = _build_simple_graph().stream_v2({"value": "x", "items": []})
run = _build_simple_graph().stream_events({"value": "x", "items": []}, version="v3")
with pytest.raises(TypeError):
run.extensions["new_key"] = object() # type: ignore[index]
with pytest.raises(TypeError):
del run.extensions["values"] # type: ignore[attr-defined]
def test_custom_stream_events(self) -> None:
run = _build_custom_stream_graph().stream_v2(
{"value": "x", "items": []},
transformers=[_CustomPassthroughTransformer],
run = _build_custom_stream_graph().stream_events({"value": "x", "items": []}, version="v3", transformers=[_CustomPassthroughTransformer],
)
custom_events = [e for e in run if e["method"] == "custom"]
assert len(custom_events) == 2
@@ -444,18 +442,18 @@ class TestStreamV2Sync:
def test_custom_events_suppressed_without_transformer(self) -> None:
"""Without a transformer declaring `"custom"`, no custom events flow.
`stream_v2` asks the graph only for the modes that registered
`stream_events(version="v3")` asks the graph only for the modes that registered
transformers require. Built-ins cover `values` / `messages`;
consumers that want raw custom events surface them by
registering a transformer whose `required_stream_modes`
includes `"custom"`.
"""
run = _build_custom_stream_graph().stream_v2({"value": "x", "items": []})
run = _build_custom_stream_graph().stream_events({"value": "x", "items": []}, version="v3")
custom_events = [e for e in run if e["method"] == "custom"]
assert custom_events == []
def test_interleave_values_and_messages(self) -> None:
run = _build_simple_graph().stream_v2({"value": "x", "items": []})
run = _build_simple_graph().stream_events({"value": "x", "items": []}, version="v3")
tagged = list(run.interleave("values", "messages"))
names = [name for name, _ in tagged]
assert set(names).issubset({"values", "messages"})
@@ -465,7 +463,7 @@ class TestStreamV2Sync:
assert run.extensions["messages"]._subscribed is False
def test_abort_marks_exhausted_and_closes_mux(self) -> None:
run = _build_simple_graph().stream_v2({"value": "x", "items": []})
run = _build_simple_graph().stream_events({"value": "x", "items": []}, version="v3")
values_iter = iter(run.values)
_ = next(values_iter)
run.abort()
@@ -474,56 +472,53 @@ class TestStreamV2Sync:
run.abort() # idempotent
def test_context_manager_calls_abort_on_exit(self) -> None:
with _build_simple_graph().stream_v2({"value": "x", "items": []}) as run:
with _build_simple_graph().stream_events({"value": "x", "items": []}, version="v3") as run:
_ = next(iter(run.values))
assert run._exhausted is True
def test_interleave_unknown_projection(self) -> None:
run = _build_simple_graph().stream_v2({"value": "x", "items": []})
run = _build_simple_graph().stream_events({"value": "x", "items": []}, version="v3")
with pytest.raises(KeyError):
list(run.interleave("values", "does_not_exist"))
class TestStreamV2SyncErrors:
def test_error_propagation_output(self) -> None:
run = _build_error_graph().stream_v2({"value": "x", "items": []})
run = _build_error_graph().stream_events({"value": "x", "items": []}, version="v3")
with pytest.raises(ValueError, match="boom"):
_ = run.output
def test_error_propagation_values(self) -> None:
run = _build_error_graph().stream_v2({"value": "x", "items": []})
run = _build_error_graph().stream_events({"value": "x", "items": []}, version="v3")
with pytest.raises(ValueError, match="boom"):
list(run.values)
def test_error_propagation_raw_events(self) -> None:
run = _build_error_graph().stream_v2({"value": "x", "items": []})
run = _build_error_graph().stream_events({"value": "x", "items": []}, version="v3")
with pytest.raises(ValueError, match="boom"):
list(run)
def test_error_propagation_interrupted(self) -> None:
run = _build_error_graph().stream_v2({"value": "x", "items": []})
run = _build_error_graph().stream_events({"value": "x", "items": []}, version="v3")
with pytest.raises(ValueError, match="boom"):
_ = run.interrupted
def test_error_propagation_interrupts(self) -> None:
run = _build_error_graph().stream_v2({"value": "x", "items": []})
run = _build_error_graph().stream_events({"value": "x", "items": []}, version="v3")
with pytest.raises(ValueError, match="boom"):
_ = run.interrupts
class TestStreamV2SyncInterrupt:
def test_interrupted(self) -> None:
run = _build_interrupt_graph().stream_v2(
{"value": "x", "items": []},
{"configurable": {"thread_id": "t1"}},
)
run = _build_interrupt_graph().stream_events({"value": "x", "items": []}, {"configurable": {"thread_id": "t1"}}, version="v3")
_ = run.output
assert run.interrupted is True
assert len(run.interrupts) > 0
# ---------------------------------------------------------------------------
# astream_v2 async tests
# astream_events(version="v3") async tests
# ---------------------------------------------------------------------------
@@ -531,26 +526,26 @@ class TestStreamV2SyncInterrupt:
@NEEDS_CONTEXTVARS
class TestStreamV2Async:
async def test_values_projection(self) -> None:
run = await _build_simple_graph().astream_v2({"value": "x", "items": []})
run = await _build_simple_graph().astream_events({"value": "x", "items": []}, version="v3")
snapshots = [s async for s in run.values]
assert len(snapshots) >= 1
last = snapshots[-1]
assert "A" in last["value"] and "B" in last["value"]
async def test_output(self) -> None:
run = await _build_simple_graph().astream_v2({"value": "x", "items": []})
run = await _build_simple_graph().astream_events({"value": "x", "items": []}, version="v3")
output = await run.output()
assert output == {"value": "xAB", "items": ["a", "b"]}
async def test_raw_event_iteration(self) -> None:
run = await _build_simple_graph().astream_v2({"value": "x", "items": []})
run = await _build_simple_graph().astream_events({"value": "x", "items": []}, version="v3")
events = [e async for e in run]
assert len(events) > 0
for event in events:
assert event["type"] == "event"
async def test_abort_marks_exhausted_and_closes_mux(self) -> None:
run = await _build_simple_graph().astream_v2({"value": "x", "items": []})
run = await _build_simple_graph().astream_events({"value": "x", "items": []}, version="v3")
values_iter = aiter(run.values)
_ = await anext(values_iter)
await run.abort()
@@ -560,22 +555,20 @@ class TestStreamV2Async:
await run.abort() # idempotent
async def test_context_manager_calls_abort_on_exit(self) -> None:
run = await _build_simple_graph().astream_v2({"value": "x", "items": []})
run = await _build_simple_graph().astream_events({"value": "x", "items": []}, version="v3")
async with run:
_ = await anext(aiter(run.values))
assert run._exhausted is True
async def test_extensions_has_native_keys(self) -> None:
run = await _build_simple_graph().astream_v2({"value": "x", "items": []})
run = await _build_simple_graph().astream_events({"value": "x", "items": []}, version="v3")
_ = await run.output()
assert "values" in run.extensions and "messages" in run.extensions
assert run.values is run.extensions["values"]
assert run.messages is run.extensions["messages"]
async def test_custom_stream_events(self) -> None:
run = await _build_custom_stream_graph().astream_v2(
{"value": "x", "items": []},
transformers=[_CustomPassthroughTransformer],
run = await _build_custom_stream_graph().astream_events({"value": "x", "items": []}, version="v3", transformers=[_CustomPassthroughTransformer],
)
events = [e async for e in run]
custom_events = [e for e in events if e["method"] == "custom"]
@@ -588,29 +581,29 @@ class TestStreamV2Async:
@NEEDS_CONTEXTVARS
class TestStreamV2AsyncErrors:
async def test_error_propagation_output(self) -> None:
run = await _build_error_graph().astream_v2({"value": "x", "items": []})
run = await _build_error_graph().astream_events({"value": "x", "items": []}, version="v3")
with pytest.raises(ValueError, match="boom"):
await run.output()
async def test_error_propagation_values(self) -> None:
run = await _build_error_graph().astream_v2({"value": "x", "items": []})
run = await _build_error_graph().astream_events({"value": "x", "items": []}, version="v3")
with pytest.raises(ValueError, match="boom"):
async for _ in run.values:
pass
async def test_error_propagation_raw_events(self) -> None:
run = await _build_error_graph().astream_v2({"value": "x", "items": []})
run = await _build_error_graph().astream_events({"value": "x", "items": []}, version="v3")
with pytest.raises(ValueError, match="boom"):
async for _ in run:
pass
async def test_error_propagation_interrupted(self) -> None:
run = await _build_error_graph().astream_v2({"value": "x", "items": []})
run = await _build_error_graph().astream_events({"value": "x", "items": []}, version="v3")
with pytest.raises(ValueError, match="boom"):
await run.interrupted()
async def test_error_propagation_interrupts(self) -> None:
run = await _build_error_graph().astream_v2({"value": "x", "items": []})
run = await _build_error_graph().astream_events({"value": "x", "items": []}, version="v3")
with pytest.raises(ValueError, match="boom"):
await run.interrupts()
@@ -619,10 +612,7 @@ class TestStreamV2AsyncErrors:
@NEEDS_CONTEXTVARS
class TestStreamV2AsyncInterrupt:
async def test_interrupted(self) -> None:
run = await _build_interrupt_graph().astream_v2(
{"value": "x", "items": []},
{"configurable": {"thread_id": "t2"}},
)
run = await _build_interrupt_graph().astream_events({"value": "x", "items": []}, {"configurable": {"thread_id": "t2"}}, version="v3")
_ = await run.output()
assert await run.interrupted() is True
assert len(await run.interrupts()) > 0
@@ -985,8 +975,7 @@ class TestCustomTransformer:
self._channel.push(self._count)
return True
run = _build_simple_graph().stream_v2(
{"value": "x", "items": []}, transformers=[CounterTransformer]
run = _build_simple_graph().stream_events({"value": "x", "items": []}, version="v3", transformers=[CounterTransformer]
)
assert "counter" in run.extensions
counter_iter = iter(run.extensions["counter"])
@@ -1011,15 +1000,14 @@ class TestCustomTransformer:
self._log.push("saw_values")
return True
run = _build_simple_graph().stream_v2(
{"value": "x", "items": []}, transformers=[FooTransformer]
run = _build_simple_graph().stream_events({"value": "x", "items": []}, version="v3", transformers=[FooTransformer]
)
foo_iter = iter(run.foo)
_ = run.output
assert "foo" in run.extensions and run.foo is run.extensions["foo"]
assert "saw_values" in list(foo_iter)
def test_stream_v2_rejects_transformer_instances(self) -> None:
def test_stream_events_v3_rejects_transformer_instances(self) -> None:
class InstanceTransformer(StreamTransformer):
def init(self) -> dict[str, Any]:
return {}
@@ -1028,8 +1016,7 @@ class TestCustomTransformer:
return True
with pytest.raises(TypeError, match="pre-built instance"):
_build_simple_graph().stream_v2(
{"value": "x", "items": []}, transformers=[InstanceTransformer()]
_build_simple_graph().stream_events({"value": "x", "items": []}, version="v3", transformers=[InstanceTransformer()]
)
def test_stream_channel_auto_forward(self) -> None:
@@ -1048,8 +1035,7 @@ class TestCustomTransformer:
self._channel.push("emitted")
return True
run = _build_simple_graph().stream_v2(
{"value": "x", "items": []}, transformers=[EmitterTransformer]
run = _build_simple_graph().stream_events({"value": "x", "items": []}, version="v3", transformers=[EmitterTransformer]
)
custom_events = [e for e in run if e["method"] == "custom:emitter"]
assert len(custom_events) > 0
@@ -1093,8 +1079,7 @@ class TestCustomTransformer:
return True
with pytest.raises(ValueError, match=r"conflict.*'values'.*ValuesTransformer"):
_build_simple_graph().stream_v2(
{"value": "x", "items": []}, transformers=[ConflictTransformer]
_build_simple_graph().stream_events({"value": "x", "items": []}, version="v3", transformers=[ConflictTransformer]
)
@@ -1176,8 +1161,7 @@ class TestStreamChannelAutoLifecycle:
self._log.push("got_it")
return True
run = _build_simple_graph().stream_v2(
{"value": "x", "items": []}, transformers=[MinimalTransformer]
run = _build_simple_graph().stream_events({"value": "x", "items": []}, version="v3", transformers=[MinimalTransformer]
)
minimal_iter = iter(run.extensions["minimal"])
_ = run.output
@@ -1437,8 +1421,7 @@ class TestAsyncTransformerLane:
async def afinalize(self) -> None:
self._log.close()
run = await _build_simple_graph().astream_v2(
{"value": "x", "items": []}, transformers=[Scorer]
run = await _build_simple_graph().astream_events({"value": "x", "items": []}, version="v3", transformers=[Scorer]
)
scores_cursor = aiter(run.extensions["scores"])
_ = await run.output()
@@ -1454,7 +1437,7 @@ class TestAsyncTransformerLane:
@NEEDS_CONTEXTVARS
class TestMemoryBounds:
def test_sync_subscribed_buffer_stays_at_most_one_between_yields(self) -> None:
run = _build_simple_graph().stream_v2({"value": "x", "items": []})
run = _build_simple_graph().stream_events({"value": "x", "items": []}, version="v3")
events_iter = iter(run)
max_buffered = 0
count = 0
@@ -1467,7 +1450,7 @@ class TestMemoryBounds:
)
def test_unsubscribed_projections_never_accumulate(self) -> None:
run = _build_simple_graph().stream_v2({"value": "x", "items": []})
run = _build_simple_graph().stream_events({"value": "x", "items": []}, version="v3")
list(run)
values_log = run.extensions["values"]
messages_log = run.extensions["messages"]
@@ -1475,19 +1458,19 @@ class TestMemoryBounds:
assert len(messages_log._items) == 0 and not messages_log._subscribed
def test_output_path_does_not_retain_values(self) -> None:
run = _build_simple_graph().stream_v2({"value": "x", "items": []})
run = _build_simple_graph().stream_events({"value": "x", "items": []}, version="v3")
_ = run.output
values_log = run.extensions["values"]
assert len(values_log._items) == 0 and not values_log._subscribed
def test_drained_subscriber_buffer_returns_to_empty(self) -> None:
run = _build_simple_graph().stream_v2({"value": "x", "items": []})
run = _build_simple_graph().stream_events({"value": "x", "items": []}, version="v3")
list(run.values)
assert len(run.extensions["values"]._items) == 0
@pytest.mark.anyio
async def test_async_single_consumer_buffer_stays_at_most_one(self) -> None:
run = await _build_simple_graph().astream_v2({"value": "x", "items": []})
run = await _build_simple_graph().astream_events({"value": "x", "items": []}, version="v3")
max_buffered = 0
count = 0
async for _ in run:
@@ -1498,7 +1481,7 @@ class TestMemoryBounds:
@pytest.mark.anyio
async def test_async_unsubscribed_projections_never_accumulate(self) -> None:
run = await _build_simple_graph().astream_v2({"value": "x", "items": []})
run = await _build_simple_graph().astream_events({"value": "x", "items": []}, version="v3")
_ = await run.output()
values_log = run.extensions["values"]
messages_log = run.extensions["messages"]
libs/langgraph/tests/test_stream_data_transformers.py
+19 -30
View File
@@ -4,7 +4,7 @@ These transformers capture raw protocol events for their respective stream
modes and expose them as native projections on the run stream (run.custom,
run.updates, run.checkpoints, run.debug, run.tasks). Tests dispatch synthetic
protocol events through a StreamMux to isolate transformer logic; the final
group exercises real graphs through stream_v2.
group exercises real graphs through stream_events(version="v3").
"""
from __future__ import annotations
@@ -478,7 +478,7 @@ def test_unrelated_events_ignored_by_all() -> None:
# ---------------------------------------------------------------------------
# End-to-end: real graphs through stream_v2
# End-to-end: real graphs through stream_events(version="v3")
# ---------------------------------------------------------------------------
@@ -503,11 +503,10 @@ def _make_simple_graph() -> Any:
return builder.compile()
def test_stream_v2_custom_projection_opt_in() -> None:
def test_stream_events_v3_custom_projection_opt_in() -> None:
"""run.custom surfaces get_stream_writer() payloads when opted in."""
graph = _make_simple_graph()
run = graph.stream_v2(
{"value": "hello", "items": []}, transformers=[CustomTransformer]
run = graph.stream_events({"value": "hello", "items": []}, version="v3", transformers=[CustomTransformer]
)
custom_events = list(run.custom)
@@ -515,11 +514,10 @@ def test_stream_v2_custom_projection_opt_in() -> None:
assert any(e.get("status") == "working" for e in custom_events)
def test_stream_v2_custom_and_values_coexist() -> None:
def test_stream_events_v3_custom_and_values_coexist() -> None:
"""Both run.custom and run.values work in the same run."""
graph = _make_simple_graph()
run = graph.stream_v2(
{"value": "hello", "items": []}, transformers=[CustomTransformer]
run = graph.stream_events({"value": "hello", "items": []}, version="v3", transformers=[CustomTransformer]
)
custom_events = list(run.custom)
@@ -528,10 +526,10 @@ def test_stream_v2_custom_and_values_coexist() -> None:
assert len(custom_events) >= 1
def test_stream_v2_tasks_projection_opt_in() -> None:
def test_stream_events_v3_tasks_projection_opt_in() -> None:
"""run.tasks surfaces raw task events when opted in via transformers=."""
graph = _make_simple_graph()
run = graph.stream_v2({"value": "x", "items": []}, transformers=[TasksTransformer])
run = graph.stream_events({"value": "x", "items": []}, transformers=[TasksTransformer], version="v3")
tasks_events = list(run.tasks)
assert len(tasks_events) >= 1
@@ -539,10 +537,10 @@ def test_stream_v2_tasks_projection_opt_in() -> None:
assert "my_node" in names
def test_stream_v2_debug_projection_opt_in() -> None:
def test_stream_events_v3_debug_projection_opt_in() -> None:
"""run.debug surfaces debug events when opted in via transformers=."""
graph = _make_simple_graph()
run = graph.stream_v2({"value": "x", "items": []}, transformers=[DebugTransformer])
run = graph.stream_events({"value": "x", "items": []}, transformers=[DebugTransformer], version="v3")
debug_events = list(run.debug)
assert len(debug_events) >= 1
@@ -550,11 +548,10 @@ def test_stream_v2_debug_projection_opt_in() -> None:
assert types & {"checkpoint", "task", "task_result"}
def test_stream_v2_updates_projection_opt_in() -> None:
def test_stream_events_v3_updates_projection_opt_in() -> None:
"""run.updates surfaces node output dicts when opted in via transformers=."""
graph = _make_simple_graph()
run = graph.stream_v2(
{"value": "x", "items": []}, transformers=[UpdatesTransformer]
run = graph.stream_events({"value": "x", "items": []}, version="v3", transformers=[UpdatesTransformer]
)
updates = list(run.updates)
@@ -563,12 +560,10 @@ def test_stream_v2_updates_projection_opt_in() -> None:
assert "my_node" in node_names
def test_stream_v2_all_transformers_interleaved() -> None:
def test_stream_events_v3_all_transformers_interleaved() -> None:
"""All five transformers registered together, consumed via interleave."""
graph = _make_simple_graph()
run = graph.stream_v2(
{"value": "x", "items": []},
transformers=[
run = graph.stream_events({"value": "x", "items": []}, version="v3", transformers=[
CustomTransformer,
UpdatesTransformer,
CheckpointsTransformer,
@@ -599,7 +594,7 @@ def test_stream_v2_all_transformers_interleaved() -> None:
assert run.output["value"] == "x!"
def test_stream_v2_all_transformers_with_checkpointer() -> None:
def test_stream_events_v3_all_transformers_with_checkpointer() -> None:
"""All transformers with a checkpointer — run.checkpoints populated."""
from langgraph.checkpoint.memory import InMemorySaver
@@ -609,9 +604,7 @@ def test_stream_v2_all_transformers_with_checkpointer() -> None:
builder.add_edge("my_node", END)
graph = builder.compile(checkpointer=InMemorySaver())
run = graph.stream_v2(
{"value": "x", "items": []},
config={"configurable": {"thread_id": "test-all"}},
run = graph.stream_events({"value": "x", "items": []}, version="v3", config={"configurable": {"thread_id": "test-all"}},
transformers=[
CustomTransformer,
UpdatesTransformer,
@@ -637,7 +630,7 @@ def test_stream_v2_all_transformers_with_checkpointer() -> None:
assert len(collected["custom"]) >= 1
def test_stream_v2_checkpoints_projection_opt_in() -> None:
def test_stream_events_v3_checkpoints_projection_opt_in() -> None:
"""run.checkpoints surfaces checkpoint data when opted in with a checkpointer."""
from langgraph.checkpoint.memory import InMemorySaver
@@ -647,9 +640,7 @@ def test_stream_v2_checkpoints_projection_opt_in() -> None:
builder.add_edge("my_node", END)
graph = builder.compile(checkpointer=InMemorySaver())
run = graph.stream_v2(
{"value": "x", "items": []},
config={"configurable": {"thread_id": "test-ckpt-standalone"}},
run = graph.stream_events({"value": "x", "items": []}, version="v3", config={"configurable": {"thread_id": "test-ckpt-standalone"}},
transformers=[CheckpointsTransformer],
)
@@ -688,9 +679,7 @@ def test_tasks_and_lifecycle_coregistration_e2e() -> None:
is present and suppressing them from the main log.
"""
graph = _make_simple_graph()
run = graph.stream_v2(
{"value": "x", "items": []},
transformers=[TasksTransformer],
run = graph.stream_events({"value": "x", "items": []}, version="v3", transformers=[TasksTransformer],
)
tasks_events = list(run.tasks)
libs/langgraph/tests/test_stream_v2.py → libs/langgraph/tests/test_stream_events_v3.py
+3 -3
View File
@@ -231,7 +231,7 @@ class TestV2Stream:
for c in chunks:
_assert_stream_part_shape(c)
def test_stream_v2_accepts_control_for_drain(self) -> None:
def test_stream_events_v3_accepts_control_for_drain(self) -> None:
class DrainState(TypedDict, total=False):
value: str
skipped: str
@@ -253,7 +253,7 @@ class TestV2Stream:
builder.add_edge("second", END)
graph = builder.compile()
run = graph.stream_v2({}, control=control)
run = graph.stream_events({}, control=control, version="v3")
with pytest.raises(GraphDrained, match="sigterm"):
list(run.values)
@@ -1124,7 +1124,7 @@ class TestV2ValidationErrors:
_INVALID_INPUT: dict[str, Any] = {"value": [1, 2, 3], "items": []}
def test_stream_v2_pydantic_validation_error(self) -> None:
def test_stream_events_v3_pydantic_validation_error(self) -> None:
"""Invalid input to stream with v2 + pydantic state raises ValidationError."""
graph = _make_pydantic_graph()
with pytest.raises(ValidationError):
libs/langgraph/tests/test_stream_v2_e2e.py → libs/langgraph/tests/test_stream_events_v3_e2e.py
+39 -51
View File
@@ -1,9 +1,9 @@
"""End-to-end tests exercising all stream_v2 projections together.
"""End-to-end tests exercising all stream_events(version="v3") projections together.
Each test builds a realistic graph (subgraphs, LLM calls, custom writers,
interrupts) and verifies that every projection — values, messages, lifecycle,
subgraphs, raw events, output, interleave — produces correct, consistent
results through a single stream_v2 / astream_v2 run.
results through a single stream_events(version="v3") / astream_events(version="v3") run.
"""
from __future__ import annotations
@@ -218,9 +218,9 @@ class _CounterTransformer(StreamTransformer):
class TestStreamV2E2ESync:
def test_all_projections_nested_graph(self) -> None:
"""Run a nested graph through stream_v2 and verify values + lifecycle."""
"""Run a nested graph through stream_events(version="v3") and verify values + lifecycle."""
graph = _make_nested_graph()
run = graph.stream_v2({"value": "x", "items": []})
run = graph.stream_events({"value": "x", "items": []}, version="v3")
values_snapshots: list[dict[str, Any]] = []
lifecycle_events: list[dict[str, Any]] = []
@@ -246,7 +246,7 @@ class TestStreamV2E2ESync:
def test_subgraph_handles_with_drill_down(self) -> None:
"""Subgraph handles yield and support values drill-down."""
graph = _make_nested_graph()
run = graph.stream_v2({"value": "x", "items": []})
run = graph.stream_events({"value": "x", "items": []}, version="v3")
handles = []
for handle in run.subgraphs:
@@ -270,7 +270,7 @@ class TestStreamV2E2ESync:
def test_raw_events_have_monotonic_seq(self) -> None:
"""Raw protocol events have monotonically increasing seq numbers."""
graph = _make_nested_graph()
run = graph.stream_v2({"value": "x", "items": []})
run = graph.stream_events({"value": "x", "items": []}, version="v3")
events = list(run)
assert len(events) > 0
@@ -285,11 +285,11 @@ class TestStreamV2E2ESync:
def test_output_matches_final_values_snapshot(self) -> None:
"""output property returns the same state as the last values snapshot."""
run1 = _make_nested_graph().stream_v2({"value": "x", "items": []})
run1 = _make_nested_graph().stream_events({"value": "x", "items": []}, version="v3")
snapshots = list(run1.values)
final_via_values = snapshots[-1]
run2 = _make_nested_graph().stream_v2({"value": "x", "items": []})
run2 = _make_nested_graph().stream_events({"value": "x", "items": []}, version="v3")
final_via_output = run2.output
assert final_via_values == final_via_output
@@ -297,7 +297,7 @@ class TestStreamV2E2ESync:
def test_context_manager_and_abort(self) -> None:
"""Context manager calls abort, marking the stream exhausted."""
graph = _make_nested_graph()
with graph.stream_v2({"value": "x", "items": []}) as run:
with graph.stream_events({"value": "x", "items": []}, version="v3") as run:
first_val = next(iter(run.values))
assert isinstance(first_val, dict)
assert run._exhausted is True
@@ -305,7 +305,7 @@ class TestStreamV2E2ESync:
def test_extensions_has_all_native_keys(self) -> None:
"""Extensions dict exposes all native projection keys."""
graph = _make_nested_graph()
run = graph.stream_v2({"value": "x", "items": []})
run = graph.stream_events({"value": "x", "items": []}, version="v3")
_ = run.output
assert "values" in run.extensions
@@ -327,7 +327,7 @@ class TestStreamV2E2EMessages:
def test_messages_projection_from_invoke(self) -> None:
"""Messages projection captures LLM calls via model.invoke() auto-routing."""
graph = _make_messages_graph()
run = graph.stream_v2({"messages": "hi"})
run = graph.stream_events({"messages": "hi"}, version="v3")
streams = list(run.messages)
assert len(streams) >= 1
@@ -350,7 +350,7 @@ class TestStreamV2E2EMessages:
.compile()
)
run = graph.stream_v2({"messages": "go"})
run = graph.stream_events({"messages": "go"}, version="v3")
(stream,) = list(run.messages)
assert "".join(stream.text) == "streamed answer"
@@ -368,7 +368,7 @@ class TestStreamV2E2EMessages:
.compile()
)
run = graph.stream_v2({"messages": "hi"})
run = graph.stream_events({"messages": "hi"}, version="v3")
(stream,) = list(run.messages)
assert stream.output.text == "hardcoded"
assert stream.message_id == "msg-1"
@@ -376,7 +376,7 @@ class TestStreamV2E2EMessages:
def test_root_messages_only_shows_root_scope(self) -> None:
"""Root messages projection doesn't surface subgraph-scoped messages."""
graph = _make_messages_subgraph()
run = graph.stream_v2({"messages": ["hi"], "done": False})
run = graph.stream_events({"messages": ["hi"], "done": False}, version="v3")
root_streams = list(run.messages)
# The message is emitted inside the subgraph, so the root
# messages projection (scoped to root namespace) doesn't see it.
@@ -385,7 +385,7 @@ class TestStreamV2E2EMessages:
def test_subgraph_handle_messages_drill_down(self) -> None:
"""Drilling into subgraph handle's messages surfaces subgraph messages."""
graph = _make_messages_subgraph()
run = graph.stream_v2({"messages": ["hi"], "done": False})
run = graph.stream_events({"messages": ["hi"], "done": False}, version="v3")
found_messages = False
for handle in run.subgraphs:
@@ -407,9 +407,7 @@ class TestStreamV2E2ECustom:
"""Custom StreamWriter events appear on the main log when a
transformer declares the custom mode."""
graph = _make_custom_writer_graph()
run = graph.stream_v2(
{"value": "x", "items": []},
transformers=[_CustomPassthroughTransformer],
run = graph.stream_events({"value": "x", "items": []}, version="v3", transformers=[_CustomPassthroughTransformer],
)
events = list(run)
custom = [e for e in events if e["method"] == "custom"]
@@ -420,7 +418,7 @@ class TestStreamV2E2ECustom:
def test_custom_events_suppressed_without_transformer(self) -> None:
"""Without a custom-mode transformer, custom events don't flow."""
graph = _make_custom_writer_graph()
run = graph.stream_v2({"value": "x", "items": []})
run = graph.stream_events({"value": "x", "items": []}, version="v3")
events = list(run)
custom = [e for e in events if e["method"] == "custom"]
assert custom == []
@@ -428,9 +426,7 @@ class TestStreamV2E2ECustom:
def test_custom_transformer_with_stream_channel(self) -> None:
"""A custom transformer with a StreamChannel produces extension data."""
graph = _make_nested_graph()
run = graph.stream_v2(
{"value": "x", "items": []},
transformers=[_CounterTransformer],
run = graph.stream_events({"value": "x", "items": []}, version="v3", transformers=[_CounterTransformer],
)
assert "counter" in run.extensions
@@ -444,9 +440,7 @@ class TestStreamV2E2ECustom:
def test_custom_channel_events_on_main_log(self) -> None:
"""StreamChannel auto-forward injects custom:<name> events into the main log."""
graph = _make_nested_graph()
run = graph.stream_v2(
{"value": "x", "items": []},
transformers=[_CounterTransformer],
run = graph.stream_events({"value": "x", "items": []}, version="v3", transformers=[_CounterTransformer],
)
events = list(run)
counter_events = [e for e in events if e["method"] == "custom:counter"]
@@ -464,7 +458,7 @@ class TestStreamV2E2EInterrupt:
"""Interrupted run has correct flags and interrupt payloads."""
graph = _make_interrupt_graph()
config: dict[str, Any] = {"configurable": {"thread_id": "int-1"}}
run = graph.stream_v2({"value": "x", "items": []}, config)
run = graph.stream_events({"value": "x", "items": []}, config, version="v3")
output = run.output
assert output is not None
@@ -477,7 +471,7 @@ class TestStreamV2E2EInterrupt:
"""Values snapshots captured before the interrupt reflect partial state."""
graph = _make_interrupt_graph()
config: dict[str, Any] = {"configurable": {"thread_id": "int-2"}}
run = graph.stream_v2({"value": "x", "items": []}, config)
run = graph.stream_events({"value": "x", "items": []}, config, version="v3")
snapshots = list(run.values)
assert len(snapshots) >= 1
@@ -494,14 +488,14 @@ class TestStreamV2E2EErrors:
def test_subgraph_error_propagates_through_output(self) -> None:
"""Error in a subgraph propagates through output."""
graph = _make_error_subgraph()
run = graph.stream_v2({"value": "x", "items": []})
run = graph.stream_events({"value": "x", "items": []}, version="v3")
with pytest.raises(ValueError, match="subgraph explosion"):
_ = run.output
def test_subgraph_error_propagates_through_raw_events(self) -> None:
graph = _make_error_subgraph()
run = graph.stream_v2({"value": "x", "items": []})
run = graph.stream_events({"value": "x", "items": []}, version="v3")
with pytest.raises(ValueError, match="subgraph explosion"):
list(run)
@@ -509,7 +503,7 @@ class TestStreamV2E2EErrors:
def test_error_subgraph_handle_status(self) -> None:
"""Subgraph handle surfaces the error status."""
graph = _make_error_subgraph()
run = graph.stream_v2({"value": "x", "items": []})
run = graph.stream_events({"value": "x", "items": []}, version="v3")
handle = next(iter(run.subgraphs))
with pytest.raises(RuntimeError, match="subgraph explosion"):
@@ -529,7 +523,7 @@ class TestStreamV2E2EAsync:
async def test_all_projections_async(self) -> None:
"""Async run exercises values projection."""
graph = _make_nested_graph()
run = await graph.astream_v2({"value": "x", "items": []})
run = await graph.astream_events({"value": "x", "items": []}, version="v3")
values_snapshots = [s async for s in run.values]
assert len(values_snapshots) >= 1
@@ -540,7 +534,7 @@ class TestStreamV2E2EAsync:
async def test_async_output(self) -> None:
"""Async output returns the final state."""
graph = _make_nested_graph()
run = await graph.astream_v2({"value": "x", "items": []})
run = await graph.astream_events({"value": "x", "items": []}, version="v3")
output = await run.output()
assert output is not None
assert output["value"] == "x_routed_processed"
@@ -550,7 +544,7 @@ class TestStreamV2E2EAsync:
async def test_async_raw_events(self) -> None:
"""Async raw event iteration yields well-formed ProtocolEvents."""
graph = _make_nested_graph()
run = await graph.astream_v2({"value": "x", "items": []})
run = await graph.astream_events({"value": "x", "items": []}, version="v3")
events = [e async for e in run]
assert len(events) > 0
seqs = [e["seq"] for e in events]
@@ -572,7 +566,7 @@ class TestStreamV2E2EAsync:
.compile()
)
run = await graph.astream_v2({"messages": "hi"})
run = await graph.astream_events({"messages": "hi"}, version="v3")
streams = [s async for s in run.messages]
assert len(streams) >= 1
for s in streams:
@@ -583,7 +577,7 @@ class TestStreamV2E2EAsync:
"""Async interrupted run has correct flags."""
graph = _make_interrupt_graph()
config: dict[str, Any] = {"configurable": {"thread_id": "async-int-1"}}
run = await graph.astream_v2({"value": "x", "items": []}, config)
run = await graph.astream_events({"value": "x", "items": []}, config, version="v3")
output = await run.output()
assert output is not None
@@ -593,14 +587,14 @@ class TestStreamV2E2EAsync:
async def test_async_error_propagation(self) -> None:
"""Async error from subgraph propagates through output."""
graph = _make_error_subgraph()
run = await graph.astream_v2({"value": "x", "items": []})
run = await graph.astream_events({"value": "x", "items": []}, version="v3")
with pytest.raises(ValueError, match="subgraph explosion"):
await run.output()
async def test_async_context_manager(self) -> None:
"""Async context manager calls abort on exit."""
graph = _make_nested_graph()
run = await graph.astream_v2({"value": "x", "items": []})
run = await graph.astream_events({"value": "x", "items": []}, version="v3")
async with run:
_ = await anext(aiter(run.values))
assert run._exhausted is True
@@ -608,7 +602,7 @@ class TestStreamV2E2EAsync:
async def test_async_extensions_present(self) -> None:
"""Async run has all native extensions."""
graph = _make_nested_graph()
run = await graph.astream_v2({"value": "x", "items": []})
run = await graph.astream_events({"value": "x", "items": []}, version="v3")
_ = await run.output()
assert "values" in run.extensions
assert "messages" in run.extensions
@@ -618,9 +612,7 @@ class TestStreamV2E2EAsync:
async def test_async_custom_transformer(self) -> None:
"""Async custom transformer with StreamChannel works."""
graph = _make_nested_graph()
run = await graph.astream_v2(
{"value": "x", "items": []},
transformers=[_CounterTransformer],
run = await graph.astream_events({"value": "x", "items": []}, version="v3", transformers=[_CounterTransformer],
)
assert "counter" in run.extensions
counter_cursor = aiter(run.extensions["counter"])
@@ -639,7 +631,7 @@ class TestStreamV2E2ECombined:
def test_interleave_all_native_projections(self) -> None:
"""Interleave values + messages + lifecycle without deadlock."""
graph = _make_nested_graph()
run = graph.stream_v2({"value": "x", "items": []})
run = graph.stream_events({"value": "x", "items": []}, version="v3")
seen_names: set[str] = set()
for name, _item in run.interleave("values", "messages", "lifecycle"):
@@ -667,9 +659,7 @@ class TestStreamV2E2ECombined:
return True
graph = _make_nested_graph()
run = graph.stream_v2(
{"value": "x", "items": []},
transformers=[_CounterTransformer, TagTransformer],
run = graph.stream_events({"value": "x", "items": []}, version="v3", transformers=[_CounterTransformer, TagTransformer],
)
assert "counter" in run.extensions
@@ -722,7 +712,7 @@ class TestStreamV2E2ECombined:
.compile()
)
run = outer.stream_v2({"items": []})
run = outer.stream_events({"items": []}, version="v3")
handles = []
for handle in run.subgraphs:
list(handle.values)
@@ -740,13 +730,13 @@ class TestStreamV2E2ECombined:
def test_lifecycle_matches_subgraph_handles(self) -> None:
"""Lifecycle events and subgraph handles agree on discovered subgraphs."""
run1 = _make_nested_graph().stream_v2({"value": "x", "items": []})
run1 = _make_nested_graph().stream_events({"value": "x", "items": []}, version="v3")
handle_paths: list[tuple[str, ...]] = []
for handle in run1.subgraphs:
list(handle.values)
handle_paths.append(handle.path)
run2 = _make_nested_graph().stream_v2({"value": "x", "items": []})
run2 = _make_nested_graph().stream_events({"value": "x", "items": []}, version="v3")
lifecycle = list(run2.lifecycle)
started_ns = [
@@ -773,9 +763,7 @@ class TestStreamV2E2ECombined:
.compile()
)
run = graph.stream_v2(
{"messages": "hi"},
transformers=[_CounterTransformer],
run = graph.stream_events({"messages": "hi"}, version="v3", transformers=[_CounterTransformer],
)
counter_iter = iter(run.extensions["counter"])
libs/langgraph/tests/test_stream_lifecycle_transformer.py
+8 -8
View File
@@ -6,7 +6,7 @@ on the `lifecycle` channel for both in-process iteration via
events. Most tests dispatch synthetic protocol events through a
`StreamMux` to keep the inference logic isolated; the end-of-file
group exercises the path through real graphs (multi-depth
discovery, nested `stream_v2` calls with non-empty `parent_ns`).
discovery, nested `stream_events(version="v3")` calls with non-empty `parent_ns`).
"""
from __future__ import annotations
@@ -324,7 +324,7 @@ def test_tasks_events_suppressed_from_main_log() -> None:
# ---------------------------------------------------------------------------
# End-to-end: real graphs through stream_v2
# End-to-end: real graphs through stream_events(version="v3")
# ---------------------------------------------------------------------------
@@ -358,10 +358,10 @@ def _make_two_level_nested() -> Any:
return outer_b.compile()
def test_stream_v2_real_graph_emits_lifecycle_at_each_depth() -> None:
def test_stream_events_v3_real_graph_emits_lifecycle_at_each_depth() -> None:
"""Outer graph with two nested subgraphs surfaces lifecycle for both."""
graph = _make_two_level_nested()
run = graph.stream_v2({"value": "x", "items": []})
run = graph.stream_events({"value": "x", "items": []}, version="v3")
# Iterating the projection drives the pump and drains synthesized
# lifecycle events at the same time.
@@ -384,17 +384,17 @@ def test_stream_v2_real_graph_emits_lifecycle_at_each_depth() -> None:
)
def test_stream_v2_with_nested_parent_ns_scopes_lifecycle() -> None:
"""When `stream_v2` is called with a non-empty checkpoint_ns in config,
def test_stream_events_v3_with_nested_parent_ns_scopes_lifecycle() -> None:
"""When `stream_events(version="v3")` is called with a non-empty checkpoint_ns in config,
`_resolve_parent_ns` returns that namespace and the registered
`LifecycleTransformer` is constructed with `scope=parent_ns`. This
exercises the path that exists today purely for nested-stream_v2
exercises the path that exists today purely for nested-stream_events(version="v3")
callers; the test simulates such a caller by injecting a
checkpoint_ns into the config.
"""
graph = _make_two_level_nested()
config = {CONF: {CONFIG_KEY_CHECKPOINT_NS: "outer:abc"}}
run = graph.stream_v2({"value": "x", "items": []}, config=config)
run = graph.stream_events({"value": "x", "items": []}, config=config, version="v3")
payloads = list(run.lifecycle)
# Every emitted lifecycle namespace must extend the caller's scope —
libs/langgraph/tests/test_stream_messages_transformer.py
+26 -26
View File
@@ -1,5 +1,5 @@
"""Tests for MessagesTransformer: protocol event routing, whole-message fallback,
legacy v1 chunk filtering, and end-to-end via stream_v2 / astream_v2."""
legacy v1 chunk filtering, and end-to-end via stream_events(version="v3") / astream_events(version="v3")."""
from __future__ import annotations
@@ -152,7 +152,7 @@ def _lifecycle(
def _simple_graph():
def call_model(state: MessagesState) -> dict[str, Any]:
model = GenericFakeChatModel(messages=iter(["hello world"]))
stream = model.stream_v2(state["messages"])
stream = model.stream_events(state["messages"], version="v3")
return {"messages": stream.output}
return (
@@ -327,7 +327,7 @@ class TestFiltering:
def test_legacy_v1_chunks_ignored(self) -> None:
# v1 AIMessageChunk tuples (from on_llm_new_token) are not streamed
# into this projection; callers must migrate to stream_v2.
# into this projection; callers must migrate to stream_events(version="v3").
t, log = _make_sync_transformer()
t.process(_v1_chunk("hello"))
t.process(_v1_chunk(" world", finish=True))
@@ -477,18 +477,18 @@ class TestViaMux:
# ---------------------------------------------------------------------------
# End-to-end: graph → stream_v2 → run.messages (node calls stream_v2)
# End-to-end: graph → stream_events(version="v3") → run.messages (node calls stream_events)
# ---------------------------------------------------------------------------
class TestEndToEnd:
"""stream_v2 path: node calls model.stream_v2() explicitly."""
"""stream_events(version="v3") path: node calls model.stream_events() explicitly."""
def test_node_calling_stream_v2_populates_messages(self) -> None:
model = GenericFakeChatModel(messages=iter(["hello world"]))
def call_model(state: MessagesState) -> dict[str, Any]:
stream = model.stream_v2(state["messages"])
stream = model.stream_events(state["messages"], version="v3")
return {"messages": stream.output}
graph = (
@@ -499,7 +499,7 @@ class TestEndToEnd:
.compile()
)
run = graph.stream_v2({"messages": "hi"})
run = graph.stream_events({"messages": "hi"}, version="v3")
(stream,) = list(run.messages)
assert isinstance(stream, ChatModelStream)
assert stream.output.text == "hello world"
@@ -509,7 +509,7 @@ class TestEndToEnd:
model = GenericFakeChatModel(messages=iter(["streamed answer"]))
def call_model(state: MessagesState) -> dict[str, Any]:
stream = model.stream_v2(state["messages"])
stream = model.stream_events(state["messages"], version="v3")
return {"messages": stream.output}
graph = (
@@ -520,7 +520,7 @@ class TestEndToEnd:
.compile()
)
run = graph.stream_v2({"messages": "go"})
run = graph.stream_events({"messages": "go"}, version="v3")
(stream,) = list(run.messages)
assert "".join(stream.text) == "streamed answer"
@@ -538,7 +538,7 @@ class TestEndToEnd:
.compile()
)
run = graph.stream_v2({"messages": "hi"})
run = graph.stream_events({"messages": "hi"}, version="v3")
(stream,) = list(run.messages)
assert stream.output.text == "hardcoded"
@@ -547,7 +547,7 @@ class TestEndToEnd:
model = GenericFakeChatModel(messages=iter(["async answer"]))
async def call_model(state: MessagesState) -> dict[str, Any]:
stream = await model.astream_v2(state["messages"])
stream = await model.astream_events(state["messages"], version="v3")
return {"messages": await stream}
graph = (
@@ -558,7 +558,7 @@ class TestEndToEnd:
.compile()
)
run = await graph.astream_v2({"messages": "hi"})
run = await graph.astream_events({"messages": "hi"}, version="v3")
streams = [s async for s in run.messages]
assert len(streams) == 1
assert isinstance(streams[0], AsyncChatModelStream)
@@ -572,7 +572,7 @@ class TestEndToEnd:
model = GenericFakeChatModel(messages=iter(["hello world"]))
async def call_model(state: MessagesState) -> dict[str, Any]:
stream = await model.astream_v2(state["messages"])
stream = await model.astream_events(state["messages"], version="v3")
return {"messages": await stream}
graph = (
@@ -583,7 +583,7 @@ class TestEndToEnd:
.compile()
)
run = await graph.astream_v2({"messages": "hi"})
run = await graph.astream_events({"messages": "hi"}, version="v3")
async def consume() -> list[str]:
collected: list[str] = []
@@ -596,12 +596,12 @@ class TestEndToEnd:
# ---------------------------------------------------------------------------
# End-to-end: graph → stream_v2 → run.messages (node calls invoke)
# End-to-end: graph → stream_events(version="v3") → run.messages (node calls invoke)
# ---------------------------------------------------------------------------
class TestEndToEndV2Invoke:
"""Auto-routing path: stream_v2 injects CONFIG_KEY_STREAM_MESSAGES_V2,
"""Auto-routing path: stream_events(version="v3") injects CONFIG_KEY_STREAM_MESSAGES_V2,
causing BaseChatModel to drive the v2 protocol event generator even for
model.invoke()."""
@@ -620,7 +620,7 @@ class TestEndToEndV2Invoke:
def test_invoke_populates_messages(self) -> None:
run = self._graph(
GenericFakeChatModel(messages=iter(["hello world"]))
).stream_v2({"messages": "hi"})
).stream_events({"messages": "hi"}, version="v3")
(stream,) = list(run.messages)
assert isinstance(stream, ChatModelStream)
assert stream.output.text == "hello world"
@@ -629,7 +629,7 @@ class TestEndToEndV2Invoke:
"""Iterating the stream yields the full v2 lifecycle, not v1 chunks."""
run = self._graph(
GenericFakeChatModel(messages=iter(["streamed answer"]))
).stream_v2({"messages": "go"})
).stream_events({"messages": "go"}, version="v3")
(stream,) = list(run.messages)
events = list(stream)
@@ -650,7 +650,7 @@ class TestEndToEndV2Invoke:
def test_invoke_text_deltas_iterate(self) -> None:
run = self._graph(
GenericFakeChatModel(messages=iter(["delta streaming works"]))
).stream_v2({"messages": "hi"})
).stream_events({"messages": "hi"}, version="v3")
(stream,) = list(run.messages)
assert "".join(stream.text) == "delta streaming works"
@@ -674,7 +674,7 @@ class TestEndToEndV2Invoke:
.compile()
)
streams = list(graph.stream_v2({"messages": "hi"}).messages)
streams = list(graph.stream_events({"messages": "hi"}, version="v3").messages)
assert len(streams) == 2
assert {s.output.text for s in streams} == {"alpha", "beta"}
@@ -698,7 +698,7 @@ class TestEndToEndV2Invoke:
.compile()
)
run = graph.stream_v2({"messages": "hi"})
run = graph.stream_events({"messages": "hi"}, version="v3")
streams = list(run.messages)
assert len(streams) == 2
assert streams[0].node == "streaming_node"
@@ -722,7 +722,7 @@ class TestEndToEndV2Invoke:
.compile()
)
run = await graph.astream_v2({"messages": "hi"})
run = await graph.astream_events({"messages": "hi"}, version="v3")
streams = [s async for s in run.messages]
assert len(streams) == 1
assert isinstance(streams[0], AsyncChatModelStream)
@@ -737,7 +737,7 @@ class TestEndToEndV2Invoke:
class TestDirectMessagesModeStaysV1:
def test_direct_graph_stream_messages_yields_ai_message_chunks(self) -> None:
"""graph.stream(stream_mode="messages") must not leak v2 event dicts —
the v2 flag is only injected by stream_v2 / astream_v2."""
the v2 flag is only injected by stream_events(version="v3") / astream_events(version="v3")."""
model = GenericFakeChatModel(messages=iter(["legacy path"]))
def call_model(state: MessagesState) -> dict[str, Any]:
@@ -760,8 +760,8 @@ class TestDirectMessagesModeStaysV1:
== "legacy path"
)
def test_nested_graph_stream_messages_stays_v1_under_outer_stream_v2(self) -> None:
"""An outer `stream_v2()` run must not flip an inner direct
def test_nested_graph_stream_messages_stays_v1_under_outer_stream_events_v3(self) -> None:
"""An outer `stream_events(version="v3")` run must not flip an inner direct
`stream_mode="messages"` call onto the v2 event protocol."""
model = GenericFakeChatModel(messages=iter(["nested legacy path"]))
@@ -812,7 +812,7 @@ class TestDirectMessagesModeStaysV1:
.compile()
)
result = outer.stream_v2({}).output
result = outer.stream_events({}, version="v3").output
assert result is not None
assert result["saw_only_chunks"] is True
libs/langgraph/tests/test_stream_subgraph_transformer.py
+9 -9
View File
@@ -4,7 +4,7 @@ Subscribes to `tasks` events and produces in-process `SubgraphRunStream`
handles backed by mini-muxes (built via `StreamMux._make_child`). The
synthetic-event tests isolate the inference / mini-mux wiring; the
real-graph tests exercise the end-to-end navigation path through
`stream_v2`.
`stream_events(version="v3")`.
"""
from __future__ import annotations
@@ -93,7 +93,7 @@ def _tasks_result(
def _native_factories() -> list[Any]:
"""Mirror the factory list `Pregel.stream_v2` registers."""
"""Mirror the factory list `Pregel.stream_events(version="v3")` registers."""
return [
ValuesTransformer,
MessagesTransformer,
@@ -761,10 +761,10 @@ def _make_failing_nested() -> Any:
return outer_b.compile()
def test_stream_v2_real_graph_yields_subgraph_handles() -> None:
def test_stream_events_v3_real_graph_yields_subgraph_handles() -> None:
"""Iterating `run.subgraphs` yields handles for direct-child subgraphs."""
graph = _make_two_level_nested()
run = graph.stream_v2({"value": "x", "items": []})
run = graph.stream_events({"value": "x", "items": []}, version="v3")
handle_paths: list[tuple[str, ...]] = []
final_status: dict[tuple[str, ...], str] = {}
@@ -780,10 +780,10 @@ def test_stream_v2_real_graph_yields_subgraph_handles() -> None:
assert final_status[handle_paths[0]] == "completed"
def test_stream_v2_grandchild_visible_on_child_handle() -> None:
def test_stream_events_v3_grandchild_visible_on_child_handle() -> None:
"""Drilling into `handle.subgraphs` surfaces nested grandchildren."""
graph = _make_two_level_nested()
run = graph.stream_v2({"value": "x", "items": []})
run = graph.stream_events({"value": "x", "items": []}, version="v3")
grandchild_paths: list[tuple[str, ...]] = []
middle_path: tuple[str, ...] | None = None
@@ -810,7 +810,7 @@ def test_subgraph_output_stops_at_own_terminal_without_draining_siblings() -> No
inside the loop body misses its events.
"""
graph = _make_two_sibling_subgraphs()
run = graph.stream_v2({"value": "x", "items": []})
run = graph.stream_events({"value": "x", "items": []}, version="v3")
paths: list[tuple[str, ...]] = []
second_values: list[dict[str, Any]] = []
@@ -829,7 +829,7 @@ def test_subgraph_output_stops_at_own_terminal_without_draining_siblings() -> No
def test_aborted_subgraph_handle_does_not_fail_parent_forwarding() -> None:
graph = _make_two_sibling_subgraphs()
run = graph.stream_v2({"value": "x", "items": []})
run = graph.stream_events({"value": "x", "items": []}, version="v3")
seen: list[str | None] = []
for handle in run.subgraphs:
@@ -847,7 +847,7 @@ def test_aborted_subgraph_handle_does_not_fail_parent_forwarding() -> None:
def test_failed_subgraph_output_raises_terminal_error() -> None:
graph = _make_failing_nested()
run = graph.stream_v2({"value": "x", "items": []})
run = graph.stream_events({"value": "x", "items": []}, version="v3")
handle = next(iter(run.subgraphs))
with pytest.raises(RuntimeError, match="child boom"):
libs/prebuilt/tests/test_tool_call_transformer.py
+5 -6
View File
@@ -199,7 +199,7 @@ class TestToolCallTransformerEndToEnd:
}
graph = _build_graph(caller, [streamer])
run = graph.stream_v2({"messages": []}, transformers=[ToolCallTransformer])
run = graph.stream_events({"messages": []}, transformers=[ToolCallTransformer], version="v3")
tool_calls: list[ToolCallStream] = []
for tc in run.tool_calls:
@@ -235,11 +235,11 @@ class TestToolCallTransformerEndToEnd:
# Without ToolCallTransformer, no tool_calls projection is
# exposed and no `tools` events flow through (required_stream_modes
# omits it).
run_no_tc = graph.stream_v2({"messages": []})
run_no_tc = graph.stream_events({"messages": []}, version="v3")
assert "tool_calls" not in run_no_tc._mux.extensions # type: ignore[attr-defined]
# With ToolCallTransformer, the projection is present.
run = graph.stream_v2({"messages": []}, transformers=[ToolCallTransformer])
run = graph.stream_events({"messages": []}, transformers=[ToolCallTransformer], version="v3")
assert "tool_calls" in run._mux.extensions # type: ignore[attr-defined]
# Drain so the run closes cleanly.
list(run.tool_calls)
@@ -266,8 +266,7 @@ class TestToolCallTransformerEndToEnd:
}
graph = _build_graph(caller, [astreamer])
run = await graph.astream_v2(
{"messages": []}, transformers=[ToolCallTransformer]
run = await graph.astream_events({"messages": []}, version="v3", transformers=[ToolCallTransformer]
)
collected: list[ToolCallStream] = []
@@ -296,7 +295,7 @@ class TestToolCallTransformerEndToEnd:
}
graph = _build_graph(caller, [boom])
run = graph.stream_v2({"messages": []}, transformers=[ToolCallTransformer])
run = graph.stream_events({"messages": []}, transformers=[ToolCallTransformer], version="v3")
collected: list[ToolCallStream] = []
with pytest.raises(ValueError, match="nope"):