From f93512e3b33346e67880bb55e3a2649147ca64cf Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Thu, 22 Aug 2024 12:33:25 -0700 Subject: [PATCH] Remove unused when values for Interrupt --- libs/langgraph/langgraph/constants.py | 4 ++-- libs/langgraph/langgraph/errors.py | 2 +- libs/langgraph/tests/test_pregel.py | 2 +- libs/langgraph/tests/test_pregel_async.py | 2 +- 4 files changed, 5 insertions(+), 5 deletions(-) diff --git a/libs/langgraph/langgraph/constants.py b/libs/langgraph/langgraph/constants.py index 51b6e1437..21844a565 100644 --- a/libs/langgraph/langgraph/constants.py +++ b/libs/langgraph/langgraph/constants.py @@ -102,5 +102,5 @@ class Send: @dataclass class Interrupt: - when: Literal["before", "during", "after"] - value: Any = None + value: Any + when: Literal["during"] = "during" diff --git a/libs/langgraph/langgraph/errors.py b/libs/langgraph/langgraph/errors.py index 27a7689fa..25ba93a72 100644 --- a/libs/langgraph/langgraph/errors.py +++ b/libs/langgraph/langgraph/errors.py @@ -40,7 +40,7 @@ class NodeInterrupt(GraphInterrupt): """Raised by a node to interrupt execution.""" def __init__(self, value: Any) -> None: - super().__init__([Interrupt("during", value)]) + super().__init__([Interrupt(value)]) class EmptyInputError(Exception): diff --git a/libs/langgraph/tests/test_pregel.py b/libs/langgraph/tests/test_pregel.py index 56e7ed5a3..015bba9a6 100644 --- a/libs/langgraph/tests/test_pregel.py +++ b/libs/langgraph/tests/test_pregel.py @@ -6383,7 +6383,7 @@ def test_dynamic_interrupt(snapshot: SnapshotAssertion) -> None: PregelTask( AnyStr(), "tool_two", - interrupts=(Interrupt("during", "Just because..."),), + interrupts=(Interrupt("Just because..."),), ), ), config=tool_two.checkpointer.get_tuple(thread1).config, diff --git a/libs/langgraph/tests/test_pregel_async.py b/libs/langgraph/tests/test_pregel_async.py index 0308deb54..26bea855a 100644 --- a/libs/langgraph/tests/test_pregel_async.py +++ b/libs/langgraph/tests/test_pregel_async.py @@ -285,7 +285,7 @@ async def test_dynamic_interrupt( PregelTask( AnyStr(), "tool_two", - interrupts=(Interrupt("during", "Just because..."),), + interrupts=(Interrupt("Just because..."),), ), ), config=tup.config,