tj hdZddlmZddlmZddlmZerddlmZddl m Z m Z GddZ d S) z@Replay state for subgraph checkpoint loading during time-travel.) annotations) TYPE_CHECKING)NS_END)RunnableConfig)BaseCheckpointSaverCheckpointTuplec6eZdZdZdZddZdd ZddZddZdS) ReplayStateaTracks which subgraphs have already loaded their pre-replay checkpoint. During a parent replay, each subgraph's first invocation should restore the checkpoint from before the replay point. Subsequent invocations of the same subgraph (e.g. in a loop) should use normal checkpoint loading so they pick up freshly created checkpoints. The single `ReplayState` instance is shared by reference across all derived configs within one parent execution. ) checkpoint_id _visited_nsr strreturnNonec:||_t|_dS)N)r setr )selfr s Y/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/langgraph/_internal/_replay.py__init__zReplayState.__init__s*&)UU checkpoint_nsboolct|vr!|tddn|}||jvrdS|j|dS)a Return True the first time a subgraph namespace is seen. The task-id suffix is stripped so that the same logical subgraph (e.g. ``"sub_node"``) is recognized across loop iterations even though each iteration has a different task id. rFT)rrsplitr add)rr stable_nss r_is_first_visitzReplayState._is_first_visit"se&&   + +A . .  ( ( (5 Y'''tr checkpointerrcheckpoint_configrCheckpointTuple | Nonec||r)||dd|jiidD]}|cSdS||S)aLLoad the right checkpoint for a subgraph during replay. On the first call for a given subgraph namespace, returns the latest checkpoint created *before* the replay point. On subsequent calls (e.g. the same subgraph in a later loop iteration), falls back to normal latest-checkpoint loading. configurabler rbeforelimitN)rlistr get_tuplerrrrsaveds rget_checkpointzReplayState.get_checkpoint4sy    . . %**!&$:L(MN+    4%%&7888rcK||r/||dd|jiid23d{V}|cS6dS||d{VS)z"Async version of `get_checkpoint`.r"r rr#N)ralistr aget_tupler(s raget_checkpointzReplayState.aget_checkpointKs    . . +11!&$:L(MN 2         e    4!,,->?????????sAN)r r rr)rr rr)rr rrrrrr ) __name__ __module__ __qualname____doc__ __slots__rrr*r.rrr r s{  1I++++ $9999.@@@@@@rr N) r2 __future__rtypingrlanggraph._internal._constantsrlangchain_core.runnablesrlanggraph.checkpoint.baserrr r4rrr:sFF"""""" 111111O777777NNNNNNNNL@L@L@L@L@L@L@L@L@L@r