@'h? ddlmZmZmZmZddlmZmZddlmZm Z m Z m Z m Z m Z mZmZmZmZmZddlmZmZddlmZddlmZmZddlmZddlmZmZm Z m!Z!m"Z"m#Z#ed e$e%e&Z'ee&e&efZ(d Z)Gd d ed Z*GddeZ+e,e&ee&e$e%ffZ-GddeZ.de.fdZ/de.de.fdZ0ddde.deee&e"fde$dee&de.f dZ1Gdde Z2ede&d dd!d"#Z3ed$e&d%d&d!d"#Z4ed'ee&d(d)dd"#Z5Gd*d+e e'Z6Gd,d-e7Z8d.edee&fd/Z9d.ed0e*de*fd1Z: ed2e!d3ed4e d5iZ;hd6Z/9//J/P/P/RStq!q!&&({S nn_b9>>@  Ts BN)rAchannelsr)rAc tjtjj }||d}n6i}|j D]!\}}||dvr |j ||<#tt||xstt|||d|d|jdgS#t$rYrwxYw)z+Create a checkpoint for the given channels.rCrDrKrErFrT) rrNrrOrPrVrREmptyChannelErrorr?rMr3rrW)rRrZr)rArBvaluesrXr@s r6create_checkpointr^s hll # - - /B,-NN$ DAq #566 LLNq     +UT*+#$67 1 nn_b9 %  sB33 B?>B?c^eZdZUdZeed<eed<eed<dZe eed<dZ e e e ed<y)CheckpointTuplez8A tuple containing a checkpoint and its associated data.configrRmetadataN parent_configpending_writes) r,r-r.r/rr0r?r#rcrrdr PendingWriter4r5r6r`r`s:B   .2M8N+237NHT,/07r5r` thread_idz Thread IDT)rA annotationname descriptiondefault is_shared checkpoint_nsz Checkpoint NSzCheckpoint namespace. Denotes the path to the subgraph node the checkpoint originates from, separated by `|` character, e.g. `"child|grandchild"`. Defaults to "" (root graph). checkpoint_idz Checkpoint IDzHPass to fetch a past checkpoint. If None, fetches the latest checkpoint.c (eZdZUdZeZeed<dddeeddfdZ e de e fdZ dedeefd Zdedeefd Zdddd deed eeeefd eedeedeef dZ dededededef dZ d#dedeeeefdededdf dZdeddfdZdedeefdZdedeefdZ dddd deed eeeefd eedeede!ef dZ"dededededef dZ# d#dedeeeefdededdf dZ$deddfdZ%d ee&d!e'de&fd"Z(y)$BaseCheckpointSaveraBase class for creating a graph checkpointer. Checkpointers allow LangGraph agents to persist their state within and across multiple interactions. Attributes: serde (SerializerProtocol): Serializer for encoding/decoding checkpoints. Note: When creating a custom checkpoint saver, consider implementing async versions to avoid blocking the main thread. serdeN)rqrHc@t|xs |j|_y)N)rrq)selfrqs r6__init__zBaseCheckpointSaver.__init__s -U-@djjA r5c$tttgS)zDefine the configuration options for the checkpoint saver. Returns: list[ConfigurableFieldSpec]: List of configuration field specs. )CheckpointThreadId CheckpointNS CheckpointId)rss r6 config_specsz BaseCheckpointSaver.config_specss#L,??r5racB|j|x}r |jSy)zFetch a checkpoint using the given configuration. Args: config: Configuration specifying which checkpoint to retrieve. Returns: Optional[Checkpoint]: The requested checkpoint, or None if not found. N) get_tuplerRrsravalues r6rWzBaseCheckpointSaver.gets)NN6* *5 *## # +r5ct)arFetch a checkpoint tuple using the given configuration. Args: config: Configuration specifying which checkpoint to retrieve. Returns: Optional[CheckpointTuple]: The requested checkpoint tuple, or None if not found. Raises: NotImplementedError: Implement this method in your custom checkpoint saver. NotImplementedErrorrsras r6r{zBaseCheckpointSaver.get_tuples "!r5)filterbeforelimitrrrct)a List checkpoints that match the given criteria. Args: config: Base configuration for filtering checkpoints. filter: Additional filtering criteria. before: List checkpoints created before this configuration. limit: Maximum number of checkpoints to return. Returns: Iterator[CheckpointTuple]: Iterator of matching checkpoint tuples. Raises: NotImplementedError: Implement this method in your custom checkpoint saver. rrsrarrrs r6listzBaseCheckpointSaver.lists ,"!r5rRrb new_versionsct)aStore a checkpoint with its configuration and metadata. Args: config: Configuration for the checkpoint. checkpoint: The checkpoint to store. metadata: Additional metadata for the checkpoint. new_versions: New channel versions as of this write. Returns: RunnableConfig: Updated configuration after storing the checkpoint. Raises: NotImplementedError: Implement this method in your custom checkpoint saver. rrsrarRrbrs r6putzBaseCheckpointSaver.puts *"!r5r*task_id task_pathct)aStore intermediate writes linked to a checkpoint. Args: config: Configuration of the related checkpoint. writes: List of writes to store. task_id: Identifier for the task creating the writes. task_path: Path of the task creating the writes. Raises: NotImplementedError: Implement this method in your custom checkpoint saver. rrsrar*rrs r6 put_writeszBaseCheckpointSaver.put_writes0s $"!r5rfctzDelete all checkpoints and writes associated with a specific thread ID. Args: thread_id: The thread ID whose checkpoints should be deleted. rrsrfs r6 delete_threadz!BaseCheckpointSaver.delete_threadDs "!r5c^K|j|d{x}r |jSy7w)aAsynchronously fetch a checkpoint using the given configuration. Args: config: Configuration specifying which checkpoint to retrieve. Returns: Optional[Checkpoint]: The requested checkpoint, or None if not found. N) aget_tuplerRr|s r6agetzBaseCheckpointSaver.agetOs3//&11 15 1## # 21s -+-cKtw)aAsynchronously fetch a checkpoint tuple using the given configuration. Args: config: Configuration specifying which checkpoint to retrieve. Returns: Optional[CheckpointTuple]: The requested checkpoint tuple, or None if not found. Raises: NotImplementedError: Implement this method in your custom checkpoint saver. rrs r6rzBaseCheckpointSaver.aget_tuple[s"! cKtw)a1Asynchronously list checkpoints that match the given criteria. Args: config: Base configuration for filtering checkpoints. filter: Additional filtering criteria for metadata. before: List checkpoints created before this configuration. limit: Maximum number of checkpoints to return. Returns: AsyncIterator[CheckpointTuple]: Async iterator of matching checkpoint tuples. Raises: NotImplementedError: Implement this method in your custom checkpoint saver. rrs r6alistzBaseCheckpointSaver.alistis,"!rcKtw)aAsynchronously store a checkpoint with its configuration and metadata. Args: config: Configuration for the checkpoint. checkpoint: The checkpoint to store. metadata: Additional metadata for the checkpoint. new_versions: New channel versions as of this write. Returns: RunnableConfig: Updated configuration after storing the checkpoint. Raises: NotImplementedError: Implement this method in your custom checkpoint saver. rrs r6aputzBaseCheckpointSaver.aputs*"!rcKtw)aAsynchronously store intermediate writes linked to a checkpoint. Args: config: Configuration of the related checkpoint. writes: List of writes to store. task_id: Identifier for the task creating the writes. task_path: Path of the task creating the writes. Raises: NotImplementedError: Implement this method in your custom checkpoint saver. rrs r6 aput_writeszBaseCheckpointSaver.aput_writess$"!rcKtwrrrs r6adelete_threadz"BaseCheckpointSaver.adelete_threads"!rcurrentchannelc>t|trt|y|dzS)aGenerate the next version ID for a channel. Default is to use integer versions, incrementing by 1. If you override, you can use str/int/float versions, as long as they are monotonically increasing. Args: current: The current version identifier (int, float, or str). channel: The channel being versioned. Returns: V: The next version identifier, which must be increasing. ) isinstancer3r)rsrrs r6get_next_versionz$BaseCheckpointSaver.get_next_versions& gs #% % _Q; r5)rg))r,r-r.r/rrqrr0rrtpropertyrrryrr?rWr`r{r r3r r1rr#rGrrrrrrrrrrrrr rrr4r5r6rprps !3 4E 4 /3B*+B  B@d#89@@ $. $Xj-A $ " "8O3L "$,0+/# "("c3h( " ( " } " / ""0"""% " & "  "8 ""sCx)" "  "  "( " "  " $ $HZ4H $ "~ "(?:S "$,0+/# (c3h(  (  }   '2"""% " & "  "8 ""sCx)" "  "  "( " "  " oRSr5rpceZdZdZy)r\ziRaised when attempting to get the value of a channel that hasn't been updated for the first time yet.N)r,r-r.r/r4r5r6r\r\s  r5r\racP|djd|djdS)zKGet checkpoint ID in a backwards-compatible manner (fallback on thread_ts). configurablern thread_ts)rW)ras r6get_checkpoint_idrs/ . ! % %/33K@ r5rbc |j}|jd|jdfD]U}|s|D]K}||vs|tvs|jdr!||}t |t t ttfsG|||<MW|S)z9Get checkpoint metadata in a backwards-compatible manner.rbr__) rUrWEXCLUDED_METADATA_KEYS startswithrr3r1boolfloat)rarbobjkeyr@s r6get_checkpoint_metadatars}}H :& >(BC"  "Ch#)?"?3>>RVCWCA!c3e45 !  "" Or5rJ>rnrmcheckpoint_map)=collections.abcrrrrrrtypingr r r r r rrrrrrlangchain_core.runnablesrrlanggraph.checkpoint.base.idrlanggraph.checkpoint.serde.baserr#langgraph.checkpoint.serde.jsonplusr langgraph.checkpoint.serde.typesrrrrrrr1rr3r rerMr#r9r2rGr?rQrYr^r`rvrwrxrp Exceptionr\rrWRITES_IDX_MAPrr4r5r6rsFF'    K.WB CeS!S#s]# %<5y5sE#sE/223''< *   z $ wsO345     >8j8+   % B   %} Z   C'!*CL  n# &8"YIr62Fr5