WEBDIENSTE/langgraph
1
0
Fork 0
mirror of https://github.com/langchain-ai/langgraph.git synced 2026-08-17 21:25:46 +02:00
Code Issues Packages Projects Releases Wiki Activity

Update

Browse Source
This commit is contained in:
Nuno Campos
2025-01-16 13:44:19 -08:00
parent 86b6afc982
commit b98a7b09a3
2 changed files with 89 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libs/langgraph/tests/__snapshots__/test_pregel.ambr
+85
View File
@@ -3627,6 +3627,91 @@
'''
# ---
# name: test_interrupt_functional[postgres]
'''
%%{init: {'flowchart': {'curve': 'linear'}}}%%
graph TD;
__graph(graph)
__bar(bar)
__foo(foo)
__graph -.-> __bar;
__bar --> __graph;
__graph -.-> __foo;
__foo --> __graph;
classDef default fill:#f2f0ff,line-height:1.2
classDef first fill-opacity:0
classDef last fill:#bfb6fc
'''
# ---
# name: test_interrupt_functional[postgres_pipe]
'''
%%{init: {'flowchart': {'curve': 'linear'}}}%%
graph TD;
__graph(graph)
__bar(bar)
__foo(foo)
__graph -.-> __bar;
__bar --> __graph;
__graph -.-> __foo;
__foo --> __graph;
classDef default fill:#f2f0ff,line-height:1.2
classDef first fill-opacity:0
classDef last fill:#bfb6fc
'''
# ---
# name: test_interrupt_functional[postgres_pool]
'''
%%{init: {'flowchart': {'curve': 'linear'}}}%%
graph TD;
__graph(graph)
__bar(bar)
__foo(foo)
__graph -.-> __bar;
__bar --> __graph;
__graph -.-> __foo;
__foo --> __graph;
classDef default fill:#f2f0ff,line-height:1.2
classDef first fill-opacity:0
classDef last fill:#bfb6fc
'''
# ---
# name: test_interrupt_functional[postgres_shallow]
'''
%%{init: {'flowchart': {'curve': 'linear'}}}%%
graph TD;
__graph(graph)
__bar(bar)
__foo(foo)
__graph -.-> __bar;
__bar --> __graph;
__graph -.-> __foo;
__foo --> __graph;
classDef default fill:#f2f0ff,line-height:1.2
classDef first fill-opacity:0
classDef last fill:#bfb6fc
'''
# ---
# name: test_interrupt_functional[sqlite]
'''
%%{init: {'flowchart': {'curve': 'linear'}}}%%
graph TD;
__graph(graph)
__bar(bar)
__foo(foo)
__graph -.-> __bar;
__bar --> __graph;
__graph -.-> __foo;
__foo --> __graph;
classDef default fill:#f2f0ff,line-height:1.2
classDef first fill-opacity:0
classDef last fill:#bfb6fc
'''
# ---
# name: test_interrupt_task_functional[memory]
'''
%%{init: {'flowchart': {'curve': 'linear'}}}%%
libs/langgraph/tests/test_pregel.py
+4 -8
View File
@@ -1546,8 +1546,7 @@ def test_imp_stream_order(
fut_baz = baz(fut_bar.result())
return fut_baz.result()
if checkpointer_name == "memory":
assert graph.get_graph().draw_mermaid() == snapshot
assert graph.get_graph().draw_mermaid() == snapshot
thread1 = {"configurable": {"thread_id": "1"}}
assert [c for c in graph.stream({"a": "0"}, thread1)] == [
@@ -5009,8 +5008,7 @@ def test_interrupt_task_functional(
fut_bar = bar(fut_foo.result())
return fut_bar.result()
if checkpointer_name == "memory":
assert graph.get_graph().draw_mermaid() == snapshot
assert graph.get_graph().draw_mermaid() == snapshot
config = {"configurable": {"thread_id": "1"}}
# First run, interrupted at bar
@@ -5456,8 +5454,7 @@ def test_falsy_return_from_task(
falsy_task().result()
interrupt("test")
if checkpointer_name == "memory":
assert graph.get_graph().draw_mermaid() == snapshot
assert graph.get_graph().draw_mermaid() == snapshot
configurable = {"configurable": {"thread_id": str(uuid.uuid4())}}
graph.invoke({"a": 5}, configurable)
@@ -5491,8 +5488,7 @@ def test_multiple_interrupts_imperative(
return {"values": values}
if checkpointer_name == "memory":
assert graph.get_graph().draw_mermaid() == snapshot
assert graph.get_graph().draw_mermaid() == snapshot
configurable = {"configurable": {"thread_id": str(uuid.uuid4())}}
graph.invoke({}, configurable)