From 20ddb2b8b48be172a64a153513a5b370dace3c19 Mon Sep 17 00:00:00 2001
From: "Parker J. Rule" <pjrule@users.noreply.github.com>
Date: Thu, 25 Sep 2025 18:30:19 -0400
Subject: [PATCH] fix(langgraph): `CheckpointTask.state` can be a
 `StateSnapshot` (#6201)

This adds `StateSnapshot` to the union type annotation of
`CheckpointTask.state`.

The annotation was previously incomplete: `map_debug_checkpoint()`
generates `CheckpointPayload` objects from `PregelTask` objects, and the
`state` field in `PregelTask` is of type `None | RunnableConfig |
StateSnapshot`.
---
 libs/langgraph/langgraph/pregel/debug.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/libs/langgraph/langgraph/pregel/debug.py b/libs/langgraph/langgraph/pregel/debug.py
index 1591431d7..507fd587a 100644
--- a/libs/langgraph/langgraph/pregel/debug.py
+++ b/libs/langgraph/langgraph/pregel/debug.py
@@ -48,7 +48,7 @@ class CheckpointTask(TypedDict):
     name: str
     error: str | None
     interrupts: list[dict]
-    state: RunnableConfig | None
+    state: StateSnapshot | RunnableConfig | None
 
 
 class CheckpointPayload(TypedDict):