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

Fix tests

Browse Source
This commit is contained in:
Tat Dat Duong
2024-10-08 19:18:52 +02:00
parent 7e9cf02922
commit 4f61dd1aa6
2 changed files with 61 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libs/langgraph/tests/test_pregel.py
+9 -6
View File
@@ -6996,11 +6996,14 @@ def test_branch_then(
# test stream_mode=debug
tool_two = tool_two_graph.compile(checkpointer=checkpointer)
thread10 = {"configurable": {"thread_id": "10"}}
assert [
res = [
*tool_two.stream(
{"my_key": "value", "market": "DE"}, thread10, stream_mode="debug"
)
] == [
]
assert res == [
{
"type": "checkpoint",
"timestamp": AnyStr(),
@@ -7026,7 +7029,7 @@ def test_branch_then(
},
"parent_config": None,
"next": ["__start__"],
"tasks": [{"id": AnyStr(), "name": "__start__", "interrupts": ()}],
"tasks": [{"id": AnyStr(), "name": "__start__", "interrupts": (), "state": None}],
},
},
{
@@ -7067,7 +7070,7 @@ def test_branch_then(
},
},
"next": ["prepare"],
"tasks": [{"id": AnyStr(), "name": "prepare", "interrupts": ()}],
"tasks": [{"id": AnyStr(), "name": "prepare", "interrupts": (), "state": None}],
},
},
{
@@ -7131,7 +7134,7 @@ def test_branch_then(
},
},
"next": ["tool_two_slow"],
"tasks": [{"id": AnyStr(), "name": "tool_two_slow", "interrupts": ()}],
"tasks": [{"id": AnyStr(), "name": "tool_two_slow", "interrupts": (), "state": None}],
},
},
{
@@ -7195,7 +7198,7 @@ def test_branch_then(
},
},
"next": ["finish"],
"tasks": [{"id": AnyStr(), "name": "finish", "interrupts": ()}],
"tasks": [{"id": AnyStr(), "name": "finish", "interrupts": (), "state": None}],
},
},
{
libs/langgraph/tests/test_pregel_async.py
+52 -7
View File
@@ -5655,7 +5655,14 @@ async def test_branch_then(checkpointer_name: str) -> None:
},
"parent_config": None,
"next": ["__start__"],
"tasks": [{"id": AnyStr(), "name": "__start__", "interrupts": ()}],
"tasks": [
{
"id": AnyStr(),
"name": "__start__",
"interrupts": (),
"state": None,
}
],
},
},
{
@@ -5696,7 +5703,14 @@ async def test_branch_then(checkpointer_name: str) -> None:
},
},
"next": ["prepare"],
"tasks": [{"id": AnyStr(), "name": "prepare", "interrupts": ()}],
"tasks": [
{
"id": AnyStr(),
"name": "prepare",
"interrupts": (),
"state": None,
}
],
},
},
{
@@ -5761,7 +5775,12 @@ async def test_branch_then(checkpointer_name: str) -> None:
},
"next": ["tool_two_slow"],
"tasks": [
{"id": AnyStr(), "name": "tool_two_slow", "interrupts": ()}
{
"id": AnyStr(),
"name": "tool_two_slow",
"interrupts": (),
"state": None,
}
],
},
},
@@ -5826,7 +5845,14 @@ async def test_branch_then(checkpointer_name: str) -> None:
},
},
"next": ["finish"],
"tasks": [{"id": AnyStr(), "name": "finish", "interrupts": ()}],
"tasks": [
{
"id": AnyStr(),
"name": "finish",
"interrupts": (),
"state": None,
}
],
},
},
{
@@ -5937,7 +5963,14 @@ async def test_branch_then(checkpointer_name: str) -> None:
},
"parent_config": None,
"next": ["__start__"],
"tasks": [{"id": AnyStr(), "name": "__start__", "interrupts": ()}],
"tasks": [
{
"id": AnyStr(),
"name": "__start__",
"interrupts": (),
"state": None,
}
],
},
},
{
@@ -5978,7 +6011,14 @@ async def test_branch_then(checkpointer_name: str) -> None:
},
},
"next": ["prepare"],
"tasks": [{"id": AnyStr(), "name": "prepare", "interrupts": ()}],
"tasks": [
{
"id": AnyStr(),
"name": "prepare",
"interrupts": (),
"state": None,
}
],
},
},
{
@@ -6043,7 +6083,12 @@ async def test_branch_then(checkpointer_name: str) -> None:
},
"next": ["tool_two_slow"],
"tasks": [
{"id": AnyStr(), "name": "tool_two_slow", "interrupts": ()}
{
"id": AnyStr(),
"name": "tool_two_slow",
"interrupts": (),
"state": None,
}
],
},
},