From 005a9a33cb4fa3fa726cabf0fe9479c65ddae5f5 Mon Sep 17 00:00:00 2001
From: Tat Dat Duong <david@duong.cz>
Date: Mon, 30 Sep 2024 03:14:34 +0200
Subject: [PATCH] Fix tests

---
 libs/langgraph/tests/test_pregel_async.py | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/libs/langgraph/tests/test_pregel_async.py b/libs/langgraph/tests/test_pregel_async.py
index ae86e6bae..5ad973ce4 100644
--- a/libs/langgraph/tests/test_pregel_async.py
+++ b/libs/langgraph/tests/test_pregel_async.py
@@ -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": ()}