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

remove interrupts

Browse Source
This commit is contained in:
vbarda
2024-08-22 15:37:20 -04:00
parent 5654d8fc86
commit 6c7d9c35bc
2 changed files with 30 additions and 100 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libs/langgraph/tests/test_pregel.py
+12 -57
View File
@@ -8179,7 +8179,6 @@ def test_nested_graph_interrupts(
PregelTask(
AnyStr(),
"inner",
interrupts=(Interrupt(when="before", value=None),),
),
),
next=("inner",),
@@ -8213,7 +8212,6 @@ def test_nested_graph_interrupts(
PregelTask(
AnyStr(),
"inner_2",
interrupts=(Interrupt(when="before", value=None),),
),
),
next=("inner_2",),
@@ -8493,7 +8491,6 @@ def test_nested_graph_interrupts(
PregelTask(
AnyStr(),
"inner",
interrupts=(Interrupt(when="before", value=None),),
),
),
next=("inner",),
@@ -8568,7 +8565,6 @@ def test_nested_graph_interrupts(
PregelTask(
AnyStr(),
"inner",
interrupts=(Interrupt(when="before", value=None),),
),
),
next=("inner",),
@@ -8602,7 +8598,6 @@ def test_nested_graph_interrupts(
PregelTask(
AnyStr(),
"inner_2",
interrupts=(Interrupt(when="before", value=None),),
),
),
next=("inner_2",),
@@ -8858,7 +8853,6 @@ def test_nested_graph_interrupts(
PregelTask(
AnyStr(),
"inner",
interrupts=(Interrupt(when="before", value=None),),
),
),
next=("inner",),
@@ -8892,7 +8886,6 @@ def test_nested_graph_interrupts(
PregelTask(
AnyStr(),
name="inner_2",
interrupts=(Interrupt(when="before", value=None),),
),
),
next=("inner_2",),
@@ -9271,9 +9264,7 @@ def test_nested_graph_interrupts(
assert state_history == [
StateSnapshot(
values={"my_key": "hi my value"},
tasks=(
PregelTask(AnyStr(), "inner", interrupts=(Interrupt(when="before"),)),
),
tasks=(PregelTask(AnyStr(), "inner"),),
next=("inner",),
config={
"configurable": {
@@ -9301,11 +9292,7 @@ def test_nested_graph_interrupts(
"my_key": "hi my value here",
"my_other_key": "hi my value",
},
tasks=(
PregelTask(
AnyStr(), "inner_2", interrupts=(Interrupt(when="before"),)
),
),
tasks=(PregelTask(AnyStr(), "inner_2"),),
next=("inner_2",),
config={
"configurable": {
@@ -9389,9 +9376,7 @@ def test_nested_graph_interrupts(
"my_key": "hi my value here",
"my_other_key": "hi my value",
},
tasks=(
PregelTask(AnyStr(), "inner_2", interrupts=(Interrupt(when="before"),)),
),
tasks=(PregelTask(AnyStr(), "inner_2"),),
next=("inner_2",),
config={
"configurable": {
@@ -9440,9 +9425,7 @@ def test_nested_graph_interrupts(
assert list(app.get_state_history(config)) == [
StateSnapshot(
values={"my_key": "hi my value"},
tasks=(
PregelTask(AnyStr(), "inner", interrupts=(Interrupt(when="before"),)),
),
tasks=(PregelTask(AnyStr(), "inner"),),
next=("inner",),
config={
"configurable": {
@@ -9470,11 +9453,7 @@ def test_nested_graph_interrupts(
"my_key": "hi my value here",
"my_other_key": "hi my value",
},
tasks=(
PregelTask(
AnyStr(), "inner_2", interrupts=(Interrupt(when="before"),)
),
),
tasks=(PregelTask(AnyStr(), "inner_2"),),
next=("inner_2",),
config={
"configurable": {
@@ -9506,9 +9485,7 @@ def test_nested_graph_interrupts(
),
StateSnapshot(
values={"my_key": "hi my value"},
tasks=(
PregelTask(AnyStr(), "inner", interrupts=(Interrupt(when="before"),)),
),
tasks=(PregelTask(AnyStr(), "inner"),),
next=("inner",),
config={
"configurable": {
@@ -9536,11 +9513,7 @@ def test_nested_graph_interrupts(
"my_key": "hi my value here",
"my_other_key": "hi my value",
},
tasks=(
PregelTask(
AnyStr(), "inner_2", interrupts=(Interrupt(when="before"),)
),
),
tasks=(PregelTask(AnyStr(), "inner_2"),),
next=("inner_2",),
config={
"configurable": {
@@ -9671,9 +9644,7 @@ def test_nested_graph_interrupts(
),
StateSnapshot(
values={"my_key": "hi my value"},
tasks=(
PregelTask(AnyStr(), "inner", interrupts=(Interrupt(when="before"),)),
),
tasks=(PregelTask(AnyStr(), "inner"),),
next=("inner",),
config={
"configurable": {
@@ -9701,11 +9672,7 @@ def test_nested_graph_interrupts(
"my_key": "hi my value here",
"my_other_key": "hi my value",
},
tasks=(
PregelTask(
AnyStr(), "inner_2", interrupts=(Interrupt(when="before"),)
),
),
tasks=(PregelTask(AnyStr(), "inner_2"),),
next=("inner_2",),
config={
"configurable": {
@@ -10126,11 +10093,7 @@ def test_nested_graph_state(
# test state w/ nested subgraph state (right after interrupt)
assert app.get_state(config) == StateSnapshot(
values={"my_key": "hi my value"},
tasks=(
PregelTask(
AnyStr(), "inner", interrupts=(Interrupt(when="before", value=None),)
),
),
tasks=(PregelTask(AnyStr(), "inner"),),
next=("inner",),
config={
"configurable": {
@@ -10160,7 +10123,6 @@ def test_nested_graph_state(
AnyStr(),
name="inner_2",
error=None,
interrupts=(Interrupt(when="before", value=None),),
),
),
next=("inner_2",),
@@ -10200,7 +10162,6 @@ def test_nested_graph_state(
PregelTask(
AnyStr(),
"inner",
interrupts=(Interrupt(when="before", value=None),),
),
),
next=("inner",),
@@ -10235,7 +10196,6 @@ def test_nested_graph_state(
AnyStr(),
name="inner_2",
error=None,
interrupts=(Interrupt(when="before", value=None),),
),
),
next=("inner_2",),
@@ -10638,7 +10598,7 @@ def test_doubly_nested_graph_state(
app.invoke({"my_key": "my value"}, config, debug=True)
assert app.get_state(config) == StateSnapshot(
values={"my_key": "hi my value"},
tasks=(PregelTask(AnyStr(), "child", interrupts=(Interrupt(when="before"),)),),
tasks=(PregelTask(AnyStr(), "child"),),
next=("child",),
config={
"configurable": {
@@ -10663,11 +10623,7 @@ def test_doubly_nested_graph_state(
subgraph_state_snapshots={
"child": StateSnapshot(
values={"my_key": "hi my value"},
tasks=(
PregelTask(
AnyStr(), "child_1", interrupts=(Interrupt(when="before"),)
),
),
tasks=(PregelTask(AnyStr(), "child_1"),),
next=("child_1",),
config={
"configurable": {
@@ -10692,7 +10648,6 @@ def test_doubly_nested_graph_state(
PregelTask(
AnyStr(),
"grandchild_2",
interrupts=(Interrupt(when="before"),),
),
),
next=("grandchild_2",),
libs/langgraph/tests/test_pregel_async.py
+18 -43
View File
@@ -6801,7 +6801,6 @@ async def test_nested_graph_interrupts(
PregelTask(
AnyStr(),
"inner",
interrupts=(Interrupt(when="before", value=None),),
),
),
next=("inner",),
@@ -6835,7 +6834,6 @@ async def test_nested_graph_interrupts(
PregelTask(
AnyStr(),
"inner_2",
interrupts=(Interrupt(when="before", value=None),),
),
),
next=("inner_2",),
@@ -7120,7 +7118,6 @@ async def test_nested_graph_interrupts(
PregelTask(
AnyStr(),
"inner",
interrupts=(Interrupt(when="before", value=None),),
),
),
next=("inner",),
@@ -7195,7 +7192,6 @@ async def test_nested_graph_interrupts(
PregelTask(
AnyStr(),
"inner",
interrupts=(Interrupt(when="before", value=None),),
),
),
next=("inner",),
@@ -7229,7 +7225,6 @@ async def test_nested_graph_interrupts(
PregelTask(
AnyStr(),
"inner_2",
interrupts=(Interrupt(when="before", value=None),),
),
),
next=("inner_2",),
@@ -7488,7 +7483,6 @@ async def test_nested_graph_interrupts(
PregelTask(
AnyStr(),
"inner",
interrupts=(Interrupt(when="before", value=None),),
),
),
next=("inner",),
@@ -7522,7 +7516,6 @@ async def test_nested_graph_interrupts(
PregelTask(
AnyStr(),
name="inner_2",
interrupts=(Interrupt(when="before", value=None),),
),
),
next=("inner_2",),
@@ -7901,7 +7894,10 @@ async def test_nested_graph_interrupts(
StateSnapshot(
values={"my_key": "hi my value"},
tasks=(
PregelTask(AnyStr(), "inner", interrupts=(Interrupt(when="before"),)),
PregelTask(
AnyStr(),
"inner",
),
),
next=("inner",),
config={
@@ -7932,7 +7928,8 @@ async def test_nested_graph_interrupts(
},
tasks=(
PregelTask(
AnyStr(), "inner_2", interrupts=(Interrupt(when="before"),)
AnyStr(),
"inner_2",
),
),
next=("inner_2",),
@@ -8019,9 +8016,7 @@ async def test_nested_graph_interrupts(
"my_key": "hi my value here",
"my_other_key": "hi my value",
},
tasks=(
PregelTask(AnyStr(), "inner_2", interrupts=(Interrupt(when="before"),)),
),
tasks=(PregelTask(AnyStr(), "inner_2"),),
next=("inner_2",),
config={
"configurable": {
@@ -8070,9 +8065,7 @@ async def test_nested_graph_interrupts(
assert [s async for s in app.aget_state_history(config)] == [
StateSnapshot(
values={"my_key": "hi my value"},
tasks=(
PregelTask(AnyStr(), "inner", interrupts=(Interrupt(when="before"),)),
),
tasks=(PregelTask(AnyStr(), "inner"),),
next=("inner",),
config={
"configurable": {
@@ -8100,11 +8093,7 @@ async def test_nested_graph_interrupts(
"my_key": "hi my value here",
"my_other_key": "hi my value",
},
tasks=(
PregelTask(
AnyStr(), "inner_2", interrupts=(Interrupt(when="before"),)
),
),
tasks=(PregelTask(AnyStr(), "inner_2"),),
next=("inner_2",),
config={
"configurable": {
@@ -8136,9 +8125,7 @@ async def test_nested_graph_interrupts(
),
StateSnapshot(
values={"my_key": "hi my value"},
tasks=(
PregelTask(AnyStr(), "inner", interrupts=(Interrupt(when="before"),)),
),
tasks=(PregelTask(AnyStr(), "inner"),),
next=("inner",),
config={
"configurable": {
@@ -8166,11 +8153,7 @@ async def test_nested_graph_interrupts(
"my_key": "hi my value here",
"my_other_key": "hi my value",
},
tasks=(
PregelTask(
AnyStr(), "inner_2", interrupts=(Interrupt(when="before"),)
),
),
tasks=(PregelTask(AnyStr(), "inner_2"),),
next=("inner_2",),
config={
"configurable": {
@@ -8301,9 +8284,7 @@ async def test_nested_graph_interrupts(
),
StateSnapshot(
values={"my_key": "hi my value"},
tasks=(
PregelTask(AnyStr(), "inner", interrupts=(Interrupt(when="before"),)),
),
tasks=(PregelTask(AnyStr(), "inner"),),
next=("inner",),
config={
"configurable": {
@@ -8331,11 +8312,7 @@ async def test_nested_graph_interrupts(
"my_key": "hi my value here",
"my_other_key": "hi my value",
},
tasks=(
PregelTask(
AnyStr(), "inner_2", interrupts=(Interrupt(when="before"),)
),
),
tasks=(PregelTask(AnyStr(), "inner_2"),),
next=("inner_2",),
config={
"configurable": {
@@ -8763,7 +8740,8 @@ async def test_nested_graph_state(
values={"my_key": "hi my value"},
tasks=(
PregelTask(
AnyStr(), "inner", interrupts=(Interrupt(when="before", value=None),)
AnyStr(),
"inner",
),
),
next=("inner",),
@@ -8795,7 +8773,6 @@ async def test_nested_graph_state(
AnyStr(),
name="inner_2",
error=None,
interrupts=(Interrupt(when="before", value=None),),
),
),
next=("inner_2",),
@@ -8835,7 +8812,6 @@ async def test_nested_graph_state(
PregelTask(
AnyStr(),
"inner",
interrupts=(Interrupt(when="before", value=None),),
),
),
next=("inner",),
@@ -8870,7 +8846,6 @@ async def test_nested_graph_state(
AnyStr(),
name="inner_2",
error=None,
interrupts=(Interrupt(when="before", value=None),),
),
),
next=("inner_2",),
@@ -9273,7 +9248,7 @@ async def test_doubly_nested_graph_state(
await app.ainvoke({"my_key": "my value"}, config, debug=True)
assert await app.aget_state(config) == StateSnapshot(
values={"my_key": "hi my value"},
tasks=(PregelTask(AnyStr(), "child", interrupts=(Interrupt(when="before"),)),),
tasks=(PregelTask(AnyStr(), "child"),),
next=("child",),
config={
"configurable": {
@@ -9300,7 +9275,8 @@ async def test_doubly_nested_graph_state(
values={"my_key": "hi my value"},
tasks=(
PregelTask(
AnyStr(), "child_1", interrupts=(Interrupt(when="before"),)
AnyStr(),
"child_1",
),
),
next=("child_1",),
@@ -9327,7 +9303,6 @@ async def test_doubly_nested_graph_state(
PregelTask(
AnyStr(),
"grandchild_2",
interrupts=(Interrupt(when="before"),),
),
),
next=("grandchild_2",),