iWSSKJr SSKrSSKrSSKrSSKrSSKrSSKJr SSK J r J r J r SSK JrJrJr SSKJr SSKJr SSKJr SS KJrJrJrJrJrJrJrJrJ r \RB"\"5r#"S S \\$\\5r%\%r&"S S \5r'g)) annotationsN) defaultdict) AsyncIteratorIteratorSequence)AbstractAsyncContextManagerAbstractContextManager ExitStack) TracebackType)Any)RunnableConfig) WRITES_IDX_MAPBaseCheckpointSaverChannelVersions CheckpointCheckpointMetadataCheckpointTupleSerializerProtocolget_checkpoint_idget_checkpoint_metadatac^\rSrSr%SrS\S'S\S'S\S'S \S .SU4S jjjrSS jrS S jr SSjr S!Sjr S"Sjr S#Sjr S S S S.S$SjjrS%SjrS&S'SjjrS(SjrS#SjrS S S S.S)SjjrS%SjrS&S'SjjrS(SjrS*SjrSrU=r$)+ InMemorySaveraaAn 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. Examples: 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)serdefactoryc>[TU]US9 U"S5UlU"[5UlU"5Ul[ 5UlU[LapURRUR5 URRUR5 URRUR 5 gg)N)rc [[5$N)rdict^/home/james-whalen/.local/lib/python3.13/site-packages/langgraph/checkpoint/memory/__init__.py(InMemorySaver.__init__..Ys {4'8r$) super__init__rr"rrr stackr enter_context)selfrr __class__s r%r)InMemorySaver.__init__Rs u%89 dm Y [ + % JJ $ $T\\ 2 JJ $ $T[[ 1 JJ $ $TZZ 0 &r$c6URR5$r!r* __enter__r,s r%r1InMemorySaver.__enter__bszz##%%r$c:URRXU5$r!r*__exit__)r,exc_type exc_value tracebacks r%r6InMemorySaver.__exit__es zz""8 BBr$c># URR5$7fr!r0r2s r% __aenter__InMemorySaver.__aenter__mszz##%%scB# URRXU5$7fr!r5)r,_InMemorySaver__exc_type_InMemorySaver__exc_value_InMemorySaver__tracebacks r% __aexit__InMemorySaver.__aexit__ps zz"":KHHsc0nUR5HQupVXXV4nXpR;dMURUnUSS:wdM4URRU5XE'MS U$)Nrempty)itemsrr loads_typed) r, thread_id checkpoint_nsversionschannel_valueskvkkvvs r% _load_blobsInMemorySaver._load_blobsxsh*,NN$DAA1BZZZZ^a5G#(, (>(>r(BN% % r$cXUSSnUSRSS5n[U5=n(aURUURU5=n(aUupgnURX#U4R 5n UR R U5n [U0U ESURX#U S50EUR R U5U V V V Vs/sH#uppXUR R U 54PM% snn n n U(a SUUUS.0S 9$SS 9$gURUU=n(a[UR55nXupgnURX#U4R 5n UR R U5n [SUUUS.00U ESURX#U S50EUR R U5U V V V Vs/sH#uppXUR R U 54PM% snn n n U(a SUUUS.0S 9$SS 9$gs snn n n fs snn n n f) aGet 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. configurablerHrIrKchannel_versionsrHrI checkpoint_idN)config checkpointmetadatapending_writes parent_config) getrrrvaluesrrGrrPmaxkeys)r,rXrHrIrWsavedrYrZparent_checkpoint_idr checkpoint_idcrM_ checkpointss r% get_tupleInMemorySaver.get_tuples / < #N377L -f5 5= 5 Y/ >BB=QQuQ=B: &:i %NOVVX*.***@*@*L &! % ($*:*:%kBT6U+  "ZZ33H=NT$NT{ra 6 6q 9:f$0 +-61>1E-*"+ R:#ll95mDD{D #K$4$4$6 7 =H=W: &:i %NOVVX"jj44Z@ &&)2-:-:) % ($*:*:%kBT6U+  "ZZ33H=NT$NT{ra 6 6q 9:f$0 +-61>1E-'6"7 E!$H$s *H*H$filterbeforelimitc#^# U(a USS4O URnU(aUSRS5OSnU(a [U5OSnUGHnURUR5GHn UbX:waM[ URUU R 5SSS9GHIun un n n U(aX:waMU(a[U5=n(aX:aM9UR RU 5mU(a*[U4SjUR 555(dMUb US ::a MUbUS -nURXU 4R5nUR RU 5n[SUU U S .00UES URUU US 50ETU (aSUU U S .0OSUVVVVs/sH&unnnnUUUR RU54PM( snnnnS9v GML GM GM gs snnnnf7f)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. rSrHrINc US$)Nrr#)xs r%r&$InMemorySaver.list..s!A$r$T)keyreversec3P># UHupUTRU5:Hv M g7fr!)r]).0 query_key query_valuerZs r% %InMemorySaver.list..s)*6D2I$x||I'>>6Ds#&rrVrKrU)rXrYrZr\r[) rr]rr`sortedrFrrGallrr^rrP)r,rXrkrlrm thread_idsconfig_checkpoint_nsconfig_checkpoint_idrHrIrWrY metadata_brbbefore_checkpoint_idrrcrdrerMrfrZs @r%listInMemorySaver.listsH,@Ff^,[9;4<< ;AF> " & & 7t =C08#I!%i!8!=!=!? (4%= LL+M:@@B&  M$(, 0U 5Fv5NN1N)A  $zz55jAHc*61>- $)$,d.>.> ) - +,> ?/$"* 4 !/1:5B5I1""&SY(RX;2q!QRDJJ$:$:1$=>RX(9K"@$Z(sF;H >-H+"H c UR5nUSSnUSSnURS5nUR5H8upX;aURR X5OSUR XgX4'M: UR UURUSURR U5URR [X55USRS5405 SUUUSS.0$) aSave 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. rSrHrIrK)rEr$rdrWrV) copypoprFr dumps_typedrrupdaterr]) r,rXrYrZ new_versionsrerHrIr^rLrMs r%putInMemorySaver.putCs* OO >*;7 ~.? !"'7!8 &&(DA56[ &&vy1n JJ !7 8) Y .554 JJ**1-JJ**+B6+TU>*..?#   &!.!+D!1  r$cvUSSnUSRSS5nUSSnXVU4nURRU5n [U5Hfun upU[R"X54n U SS:aU (aX;aM7UU URR U 5U4URUU 'Mh g) 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. rSrHrIrTrWrzrN)r]r enumeraterrr)r,rXrtask_id task_pathrHrIrW outer_key outer_writes_idxrerM inner_keys r% put_writesInMemorySaver.put_writesqs*>*;7 ~.22?BG ~.? }=   2 $V,KC! ."4"4Q"<=I|q ]y7Q &&q) 1DKK "9 - -r$cDXR;a URU [URR55HnUSU:XdMURU M [URR55HnUSU:XdMURU M g)Delete all checkpoints and writes associated with a thread ID. Args: thread_id: The thread ID to delete. Returns: None rN)rrrr`r)r,rHrLs r% delete_threadInMemorySaver.delete_threads  $ Y'dkk&&()Aty KKN*djjoo'(Aty JJqM)r$c,# URU5$7f)aDAsynchronous 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. )rh)r,rXs r% aget_tupleInMemorySaver.aget_tuples~~f%%cD# URXX4S9HnU7v M g7f)aAsynchronous 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. rjN)r)r,rXrkrlrmitems r%alistInMemorySaver.alists$&IIfFIPDJQs c.# URXX45$7f)aTAsynchronous 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,rXrYrZrs r%aputInMemorySaver.aputs$xxHCCc.# URXX45$7f)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,rXrrrs r% aput_writesInMemorySaver.aput_writess*vwBBrc,# URU5$7f)r)r)r,rHs r%adelete_threadInMemorySaver.adelete_threads!!),,rcUcSnO5[U[5(aUnO[URS5S5nUS-n[R"5nUSSUS3$)Nr.rz032016) isinstanceintsplitrandom)r,currentchannel current_vnext_vnext_hs r%get_next_versionInMemorySaver.get_next_versions_ ?I  % %IGMM#.q12IQQvcl++r$)rr*rr)rzSerializerProtocol | Nonerztype[defaultdict]returnNone)rr)r7type[BaseException] | Noner8BaseException | Noner9TracebackType | Noner bool | None)r?rr@rrArrr)rHstrrIrrJrrzdict[str, Any])rXr rzCheckpointTuple | None) rXRunnableConfig | Nonerkdict[str, Any] | Nonerlrrm int | NonerzIterator[CheckpointTuple]) rXr rYrrZrrrrr )rT) rXr rzSequence[tuple[str, Any]]rrrrrr)rHrrr) rXrrkrrlrrmrrzAsyncIterator[CheckpointTuple])rz str | Nonerrrr)__name__ __module__ __qualname____firstlineno____doc____annotations__rr)r1r6r<rBrPrhrrrrrrrrrr__static_attributes__ __classcell__r-s@r%rrsy@   ,0%0 1)1# 1  11 &C,C(C( C  C&I.I*I* I  I  -0 SUlSUlSUlXl[TU]"U0UD6 g)Nrepickle)flagmodeformatfilenamer(r))r,rargskwdsr-s r%r)PersistentDict.__init__"s/     $'$'r$cURS:XagURS-n[XRS:XaSOS5nUR U5 UR5 [R"XR5 URb,[ R"URUR5 gg![ a [ R"U5 ef=f!UR5 f=f)zWrite dict to diskrNz.tmprwbw) rropenrdump Exceptionosremovecloseshutilmoverchmod)r,tempnamefileobjs r%syncPersistentDict.sync)s 99  ==6)x)@cJ  IIg  MMO Hmm, 99 HHT]]DII . !   IIh    MMOsB;;"CC C2cDUR5 UR5 gr!)rclearr2s r%rPersistentDict.close:s  r$cU$r!r#r2s r%r1PersistentDict.__enter__>s r$c$UR5 gr!)r)r,exc_infos r%r6PersistentDict.__exit__As  r$cURS:Xa"[R"[U5US5 g[ S[ UR5-5e)NrzUnknown format: )rrrr"NotImplementedErrorrepr)r,rs r%rPersistentDict.dumpDs< ;;( " KKT GQ /%&84 ;L&LM Mr$cURS:Xag[URURS:XaSOS5n[R 4H5nUR S5 URU"U55s sSSS5 $ [S5e![a  SSS5 g[a$ [RSUR35 ef=f!,(df  g=f)NnrrbrrzFailed to load file: zFile not in a supported format)rrrrrloadseekrEOFErrorrloggererrorname ValueError)r,rloaders r%rPersistentDict.loadJs 99   $--)@c Jg!;;. Q;;vg77 K J(=> >   K J!LL#8!GHK Js5&C!B7C C CC%-CC C#)rrrr)rr rrrr rr)rr)rr)rr rr)rr rr)rrrrrr)rrr1r6rrrrrs@r%rrs1 (/"N ??r$r)( __future__rloggingrrrr collectionsrcollections.abcrrr contextlibrr r typesr typingr langchain_core.runnablesr langgraph.checkpoint.baserrrrrrrrr getLoggerrrrr MemorySaverrr#r$r%rs" #==UU3      8 $m,46Qm,` F?[F?r$