Fix get_state

This commit is contained in:
Nuno Campos
2024-12-04 15:39:16 -08:00
parent 09ca964714
commit 2fe38f3940
2 changed files with 3 additions and 0 deletions
@@ -71,6 +71,7 @@ def entrypoint(
channels={START: EphemeralValue(Any), END: LastValue(Any, END)},
input_channels=START,
output_channels=END,
stream_channels=END,
stream_mode="updates",
checkpointer=checkpointer,
store=store,
+2
View File
@@ -2046,6 +2046,8 @@ def test_imp_stream_order(
{"graph": {"a": "0foobarbaz", "c": "something else"}},
]
assert graph.get_state(thread1).values == {"a": "0foobarbaz", "c": "something else"}
@pytest.mark.parametrize("checkpointer_name", ALL_CHECKPOINTERS_SYNC)
def test_send_dedupe_on_resume(