Undo

langchain-ai · Aug 20, 2024 · 84da8e2 · 84da8e2
1 parent 7b4f8ea
commit 84da8e2
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 14 deletions.
diff --git a/libs/checkpoint/langgraph/checkpoint/serde/jsonplus.py b/libs/checkpoint/langgraph/checkpoint/serde/jsonplus.py
@@ -23,7 +23,7 @@
 from zoneinfo import ZoneInfo
 
 from langgraph.checkpoint.serde.base import SerializerProtocol
-from langgraph.checkpoint.serde.types import InterruptProtocol, SendProtocol
+from langgraph.checkpoint.serde.types import SendProtocol
 
 LC_REVIVER = Reviver()
 
@@ -106,11 +106,6 @@ def _default(self, obj):
             return self._encode_constructor_args(
                 obj.__class__, kwargs={"node": obj.node, "arg": obj.arg}
             )
-        elif isinstance(obj, InterruptProtocol):
-            return self._encode_constructor_args(
-                obj.__class__,
-                kwargs={"when": obj.when, "node": obj.node, "value": obj.value},
-            )
         elif isinstance(obj, (bytes, bytearray)):
             return self._encode_constructor_args(
                 obj.__class__, method="fromhex", args=[obj.hex()]

diff --git a/libs/checkpoint/langgraph/checkpoint/serde/types.py b/libs/checkpoint/langgraph/checkpoint/serde/types.py
@@ -2,7 +2,6 @@
     Any,
     AsyncGenerator,
     Generator,
-    Literal,
     Optional,
     Protocol,
     Sequence,
@@ -65,10 +64,3 @@ def __repr__(self) -> str:
 
     def __eq__(self, value: object) -> bool:
         ...
-
-
-@runtime_checkable
-class InterruptProtocol(Protocol):
-    when: Literal["before", "during", "after"]
-    node: str
-    value: Any