more test fixes

This commit is contained in:
Sydney Runkle
2025-04-23 16:19:46 -07:00
parent 9f33b4e529
commit 5929b0b08d
2 changed files with 9 additions and 11 deletions
-10
View File
@@ -5468,11 +5468,6 @@ def test_interrupt_task_functional(
resumable=True,
ns=[AnyStr("graph:"), AnyStr("bar:")],
),
Interrupt(
value="Provide value for bar:",
resumable=True,
ns=[AnyStr("graph:"), AnyStr("bar:")],
),
]
}
# Resume with an answer
@@ -5497,11 +5492,6 @@ def test_interrupt_task_functional(
resumable=True,
ns=[AnyStr("graph:"), AnyStr("bar:")],
),
Interrupt(
value="Provide value for bar:",
resumable=True,
ns=[AnyStr("graph:"), AnyStr("bar:")],
),
]
}
# Provide resumes
+9 -1
View File
@@ -6873,7 +6873,15 @@ async def test_interrupt_task_functional(checkpointer_name: str) -> None:
config = {"configurable": {"thread_id": "1"}}
# First run, interrupted at bar
await graph.ainvoke({"a": ""}, config)
assert graph.ainvoke({"a": ""}, config) == {
"__interrupt__": [
Interrupt(
value="Provide value for bar:",
resumable=True,
ns=[AnyStr("graph:"), AnyStr("bar:")],
),
]
}
# Resume with an answer
res = await graph.ainvoke(Command(resume="bar"), config)
assert res == {"a": "foobar"}