Fix tests

This commit is contained in:
Tat Dat Duong
2024-09-30 03:14:34 +02:00
parent a8556bf2b8
commit 005a9a33cb
+23
View File
@@ -5933,6 +5933,7 @@ async def test_branch_then(checkpointer_name: str) -> None:
"step": -1,
"writes": {"__start__": {"my_key": "value", "market": "DE"}},
},
"parent_config": None,
"next": ["__start__"],
"tasks": [{"id": AnyStr(), "name": "__start__", "interrupts": ()}],
},
@@ -5963,6 +5964,17 @@ async def test_branch_then(checkpointer_name: str) -> None:
"step": 0,
"writes": None,
},
"parent_config": {
"tags": [],
"metadata": {"thread_id": "11"},
"callbacks": None,
"recursion_limit": 25,
"configurable": {
"thread_id": "11",
"checkpoint_ns": "",
"checkpoint_id": AnyStr(),
},
},
"next": ["prepare"],
"tasks": [{"id": AnyStr(), "name": "prepare", "interrupts": ()}],
},
@@ -6016,6 +6028,17 @@ async def test_branch_then(checkpointer_name: str) -> None:
"step": 1,
"writes": {"prepare": {"my_key": " prepared"}},
},
"parent_config": {
"tags": [],
"metadata": {"thread_id": "11"},
"callbacks": None,
"recursion_limit": 25,
"configurable": {
"thread_id": "11",
"checkpoint_ns": "",
"checkpoint_id": AnyStr(),
},
},
"next": ["tool_two_slow"],
"tasks": [
{"id": AnyStr(), "name": "tool_two_slow", "interrupts": ()}