tj0fddlmZddlZddlmZmZmZddlmZmZddl m Z m Z ddl m Z ddlmZmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlm Z m!Z!dZ"ee dge fZ#dIdZ$dJdZ%dKdZ&dLd!Z'dMd"Z(dNd'Z)dOd.Z*ddddd/dPd8Z+dQd=Z,ddd>dRdFZ-ddd>dRdGZ.dSdHZ/dS)T) annotationsN)CallableIterableMapping)datetimetimezone)Anycast)RunnableConfig)BaseCheckpointSaver Checkpoint)uuid6)_DeltaSnapshot)#DELTA_MAX_SUPERSTEPS_SINCE_SNAPSHOT)PUSH)MISSING) BaseChannel) DeltaChannel)ManagedValueMappingManagedValueSpecreturnr cttttdt jt jiiiS)N clock_seq)vidtschannel_valueschannel_versions versions_seen) r LATEST_VERSIONstrrrnowrutc isoformatZ/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/langgraph/pregel/_checkpoint.pyempty_checkpointr+sV   ur""" # # < % % / / 1 1    r)stepinttask_idr$c ttj|d}|dd|dd|dd|dd|d S)aSynthetic task id for exit-mode DeltaChannel writes. Embeds the superstep in the first UUID group so `ORDER BY task_id, idx` preserves chronological order while remaining a valid RFC UUID (required by Postgres `checkpoint_writes.task_id uuid` columns). -08dr)r$uuidUUIDsplit)r,r.partss r*exit_delta_task_idr9'sh  '"" # # ) )# . .E D D Dq D DE!H D DuQx D D%( D DDr)channelsMapping[str, BaseChannel]counters_since_delta_snapshotMapping[str, tuple[int, int]]set[str]c2t}|D]s\}}t|tr|s/||d\}}||jks |tkr||t|S)u7Return the set of DeltaChannel names that should snapshot now. A channel snapshots when EITHER its accumulated update count reaches `snapshot_frequency` OR the total supersteps since its last snapshot reaches `DELTA_MAX_SUPERSTEPS_SINCE_SNAPSHOT`. This is a pure predicate — no mutation. rr) setitems isinstancer is_availablegetsnapshot_frequencyradd)r:r<resultnamechupdates superstepss r*delta_channels_to_snapshotrM2suuFNN$$b"l++ 2??3D3D  ;??fMM r, , ,@@@ JJt    Mr) run_tasks Iterable[Any]cd|DS)zDChannel names written by an update_state superstep (excluding PUSH).cBh|]}|jD]\}}|tk|Sr()writesr).0taskc_s r* z2get_updated_channels_from_tasks..Ns/ I I I$ I I1qDyyAyyyyr)r()rNs r*get_updated_channels_from_tasksrXJs J I) I I IIr)c>d|DS)zFDeltaChannels to snapshot on the first update_state of a fresh thread.cjh|]0\}}t|t|.|1Sr()rCrrD)rSkrJs r*rWz7get_delta_channels_from_all_channels..UsP    Ar b, ' ' -/OO,=,=     r))rB)r:s r*$get_delta_channels_from_all_channelsr\Qs-  ^^%%   r)updated_channels prev_metadataMapping[str, Any] | Nonedict[str, tuple[int, int]]ct|pidpi}i}|D]I\}}t|ts||d\}}|dz }||vr|dz }||f||<J|S)zAdvance ``counters_since_delta_snapshot`` for update_state on a non-fresh thread. Mirrors the per-superstep counter bump in ``_loop._put_checkpoint``. r<r@r2)dictrErBrCr) r:r]r^ prev_counters new_countersch_namerJuss r*$create_metadata_for_update_state_apirh\s  "!!"ABBHbM02L~~'''' "l++    &111 Q & & & FA!"A W r)parentsdict[str, Any]saved_metadatais_fresh_threadbooltuple[set[str], dict[str, Any]]cd||d}|rt||fSt|||}t||}|D]} d|| <d|D} | r| |d<||fS)zEReturn ``(channels_to_snapshot, metadata)`` for an update_state head.update)sourcer,ri)r^r@c&i|]\}}|dk ||S)r@r(rSr[rs r* z?create_checkpoint_plan_for_update_state_api..s#EEEAf1r)r<)r\rhrMrB) r:r]r,rirkrlmetadatardchannels_to_snapshotr[non_zeros r*+create_checkpoint_plan_for_update_state_apirxus  H H3H==xGG7$L 6h MM !!! QEE!3!3!5!5EEEH=4<01  ))r))rr]get_next_versionrv checkpoint Mapping[str, BaseChannel] | Noner str | Noneset[str] | NoneryGetNextVersion | Nonervc htjtj}|p t }||d}|d} ni}t |d} |D]w} | | vr|| } | |vrB||| |vr|| | d| | <t| || <U| } | tur| || <xtt||ptt||| |d|dnt|S)uBuild a new Checkpoint from the previous one and live channel state. For each name in `channels_to_snapshot`, a `_DeltaSnapshot(value)` blob is written into `channel_values[k]`. Other delta channels are omitted from `channel_values` — the ancestor walk reconstructs their state from `checkpoint_writes`. Callers compute the set via `delta_channels_to_snapshot(channels, counters)`; defaults to empty (no snapshots) when not provided. Nr r!rr"rrrr r!r"r])rr%rr&r'rArbrrErzrr r#r$rsorted) rzr:r,rr]ryrvrvaluesr!r[rJrs r*create_checkpointrsl& hl # # - - / /B/8355,-%&89 += >?? " "A(((!B((("$/$,9I0I0I*:*:;KA;NPT*U*U$Q'*2668844q MMOOG## !F1I   +UT***++) 1!1!9vFV?W?W   r)specrstoredobjectcBt|tsdS|tuS)uTrue if `spec` is a `DeltaChannel` and no value is stored at this checkpoint, requiring an ancestor walk to reconstruct. `_DeltaSnapshot` blobs and plain values (migration) resolve directly via `from_checkpoint` — only absence (`MISSING`) triggers replay. F)rCrr)rrs r* _needs_replayrs& dL ) )u W r))saverconfigspecs,Mapping[str, BaseChannel | ManagedValueSpec]rBaseCheckpointSaver | NonerRunnableConfig | None5tuple[Mapping[str, BaseChannel], ManagedValueMapping]ci}i}|D]%\}}t|tr|||< |||<&fd|D}i} |r|||||} i} |D]\}} || vrit t | } | |} | | dt}| | d|}n4| d|t}|| |<| |fS)aHydrate channels from a checkpoint. For most channels, `spec.from_checkpoint(checkpoint["channel_values"][k])` is sufficient. `DeltaChannel` is the exception: when the channel is absent from `channel_values`, an ancestor walk via `saver.get_delta_channel_history` is required to find the nearest seed (`_DeltaSnapshot` blob or pre-migration plain value) and accumulate the writes between it and the target. All delta channels needing replay are batched into a single saver call. c xg|]6\}}t|d|t4|7Sr rrErrSr[rrzs r* z,channels_from_checkpoint..S!!! At z*:;??7KK L L! !!!r)Nrr:seedrRr ) rBrCrget_delta_channel_historyr rfrom_checkpointrEr replay_writesrrzrr channel_specs managed_specsr[rdelta_channels historiesr:r delta_spechistory replay_chrJs ` r*channels_from_checkpointrs"-/M13M !!1 a % % ! M!   M!  !!!!$**,,!!!N $&I %+0B33N4  (*H &&((  4 >>lD11JlG"227;;vw3O3OPPI  # #GH$5 6 6 6BB%%j1A&B&F&Fq'&R&RSSB ] ""r)cKi}i}|D]%\}}t|tr|||< |||<&fd|D}i} |r!|||||d{V} i} |D]\}} || vrit t | } | |} | | dt}| | d|}n4| d|t}|| |<| |fS)zAAsync version of `channels_from_checkpoint`. See docstring there.c xg|]6\}}t|d|t4|7Srrrs r*rz-achannels_from_checkpoint..(rr)NrrrRr ) rBrCraget_delta_channel_historyr rrrErrrs ` r*achannels_from_checkpointrs-/M13M !!1 a % % ! M!   M!  !!!!$**,,!!!N $&I %+0B::N;        (*H &&((  4 >>lD11JlG"227;;vw3O3OPPI  # #GH$5 6 6 6BB%%j1A&B&F&Fq'&R&RSSB ] ""r)c t|d|d|d|d|dd|dD|dd S) Nrrrr r!c>i|]\}}||Sr()copyrss r*rtz#copy_checkpoint..Is&SSStq!q!&&((SSSr)r"r]r)r rrBrE)rzs r*copy_checkpointrBs  S/ d  d !"2388::#$67<<>>SSz//J/P/P/R/RSSS#(:DAA   r))rr )r,r-r.r$rr$)r:r;r<r=rr>)rNrOrr>)r:r;rr>)r:r;r]r>r^r_rr`)r:r;r]r>r,r-rirjrkr_rlrmrrn)rzr r:r{r,r-rr|r]r}ryr~rvr}rr )rrrrrrm) rrrzr rrrrrr)rzr rr )0 __future__rr5collections.abcrrrrrtypingr r langchain_core.runnablesr langgraph.checkpoint.baser r langgraph.checkpoint.base.idr langgraph.checkpoint.serde.typesrlanggraph._internal._configrlanggraph._internal._constantsrlanggraph._internal._typingrlanggraph.channels.baserlanggraph.channels.deltarlanggraph.managed.baserrr#GetNextVersionr+r9rMrXr\rhrxrrrrrr(r)r*rs"""""" 7777777777''''''''333333/.....;;;;;;KKKKKK//////////////////111111HHHHHHHH3+s*+EEEE0JJJJ2****J(,.2,0AAAAAAH     )-$( 0#0#0#0#0#0#n)-$( '#'#'#'#'#'#T      r)