mirror of
https://github.com/langchain-ai/langgraph.git
synced 2026-09-06 17:57:49 +02:00
Nuno Campos
4e8fbe4525
- Instead store sends in a Topic channel, removing the need to fetch sends as writes against the parent checkpoint - Remove deprecated/unused functions in langgraph-checkpoint (will require bumping min range for langgraph-checkpoint in langgraph lib) - Implement migration of old pending sends in langgraph-checkpoint-postgres - Ensure parent config of `checkpoint_during=False` checkpoints always points to checkpoints that were also saved
68 lines
1.8 KiB
Python
68 lines
1.8 KiB
Python
from langgraph.constants import PULL, PUSH
|
|
from langgraph.pregel.algo import prepare_next_tasks, task_path_str
|
|
from langgraph.pregel.checkpoint import channels_from_checkpoint, empty_checkpoint
|
|
|
|
|
|
def test_prepare_next_tasks() -> None:
|
|
config = {}
|
|
processes = {}
|
|
checkpoint = empty_checkpoint()
|
|
channels, managed = channels_from_checkpoint({}, checkpoint)
|
|
|
|
assert (
|
|
prepare_next_tasks(
|
|
checkpoint,
|
|
{},
|
|
processes,
|
|
channels,
|
|
managed,
|
|
config,
|
|
0,
|
|
-1,
|
|
for_execution=False,
|
|
)
|
|
== {}
|
|
)
|
|
assert (
|
|
prepare_next_tasks(
|
|
checkpoint,
|
|
{},
|
|
processes,
|
|
channels,
|
|
managed,
|
|
config,
|
|
0,
|
|
-1,
|
|
for_execution=True,
|
|
checkpointer=None,
|
|
store=None,
|
|
manager=None,
|
|
)
|
|
== {}
|
|
)
|
|
|
|
# TODO: add more tests
|
|
|
|
|
|
def test_tuple_str() -> None:
|
|
push_path_a = (PUSH, 2)
|
|
pull_path_a = (PULL, "abc")
|
|
push_path_b = (PUSH, push_path_a, 1)
|
|
push_path_c = (PUSH, push_path_b, 3)
|
|
|
|
assert task_path_str(push_path_a) == f"~{PUSH}, 0000000002"
|
|
assert task_path_str(push_path_b) == f"~{PUSH}, ~{PUSH}, 0000000002, 0000000001"
|
|
assert (
|
|
task_path_str(push_path_c)
|
|
== f"~{PUSH}, ~{PUSH}, ~{PUSH}, 0000000002, 0000000001, 0000000003"
|
|
)
|
|
assert task_path_str(pull_path_a) == f"~{PULL}, abc"
|
|
|
|
path_list = [push_path_b, push_path_a, pull_path_a, push_path_c]
|
|
assert sorted(map(task_path_str, path_list)) == [
|
|
f"~{PULL}, abc",
|
|
f"~{PUSH}, 0000000002",
|
|
f"~{PUSH}, ~{PUSH}, 0000000002, 0000000001",
|
|
f"~{PUSH}, ~{PUSH}, ~{PUSH}, 0000000002, 0000000001, 0000000003",
|
|
]
|