g3fiWddlmZddlZddlZddlZddlZddlZddlmZddl m Z m Z m Z ddl mZmZmZddlmZddlmZddlmZdd lmZmZmZmZmZmZmZmZm Z ejBe"Z#Gd d ee$eeZ%e%Z&Gd d eZ'y)) annotationsN) defaultdict) AsyncIteratorIteratorSequence)AbstractAsyncContextManagerAbstractContextManager ExitStack) TracebackType)Any)RunnableConfig) WRITES_IDX_MAPBaseCheckpointSaverChannelVersions CheckpointCheckpointMetadataCheckpointTupleSerializerProtocolget_checkpoint_idget_checkpoint_metadataceZdZUdZded<ded<ded<ded  dfd Zdd Z dd Zdd Z ddZ d dZ d!dZ dddd d"dZ d#dZ d$ d%dZd&dZd!dZdddd d'dZ d#dZ d$ d%dZd&dZd(dZxZS)) InMemorySaveraAn in-memory checkpoint saver. This checkpoint saver stores checkpoints in memory using a `defaultdict`. Note: Only use `InMemorySaver` for debugging or testing purposes. For production use cases we recommend installing [langgraph-checkpoint-postgres](https://pypi.org/project/langgraph-checkpoint-postgres/) and using `PostgresSaver` / `AsyncPostgresSaver`. If you are using LangSmith Deployment, no checkpointer needs to be specified. The correct managed checkpointer will be used automatically. Args: serde: The serializer to use for serializing and deserializing checkpoints. Example: ```python import asyncio from langgraph.checkpoint.memory import InMemorySaver from langgraph.graph import StateGraph builder = StateGraph(int) builder.add_node("add_one", lambda x: x + 1) builder.set_entry_point("add_one") builder.set_finish_point("add_one") memory = InMemorySaver() graph = builder.compile(checkpointer=memory) coro = graph.ainvoke(1, {"configurable": {"thread_id": "thread-1"}}) asyncio.run(coro) # Output: 2 ``` z_defaultdict[str, dict[str, dict[str, tuple[tuple[str, bytes], tuple[str, bytes], str | None]]]]storagezadefaultdict[tuple[str, str, str], dict[tuple[str, int], tuple[str, str, tuple[str, bytes], str]]]writesz@dict[tuple[str, str, str, str | int | float], tuple[str, bytes]]blobsN)serdefactoryct|||d|_|t|_||_t |_|turp|jj|j|jj|j|jj|j yy)N)rc ttSN)rdictb/var/www/auto_recruiter/arenv/lib/python3.12/site-packages/langgraph/checkpoint/memory/__init__.pyz(InMemorySaver.__init__..Zs {4'8r#) super__init__rr!rrr stackr enter_context)selfrr __class__s r$r'zInMemorySaver.__init__Ss u%89 dm Y [ + % JJ $ $T\\ 2 JJ $ $T[[ 1 JJ $ $TZZ 0 &r#c:|jj|Sr r( __enter__r*s r$r.zInMemorySaver.__enter__cs  r#c<|jj|||Sr r(__exit__)r*exc_type exc_value tracebacks r$r2zInMemorySaver.__exit__gs zz""8Y BBr#cBK|jj|Swr r-r/s r$ __aenter__zInMemorySaver.__aenter__os  scDK|jj|||Swr r1)r*_InMemorySaver__exc_type_InMemorySaver__exc_value_InMemorySaver__tracebacks r$ __aexit__zInMemorySaver.__aexit__ss! zz"":{KHHs ci}|jD]P\}}||||f}||jvs|j|}|ddk7s3|jj|||<R|S)Nrempty)itemsrr loads_typed) r* thread_id checkpoint_nsversionschannel_valueskvkkvvs r$ _load_blobszInMemorySaver._load_blobs{sy*,NN$ CDAq]Aq1BTZZZZ^a5G#(, (>(>r(BN1%  C r#cL|dd}|djdd}t|x}r|j||j|x}r|\}}}|j|||fj } |j j |} t|i| d|j||| di|j j || D cgc]%\} } } }| | |j j | f'c}} } } |r d|||di Sd Sy|j||x}rt|j}||\}}}|j|||fj } |j j |} td|||dii| d|j||| di|j j || D cgc]%\} } } }| | |j j | f'c}} } } |r d|||di Sd Sycc}} } } wcc}} } } w) a9Get a checkpoint tuple from the in-memory storage. This method retrieves a checkpoint tuple from the in-memory storage based on the provided config. If the config contains a `checkpoint_id` key, the checkpoint with the matching thread ID and timestamp is retrieved. Otherwise, the latest checkpoint for the given thread ID is retrieved. Args: config: The config to use for retrieving the checkpoint. Returns: The retrieved checkpoint tuple, or None if no matching checkpoint was found. configurablerArBrDchannel_versionsrArB checkpoint_idN)config checkpointmetadatapending_writes parent_config) getrrrvaluesrr@rrImaxkeys)r*rPrArBrOsavedrQrRparent_checkpoint_idr checkpoint_idcrF_ checkpointss r$ get_tuplezInMemorySaver.get_tuples / < #N377L -f5 5= 5 Y/ >BB=QQuQ=B: H&:i %NOVVX*.***@*@*L &! % ($*:*:%}kBT6U+  "ZZ33H=NT$$?Jr1aQ 6 6q 9:$0 +-61>1E-*"+ R:#ll95mDD{D #K$4$4$6 7 =H=W: H&:i %NOVVX"jj44Z@ &&)2-:-:) % ($*:*:%}kBT6U+  "ZZ33H=NT$$?Jr1aQ 6 6q 9:$0 +-61>1E-'6"7 E!$H$s *H*Hfilterbeforelimitc#K|r |ddfn |j}|r|djdnd}|r t|nd}|D]}|j|jD]_} || |k7r t |j|| j ddD]#\} \} } } |r| |k7r|rt|x}r| |k\r(|j j| |r#tfd|j Dsh||d kr||d z}|j|| | fj}|j j| }td|| | d ii|d |j|| |d i| rd|| | d ind|Dcgc]%\}}}}|||j j|f'c}}}}&bycc}}}}ww)aList checkpoints from the in-memory storage. This method retrieves a list of checkpoint tuples from the in-memory storage based on the provided 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. Yields: An iterator of matching checkpoint tuples. rKrArBNc |dS)Nrr")xs r$r%z$InMemorySaver.list..s !A$r#T)keyreversec3LK|]\}}|j|k(ywr )rU).0 query_key query_valuerRs r$ z%InMemorySaver.list..s+*2I{$x||I'>>*s!$rrNrDrM)rPrQrRrTrS) rrUrrXsortedr?rr@allrrVrrI)r*rPrbrcrd thread_idsconfig_checkpoint_nsconfig_checkpoint_idrArBrOrQ metadata_brZbefore_checkpoint_idrr[r\r]rFr^rRs @r$listzInMemorySaver.listsc,@Ff^,[9;4<< ;AF> " & & 7t =C08#P I!%i!8!=!=!?O  (4%)== LL+M:@@B&  HM$(, AU0U 5Fv5NN1N)-AA  $zz55jAHc*61>- $)$,d.>.> ) - +,> ?/$"* 4 !/1:5B5I1""&SY((CN2q!QRDJJ$:$:1$=>(9SHO P Z(sFG *G 5Gc |j}|dd}|dd}|jd}|jD]:\} } | |vr|jj || nd|j ||| | f<<|j ||j|d|jj ||jj t|||djdfid|||ddiS) a"Save a checkpoint to the in-memory storage. This method saves a checkpoint to the in-memory storage. The checkpoint is associated with the provided config. Args: config: The config to associate with the checkpoint. checkpoint: The checkpoint to save. metadata: Additional metadata to save with the checkpoint. new_versions: New versions as of this write Returns: RunnableConfig: The updated config containing the saved checkpoint's timestamp. rKrArBrD)r>r#r\rOrN) copypopr?r dumps_typedrrupdaterrU) r*rPrQrR new_versionsr]rArBrVrErFs r$putzInMemorySaver.putFs* OO >*;7 ~.? !"'7!8 &&( DAq56&[ &&vay1n JJ =!Q7 8  Y .554 JJ**1-JJ**+B68+TU>*..?#   &!.!+D!1  r#cf|dd}|djdd}|dd}|||f}|jj|} t|D]^\} \} } |tj| | f} | ddk\r| r| | vr0|| |jj | |f|j|| <`y) aSave a list of writes to the in-memory storage. This method saves a list of writes to the in-memory storage. The writes are associated with the provided config. Args: config: The config to associate with the writes. writes: The writes to save. task_id: Identifier for the task creating the writes. task_path: Path of the task creating the writes. Returns: RunnableConfig: The updated config containing the saved writes' timestamp. rKrArBrLrOrorN)rUr enumeraterrr{)r*rPrtask_id task_pathrArBrO outer_key outer_writes_idxr]rF inner_keys r$ put_writeszInMemorySaver.put_writests*>*;7 ~.22?BG ~.?  }=   2 $V, KC!Q ."4"4Q"<=I|q ]yM7Q &&q) 1DKK "9 - r#c2||jvr |j|=t|jjD]}|d|k(s |j|=t|jjD]}|d|k(s |j|=y)Delete all checkpoints and writes associated with a thread ID. Args: thread_id: The thread ID to delete. Returns: None rN)rrwrrXr)r*rArEs r$ delete_threadzInMemorySaver.delete_threads  $ Y'dkk&&() #Aty KKN #djjoo'( "Aty JJqM "r#c,K|j|Sw)a|Asynchronous version of `get_tuple`. This method is an asynchronous wrapper around `get_tuple` that runs the synchronous method in a separate thread using asyncio. Args: config: The config to use for retrieving the checkpoint. Returns: The retrieved checkpoint tuple, or None if no matching checkpoint was found. )r`)r*rPs r$ aget_tuplezInMemorySaver.aget_tuples~~f%%cJK|j||||D]}| yw)aQAsynchronous version of `list`. This method is an asynchronous wrapper around `list` that runs the synchronous method in a separate thread using asyncio. Args: config: The config to use for listing the checkpoints. Yields: An asynchronous iterator of checkpoint tuples. raN)rw)r*rPrbrcrditems r$alistzInMemorySaver.alists-&IIfVF%IP DJ s!#c2K|j||||Sw)aAsynchronous version of `put`. Args: config: The config to associate with the checkpoint. checkpoint: The checkpoint to save. metadata: Additional metadata to save with the checkpoint. new_versions: New versions as of this write Returns: RunnableConfig: The updated config containing the saved checkpoint's timestamp. )r~)r*rPrQrRr}s r$aputzInMemorySaver.aputs$xx HlCCc2K|j||||Sw)aAsynchronous version of `put_writes`. This method is an asynchronous wrapper around `put_writes` that runs the synchronous method in a separate thread using asyncio. Args: config: The config to associate with the writes. writes: The writes to save, each as a (channel, value) pair. task_id: Identifier for the task creating the writes. task_path: Path of the task creating the writes. Returns: None )r)r*rPrrrs r$ aput_writeszInMemorySaver.aput_writess*vvw BBrc,K|j|Sw)r)r)r*rAs r$adelete_threadzInMemorySaver.adelete_threads!!),,rc|d}n0t|tr|}nt|jdd}|dz}tj}|dd|dS)Nr.ro032016) isinstanceintsplitrandom)r*currentchannel current_vnext_vnext_hs r$get_next_versionzInMemorySaver.get_next_versions\ ?I  %IGMM#.q12IQQvcl++r#)rzSerializerProtocol | Nonerztype[defaultdict]returnNone)rr)r3type[BaseException] | Noner4BaseException | Noner5TracebackType | Noner bool | None)r9rr:rr;rrr)rAstrrBrrCrrzdict[str, Any])rPr rzCheckpointTuple | None) rPRunnableConfig | Nonerbdict[str, Any] | Nonercrrd int | NonerzIterator[CheckpointTuple]) rPr rQrrRrr}rrr )rL) rPr rzSequence[tuple[str, Any]]rrrrrr)rArrr) rPrrbrrcrrdrrzAsyncIterator[CheckpointTuple])rz str | Nonerrrr)__name__ __module__ __qualname____doc____annotations__rr'r.r2r7r<rIr`rwr~rrrrrrrr __classcell__r+s@r$rrsoB   ,0%0 1)1# 1  1 C,C(C( C  CI.I*I* I  I  -0  >   ? ?!LL#8!GH ? ?s5&CB 5C C C C!,C  CC)rr rrrr rr)rr)rr)rr rr)rr rr) rrrrr'rrr.r2rrrrs@r$rrs, (/"N ?r#r)( __future__rloggingrrrr collectionsrcollections.abcrrr contextlibrr r typesr typingr langchain_core.runnablesr langgraph.checkpoint.baserrrrrrrrr getLoggerrrrr MemorySaverrr"r#r$rs" #==UU3      8 $p,46Qp,f F?[F?r#