WEBDIENSTE/langgraph
1
0
Fork 0
mirror of https://github.com/langchain-ai/langgraph.git synced 2026-09-06 01:37:49 +02:00
Code Issues Packages Projects Releases Wiki Activity

sdk too

Browse Source
This commit is contained in:
Sydney Runkle
2026-03-03 14:24:30 -08:00
parent 727a9a1f6c
commit 33289349a9
3 changed files with 15 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libs/langgraph/tests/test_stream_v2.py
+5 -2
View File
@@ -133,11 +133,14 @@ def _assert_stream_part_shape(part: StreamPart[Any, Any]) -> None:
class TestV1BackwardsCompat:
def test_stream_default_is_v1(self) -> None:
def test_stream_default_is_v2(self) -> None:
graph = _make_simple_graph().compile()
chunks = list(graph.stream(_SIMPLE_INPUT))
for chunk in chunks:
assert isinstance(chunk, dict)
assert "type" in chunk
assert "ns" in chunk
assert "data" in chunk
def test_stream_v1_updates_mode(self) -> None:
graph = _make_simple_graph().compile()
@@ -1082,7 +1085,7 @@ class TestV2TypeSafeStreaming:
f"Expected OuterState, got {type(c['data'])}"
)
# Subgraph values are streamed from the subgraph's own stream()
# which runs with default stream_version="v1", so no coercion
# which runs with default stream_version="v2", so coercion is applied
sub_values = [c for c in chunks if c["type"] == "values" and c["ns"] != ()]
assert len(sub_values) >= 1
libs/sdk-py/langgraph_sdk/_async/runs.py
+5 -5
View File
@@ -96,7 +96,7 @@ class RunsClient:
headers: Mapping[str, str] | None = None,
params: QueryParamTypes | None = None,
on_run_created: Callable[[RunCreateMetadata], None] | None = None,
stream_version: Literal["v1"] = "v1",
stream_version: Literal["v1"],
) -> AsyncIterator[StreamPart]: ...
@overload
@@ -127,7 +127,7 @@ class RunsClient:
headers: Mapping[str, str] | None = None,
params: QueryParamTypes | None = None,
on_run_created: Callable[[RunCreateMetadata], None] | None = None,
stream_version: Literal["v2"],
stream_version: Literal["v2"] = "v2",
) -> AsyncIterator[StreamPartV2]: ...
@overload
@@ -155,7 +155,7 @@ class RunsClient:
headers: Mapping[str, str] | None = None,
params: QueryParamTypes | None = None,
on_run_created: Callable[[RunCreateMetadata], None] | None = None,
stream_version: Literal["v1"] = "v1",
stream_version: Literal["v1"],
) -> AsyncIterator[StreamPart]: ...
@overload
@@ -183,7 +183,7 @@ class RunsClient:
headers: Mapping[str, str] | None = None,
params: QueryParamTypes | None = None,
on_run_created: Callable[[RunCreateMetadata], None] | None = None,
stream_version: Literal["v2"],
stream_version: Literal["v2"] = "v2",
) -> AsyncIterator[StreamPartV2]: ...
def stream(
@@ -215,7 +215,7 @@ class RunsClient:
params: QueryParamTypes | None = None,
on_run_created: Callable[[RunCreateMetadata], None] | None = None,
durability: Durability | None = None,
stream_version: StreamVersion = "v1",
stream_version: StreamVersion = "v2",
) -> AsyncIterator[StreamPart | StreamPartV2]:
"""Create a run and stream the results.
libs/sdk-py/langgraph_sdk/_sync/runs.py
+5 -5
View File
@@ -95,7 +95,7 @@ class SyncRunsClient:
headers: Mapping[str, str] | None = None,
params: QueryParamTypes | None = None,
on_run_created: Callable[[RunCreateMetadata], None] | None = None,
stream_version: Literal["v1"] = "v1",
stream_version: Literal["v1"],
) -> Iterator[StreamPart]: ...
@overload
@@ -125,7 +125,7 @@ class SyncRunsClient:
headers: Mapping[str, str] | None = None,
params: QueryParamTypes | None = None,
on_run_created: Callable[[RunCreateMetadata], None] | None = None,
stream_version: Literal["v2"],
stream_version: Literal["v2"] = "v2",
) -> Iterator[StreamPartV2]: ...
@overload
@@ -154,7 +154,7 @@ class SyncRunsClient:
headers: Mapping[str, str] | None = None,
params: QueryParamTypes | None = None,
on_run_created: Callable[[RunCreateMetadata], None] | None = None,
stream_version: Literal["v1"] = "v1",
stream_version: Literal["v1"],
) -> Iterator[StreamPart]: ...
@overload
@@ -183,7 +183,7 @@ class SyncRunsClient:
headers: Mapping[str, str] | None = None,
params: QueryParamTypes | None = None,
on_run_created: Callable[[RunCreateMetadata], None] | None = None,
stream_version: Literal["v2"],
stream_version: Literal["v2"] = "v2",
) -> Iterator[StreamPartV2]: ...
def stream(
@@ -215,7 +215,7 @@ class SyncRunsClient:
params: QueryParamTypes | None = None,
on_run_created: Callable[[RunCreateMetadata], None] | None = None,
durability: Durability | None = None,
stream_version: StreamVersion = "v1",
stream_version: StreamVersion = "v2",
) -> Iterator[StreamPart | StreamPartV2]:
"""Create a run and stream the results.