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`.
This commit is contained in:
Parker J. Rule
2025-09-25 22:30:19 +00:00
committed by GitHub
parent b0a25f2794
commit 20ddb2b8b4
+1 -1
View File
@@ -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):