k iB[dZddlmZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z mZmZmZmZmZmZmZmZmZddlZddlmZdd lmZmZmZmZmZm Z m!Z!dd l"m#Z#dd l$m%Z&dd l'm(Z)dd l*m+Z+ddl,m-Z-ddl.m/Z/m0Z0ddl1m2Z2ddl3m4Z4ddl5m6Z6m7Z7m8Z8m9Z9ddl:m;Z<ddl=m>Z>m?Z?m@Z@mAZAmBZBejCeDZEGdde9ee&e)fe+fZFdS)z/Async implementation of Redis checkpoint saver.) annotationsN) defaultdict)asynccontextmanager) TracebackType) Any AsyncIteratorDictListOptionalSequenceTupleTypeUnioncast)RunnableConfig)WRITES_IDX_MAPChannelVersions CheckpointCheckpointMetadataCheckpointTuple PendingWriteget_checkpoint_id)TASKS)Redis) RedisCluster)AsyncSearchIndex) FilterQuery)NumTag)RedisConnectionFactoryULID)CHECKPOINT_PREFIXCHECKPOINT_WRITE_PREFIXREDIS_KEY_SEPARATORBaseRedisSaver)AsyncCheckpointKeyRegistry)EMPTY_ID_SENTINELfrom_storage_safe_idfrom_storage_safe_strto_storage_safe_idto_storage_safe_strc eZdZUdZded<ded<ded<ded<d Zd ed <d Zd ed < dpd d d eeddqfdZ drdsdZ dtdZ dud Z dvd$Z dwd%Zdxd,Zdtd-Zdtd.Zdtd/Z dydzd6Zd{d:Zd d d d;d|dBZ d}d~dKZ dddPZ dddQZ dddSZd{dTZddUZee dpd d d eedddWZ dddYZ ddd\Z ddd^Z!ddbZ"dddZ#ddeZ$dfd dgdhddoZ%xZ&S)AsyncRedisSaverzAsync Redis implementation for checkpoint saver. Supports standard Redis URLs (redis://), SSL (rediss://), and Sentinel URLs (redis+sentinel://host:26379/service_name/db). str _redis_urlrcheckpoints_indexcheckpoint_writes_indexz$Union[AsyncRedis, AsyncRedisCluster]_redisNzOptional[bool] cluster_modezOptional[AsyncKeyRegistry] _key_registry) redis_clientconnection_argsttlcheckpoint_prefixcheckpoint_write_prefix redis_url Optional[str]r6.Optional[Union[AsyncRedis, AsyncRedisCluster]]r7Optional[Dict[str, Any]]r8r9r:returnNonect||||||tj|_i|_d|_t|_dS)Nr;r6r7r8r9r:) super__init__asyncioget_running_looploop _key_cache_key_cache_max_sizer% _separator)selfr;r6r7r8r9r: __class__s C:\Users\Dell Inspiron 16\Desktop\tws\AgrotaPowerBi\back-agrota-powerbi\mcp-client-agrota\venv\Lib\site-packages\langgraph/checkpoint/redis/aio.pyrEzAsyncRedisSaver.__init__Vsf %+/$;    ,.. +-#' -c`|du|_|tj|fi|pi|_dS||_dS)zConfigure the Redis client. Supports standard Redis URLs (redis://), SSL (rediss://), and Sentinel URLs (redis+sentinel://host:26379/service_name/db). N)_owns_its_clientr get_async_redis_connectionr3)rLr;r6r7s rNconfigure_clientz AsyncRedisSaver.configure_clientpsQ!- 4  0K-3DKKK'DKKKrOctj|j|j|_tj|j|j|_dS)z+Create indexes without connecting to Redis.)r6N)r from_dictcheckpoints_schemar3r1 writes_schemar2rLs rNcreate_indexeszAsyncRedisSaver.create_indexessR!1!;  #$+" " " (8'A  T[( ( ( $$$rO thread_id checkpoint_ns checkpoint_idcjd|d|d|}||jvr |j|Stt|}t|}tt|}|j|j|||g}t|j|jkr ||j|<|S)z&Optimized key generation with caching.zckpt::) rIr/r+r,rKjoin_checkpoint_prefixlenrJ) rLrZr[r\ cache_keysafe_thread_idsafe_checkpoint_nssafe_checkpoint_idkeys rN!_make_redis_checkpoint_key_cachedz1AsyncRedisSaver._make_redis_checkpoint_key_cacheds HIGG GG GG   ' '?9- -/ ::;;0?? !3M!B!BCCo""'""     t  $": : :),DOI & rOtask_ididx Optional[int]c d|d|d|d|d| }||jvr |j|Stt|}t|}tt|} |j||| |g} |"| t||j| } t|jdkr | |j|<| S)z-Optimized writes key generation with caching.zwrite:r^N') rIr/r+r,_checkpoint_write_prefixappendrKr_ra) rLrZr[r\rhrirbrcrdre key_partsrfs rN(_make_redis_checkpoint_writes_key_cachedz8AsyncRedisSaver._make_redis_checkpoint_writes_key_cachedsYYXXXXXXXXSVXX   ' '?9- -/ ::;;0?? !3M!B!BCC  )       ?   SXX & & &o""9-- t  % ' '),DOI & rOcrK|d{V|d{V|S)zAsync context manager enter.N)asetupaset_client_inforXs rN __aenter__zAsyncRedisSaver.__aenter__sQkkmm##%%%%%%%%% rO _exc_typeOptional[Type[BaseException]]_exc_valOptional[BaseException]_exc_tbOptional[TracebackType]cK|jrw|jd{Vt|jddr(|jj}|r|d{Vd|j_d|j_dSdS)zAsync context manager exit.Nconnection_pool) rQr3aclosegetattrr| disconnectr1 _redis_clientr2)rLrurwrycoros rN __aexit__zAsyncRedisSaver.__aexit__s  >+$$&& & & & & & & &t{$5t<< {2==??JJJJJJJ48D " 09=D ( 6 6 6 > >rOcK||jdd{V|jdd{V|d{Vt |j|_dS)zSet up the checkpoint saver.F) overwriteN)rYr1creater2_detect_cluster_modeAsyncKeyRegistryr3r5rXs rNrrzAsyncRedisSaver.asetups $++e+<<<<<<<<<*11E1BBBBBBBBB''))))))))).dk::rOc>K|d{VdS)a1Set up the checkpoint saver asynchronously. This method creates the necessary indices in Redis and detects cluster mode. It MUST be called before using the checkpointer. This async method follows the canonical pattern used by PostgreSQL and SQLite checkpointers in the LangGraph ecosystem. The type ignore is necessary because the base class defines a sync setup() method, but async checkpointers require an async setup() method to properly handle coroutines. Usage: await checkpointer.setup() N)rrrXs rNsetupzAsyncRedisSaver.setups.kkmmrOcK|j%td|jddSt|jt r#tdd|_dStdd|_dS)zGDetect if the Redis client is a cluster client by inspecting its class.Nz%Redis cluster_mode explicitly set to z, skipping detection.z Redis client is a cluster clientTz#Redis client is a standalone clientF)r4loggerinfo isinstancer3AsyncRedisClusterrXs rNrz$AsyncRedisSaver._detect_cluster_modes   ( KK`8I```    F dk#4 5 5 & KK: ; ; ; $D    KK= > > > %D   rOmain_key related_keysOptional[list[str]] ttl_minutesOptional[float]rcK|*|jr#d|jvr|jd}||dkr[|g|pgz}|D]N} |j|d{V$#t$rt d|YKwxYwdSt|dz}|g|pgz}|D]O} |j||d{V%#t$rt d|YLwxYwdSdS)aApply Redis native TTL to keys asynchronously. Args: main_key: The primary Redis key related_keys: Additional Redis keys that should expire at the same time ttl_minutes: Time-to-live in minutes, overrides default_ttl if provided Use -1 to remove TTL (make keys persistent) Returns: Result of the Redis operation N default_ttlz!Failed to remove TTL from key: %sT<zFailed to apply TTL to key: %s) ttl_configgetr3persist Exceptionrwarningintexpire)rLrrrall_keysrf ttl_secondss rN_apply_ttl_to_keysz"AsyncRedisSaver._apply_ttl_to_keyss"   A=DO#C#C"o11-@@  "b  %:);<#QQCQ"k11#6666666666$QQQ'JCPPPPPQtkB.//K !z\%7R8H J JJ+,,S+>>>>>>>>>> JJJNN#CSIIIIIJ47 # "s$ A%%%B  B 1!C%C;:C;configrOptional[CheckpointTuple]c $K|dd}t|}|ddd}t|}t|}|r|tkrt|}||||}|jd} | ||j r/|j dr| || d{V} | d } | sdSd} |j r"|j dr| d } | d|| d|| d || d |tj | d i| ddd} nd|d|}|j|d{V}|sdS|jd} | ||j r/|j dr| || d{V} | d } | sdSd} |j r"|j dr| d } | d|| d|| d | d tj | d i| ddd} t| d}t| d}t| d }t| d }|j r9|j drdtvr^|t|t|t|}|j |d{V} | d krg}|jr"|j|||d{V}|||r|ndd{V|jrX|j rQ|j d}|5t'|dz}|j||||d{Vt+| t,r| dnt/| dd}t+|t0rtj|}n|}|r|dnd}g}|||||||r+|||||||||||r@t=j|d{V}| |d }|d }|d}n9t=j|d{V}| |d }g}|d }t+| t,r| ddnt/| dd}t+|t0rtj|n|}d|!D} tEtF| }!d|||di}"| d} t+| t,rtj | } |$| ||}#d}$|rd|||di}$tK|"|#|!|$|S)z1Get a checkpoint tuple from Redis asynchronously. configurablerZr[F transactionrefresh_on_readNrr\parent_checkpoint_id checkpointmetadata{})rZr[r\r $.checkpoint $.metadatacheckpoint_latest:r^ current_ttlrrrchannel_versions)rZr[r\r)rZr[rrci|]G\}}|ddt|tr|ddn|HSrreplacerr/.0kvs rN z.AsyncRedisSaver.aget_tuple..4sa   1 IIh # #+5a+=+=D (B'''1   rOrZr[r\rrr parent_configpending_writes)&rrr+r,r(_make_redis_checkpoint_keyr3pipelinejsonrr8executedumpsr)r*localsr5get_write_keysrr apply_ttlrdictr~r/loadsrnaget_channel_values_aload_pending_sends_aload_pending_writesrFgather_recursive_deserializeitemsrr_load_checkpointr)%rLrrZr\r[storage_safe_thread_idstorage_safe_checkpoint_nsstorage_safe_checkpoint_idcheckpoint_keyrpipeline_resultscheckpoint_datardoclatest_pointer_key doc_thread_iddoc_checkpoint_nsdoc_checkpoint_iddoc_parent_checkpoint_id write_keysrrcheckpoint_rawcheckpoint_data_dict channel_versions_from_checkpointtasksresultschannel_values pending_sendsr raw_metadata metadata_dictsanitized_metadatar config_paramcheckpoint_paramrs% rN aget_tuplezAsyncRedisSaver.aget_tupleRs' >*;7 )&11 ~.22?BGG "4I!>!>%8%G%G" ^ ].???);M)J)J &"<<&**N{+++>>H MMOO   / / / -4?#6#67H#I#I - ^,,,&.%5%5%7%7777777 .q1O" tK 24?#6#67H#I#I 2.q1 -00>TUU!0!4!4#%?"""1!4!4#%?"")8(;(;*,F))!% ?+>+>|R+P+P Q Q-11CC$"l6L!k!kOi!k!k #';??3E#F#FFFFFFFN! t{+++>>H MMOO   / / / -4?#6#67H#I#I - ^,,,&.%5%5%7%7777777 .q1O" tK 24?#6#67H#I#I 2.q1 -00>TUU!0!4!4#%?"""1!4!4_!E!E(7(;(;+>|R+P+P Q Q-11  C-S-=>> 1#o2FGG0_1EFF#7?? &!.!2( n- ot , , :"j99O00     04 # !*%2%=!!M'')     rO)filterbeforelimitOptional[RunnableConfig]rOptional[dict[str, Any]]rrAsyncIterator[CheckpointTuple]c Kg}|r#|tdt|ddk|ddx}r3|tdt|k|ddx}r3|tdt |kt |x}r3|tdt|k|r|D]\} } | dkr'|td| k2| dkr'|td| k_| dkr4|tdt| k| dkr4|tdt| ktd| |rdt |} | rS tj | } | j } |td | kn#t$rYnwxYw|r|d nd }|d d D]}||z}t|gd|pdd}|j|d {V}g}g}g}|jD]}t#|dr|jni}t'|d}t)|d}t'|d}t'|d}|dpt+|dd }|rtt-|t.r |r|d }t-|t0r|nt3j|}||di}ni}|r |dkr|||f}|||dpt+|dd}|dkrd}n|dks|dkrd}|r|||f}|||||||||||r|nd ||d i}|r||d {V}i}|r||d {V}|D]} | d}| d}| d}| d}g}!|r|||f}||g}!| d}|dpt+| d dd!}"t-|"t<rt3j|"n|"}#t?d"|# Dr4d#|#D}$tCtD|$}%ntCtD|#}%d|||d$i}&|#| d%r| d&n | d d| d|!}'g}(| dr|||f}||g}(d })|rd|||d$i})tI|&|'|%|)|('WVd S)(z+List checkpoints from Redis asynchronously.rZrrun_idr[r\sourcestepzUnsupported filter key: checkpoint_tsr*rN)rZr[r\rrr has_writesrl)r\DESC)filter_expression return_fields num_resultssort_by__dict__rrrr@rFtrueTfalseFalse) rdoc_dictrZr[r\rrcheckpoint_dictrrrrrrc3bK|]*}t|tdt|vV+dS)rN)rr/)rrs rN z(AsyncRedisSaver.alist..#sR'(jQRTWFXFXCFF"rOci|]G\}}|ddt|tr|ddn|HSrrrs rNrz)AsyncRedisSaver.alist..&sa&&&1IIh++3=a3E3EL (B///1&&&rOrrrr)%rnrr+rr,rrr ValueErrorr"from_str timestamprrr1searchdocshasattrrr)r*r~rlistrorjsonrr_abatch_load_pending_sends_abatch_load_pending_writesr/anyvaluesrrrr)*rLrrrrrrr[r\rrbefore_checkpoint_id before_ulid before_tscombined_filterexprqueryr all_docs_datapending_sends_batch_keyspending_writes_batch_keysrrrZrrrr batch_keycheckpoint_has_writespending_sends_mappending_writes_mapdoc_datarrrrrrrrrs* rNalistzAsyncRedisSaver.alistcs    $ $K  %f^&<[&IJJK    /33H===v V!((X:LV:T:T)TUUU!'~ 6 : :? K KK} !((((,? ,N,NN!2& 9 99} !((((,>},M,MM  E  E E1==%,,S]]a-?@@@@&[[%,,S[[A-=>>>>+%%%,,S-=-=ASTUAVAV-VWWWW(]]%,,S]]>PQR>S>S-STTTT$%C%C%CDDD  #4V#<#< # "&-0D"E"EK + 5I%,,S-A-AI-MNNNN D 3DL+A..%abb) $ $D t #OO--     .55e<<<<<<<< #% $&!<A A C'.sJ'?'?Gs||RH,S-=>>I1#o2FGGM0_1EFFM#7::g^T??O $ot4499&5a&8O "/4887OOo66 "&!$>%'\5CC !%..(,%%&'115Jg5U5U(-%$ <& }E )00;;;   (!*%2%2,@'6:I'St&4"7        # &*&E&E(''!!!!!!    $ '+'G'G)((""""""  &Q Q H -I$_5M$_5M#+,B#C BDM# E& 7KL 1 5 5i D D  +H#<< 55t::L lC00" \***! ,9,@,@,B,B C&&!. 3 3 5 5 &&&"  24FGG 2MBB!*%2%2!!,L $44 129H.//!%8)*   24N % G& }E !3!7!7 2!F!F48M# "%.)6)=%%! "#+!+-      WQ Q s:AI<< J J rrrrr new_versionsr stream_modec  K|d}|dd}|d}|d} |dd} |dd} | p| p|d d} d} |d r%| r#|d | kr | } |d } t|}t | }t| }|}d|| | d i} d}| rJ d d lm}|j| }|j}n+#t$rd dl }| d z}YnwxYw||rt|nd||| rt| nd|| || dd }tfddDrd|d<d|d<||| | }d}|jr&d|jvrt!|jddz}|j|g|g|d{V|jr`|jrYd|jvrP|N |j||d{Vn+#t$rt,d|YnwxYwd|d|}|j||d{V|N |j||d{Vn+#t$rt,d|YnwxYw|S#t2j$rX|dvrQ ||rt|nd|||dr0tt7|ddnd| || id|ddd}||||}|jr5|j|d |d{Vn^|jd!}||d ||d{Vn#t$rYnwxYwt$r}|d}~wwxYw)"a2Store a checkpoint to Redis with proper transaction handling. This method ensures that all Redis operations are performed atomically using Redis transactions. In case of interruption (asyncio.CancelledError), the transaction will be aborted, ensuring consistency. Args: config: The config to associate with the checkpoint checkpoint: The checkpoint data to store metadata: Additional metadata to save with the checkpoint new_versions: New channel versions as of this write stream_mode: The streaming mode being used (values, updates, etc.) Returns: Updated configuration after storing the checkpoint Raises: asyncio.CancelledError: If the operation is cancelled/interrupted rrrZr[r\N thread_tsridrrr!rCF) rZrr[r\rrrrrc3 K|]}|vV dSN)rrfrs rNrz'AsyncRedisSaver.aput..s'AAs3(?AAAAAArO)rrrrrr)keysr8z)Failed to apply TTL to checkpoint key: %srr^z-Failed to apply TTL to latest pointer key: %s)rmessagesrT) interruptedr%)rZrr[r\rrrr$r) copypoprr+r,ulidr"r r rtime_dump_checkpoint_dump_metadataallrgrrr1loadr4r3rrrsetrFCancelledErrorr/rrrr)rLrrrr$r%rrrZr[config_checkpoint_idr'r\rrrrr0 next_configrr"ulid_objr3rrrrres ` rNaputzAsyncRedisSaver.aput]s6n-2244 !!(HLL,B,BCC $$[11 $((99 +//FF $$["55 -U UZ^^DRT=U=U  $ NN4  -$ -t$$(<<<#7 &t,M"4I!>!>%8%G%G"%7 %F%F"   &!.!. O !M 7 7)))))),t}];;H$,$6MM 777KKK$(IIKK$$6MMM 748>F,V444B!;!;,&';<<<!."33D99 //99#O"AAAA.@AAAAA ;,4X,>)*26*:'"CCNK G=DO#C#C!$/-"@2"EFF (-- !$%.       ! O "T_44++,,^[IIIIIIIIII NNC^ "l6L!k!kOi!k!k +//"4nEE E E E E E E E&+,,-?MMMMMMMMMM NNG*  %4 4 4 444,&<@F"N"4V"<"<".cs+EEadn4EEEEEErOrir/rrrci|]\}}|| Sr+r+rrirfs rNrz/AsyncRedisSaver.aput_writes..s'W'W'WXS#S'W'W'WrOrz-Failed to apply TTL to write registry key: %sT)exc_infoFrrci|]\}}|| Sr+r+rIs rNrz/AsyncRedisSaver.aput_writes..s#U#U#UcC#U#U#UrO)raise_on_errorz JSON.MERGEmergez/Failed to apply TTL to checkpoint write key: %s)&r enumerateserde dumps_typedr+r,r _encode_blobrnr6r4rpr3rr8rrrrrar5make_write_keys_zset_keyzaddrrrrrrrrMrrr/lowerrrFr9)rLrr?rhrArZr[r\writes_objectsrirCvaluetype_rE write_obj upsert_case created_keysrfrzset_key zadd_mappingrrrr merge_failedresulterr_strrr=s rN aput_writeszAsyncRedisSaver.aput_writes2s*  F>*;7 ~.22?BGG ~.? %.v%6%6 - - !C!'5*0077KE4/ ::!4]!C!C!3M!B!B"&%)'377"))  I  ! !) , , , ,l EEfEEEEEKL ] !/ - -IGG!%%!%( C+**,,00c4Y;O;OPPPPPPPPP '',,,,!&8811$Q,/ ,=,=,A,A QRR((t % "!#J%3// "KK%))#%e, #))#....!"#'#5#N#N%}m$$ (X'W:AVAV'W'W'W "k..xFFFFFFFFF ? "}/O/O*-do.A.A-.P.PSU.U*V*VK"&*k&8&8;&O&O O O O O O O O O O#,""" &$S$,-1!/!"!"!"!"!"!""7 " """ " "/O/O$ ;//E/BB"0 - -IGG!%%!%( CMMOO''S$sI2F2FGGG '',,,,"$!U%)%D%D!=-&&NMMOO)).# d?STTT%:,:#1JJ!=-  H$V#UY|=T=T#U#U#ULMM(L999!) 0 0 0 F FFFFFFF % %))F!&)44)"%f++'722g6P6P+/LL"(L ) N  04 0@0@0B0B0F0F*11++++++&+T"1"5"5l"C"C=AOL9"&+"2"2"4"4"8"8 ._##%!  &88"%dom&Dr&I"J"JK+"&+"4"4S+"F"FFFFFFFFF("NN Q #   MT_4T4T&)$/-*H2*M&N&N "k00;GGGGGGGGGGG$K$%)'    4T4T%       G s%G Y /!K'K=9Y <K==Y FY BT54Y 5 U?Y U6Y 9!VY %WY WY >X'YY YY Y-&Y((Y-ctj|||||jS)aZSynchronous wrapper for aput_writes. Args: config (RunnableConfig): Configuration of the related checkpoint. writes (List[Tuple[str, Any]]): List of writes to store. task_id (str): Identifier for the task creating the writes. task_path (str): Path of the task creating the writes. )rFrun_coroutine_threadsafer`rHr^)rLrr?rhrAs rN put_writeszAsyncRedisSaver.put_writess;/   VVW 5 5ty  &(( rODict[str, Any]c tj|jurtjdn#t$rYnwxYwtj|||||jS)zTRetrieve channel_values using efficient FT.SEARCH with checkpoint_id (sync wrapper).zSynchronous calls to AsyncRedisSaver are only allowed from a different thread. From the main thread, use the async interface.For example, use `await checkpointer.get_channel_values(...)`.)rFrGrHInvalidStateError RuntimeErrorrbrr^)rLrZr[r\s rNget_channel_valuesz"AsyncRedisSaver.get_channel_values"s '))TY66/U7     D /  $ $   I    &((  .1 >>c tj|jurtjdn#t$rYnwxYwtj|||jS)ayGet a checkpoint tuple from Redis. Args: config (RunnableConfig): The config to use for retrieving the checkpoint. Returns: Optional[CheckpointTuple]: The retrieved checkpoint tuple, or None if no matching checkpoint was found. Raises: asyncio.InvalidStateError: If called from the wrong thread/event loop zSynchronous calls to AsyncRedisSaver are only allowed from a different thread. From the main thread, use the async interface.For example, use `await checkpointer.aget_tuple(...)` or `await graph.ainvoke(...)`.)rFrGrHrfrgrbrr^)rLrs rN get_tuplezAsyncRedisSaver.get_tuple8s '))TY66/+7    D / OOF # #TY  &(( ric tj|jurtjdn#t$rYnwxYwtj||||||jS)a7Store a checkpoint to Redis. Args: config (RunnableConfig): The config to associate with the checkpoint. checkpoint (Checkpoint): The checkpoint to save. metadata (CheckpointMetadata): Additional metadata to save with the checkpoint. new_versions (ChannelVersions): New channel versions as of this write. Returns: RunnableConfig: Updated configuration after storing the checkpoint. Raises: asyncio.InvalidStateError: If called from the wrong thread/event loop zSynchronous calls to AsyncRedisSaver are only allowed from a different thread. From the main thread, use the async interface.For example, use `await checkpointer.aput(...)` or `await graph.ainvoke(...)`.)rFrGrHrfrgrbr>r^)rLrrrr$s rNputzAsyncRedisSaver.putSs* '))TY66/+7    D / IIfj(L A A49  &(( riAsyncIterator[AsyncRedisSaver]cK|||||||4d{V }|WVdddd{VdS#1d{VswxYwYdS)NrBr+)clsr;r6r7r8r9r:savers rNfrom_conn_stringz AsyncRedisSaver.from_conn_stringws3%+/$;            KKKK                              s 4 >>rc8Kt|}t|}t|}||||}|j|dd{V} | siS| d} | diS)zERetrieve channel_values using efficient FT.SEARCH with checkpoint_id.rNrr)r+r,rgr3rr) rLrZr[r\rrrrrrrs rNrz#AsyncRedisSaver.aget_channel_valuess"4I!>!>%8%G%G"%7 %F%F"??     !% 0 0 2 2 6 6~~ V VVVVVVV I%Q' ~~.333rOr#List[Tuple[str, Union[str, bytes]]]c *K|sgS|jr |j|||d{V}|dkrgS|j|||d{V}g}|D]$}t|vsd|vr||%|sgS|jd}|D])}||*| d{V} g} | D]} | r| dtkr~| | dd| d d| d d| d d| d d f| dd| DS#t$rYnwxYwttdt|ktdt|kztdt|kztdtkzgdd} |j| d{V} t%| jd}fd|DS)a3Load pending sends for a parent checkpoint. Args: thread_id: The thread ID checkpoint_ns: The checkpoint namespace parent_checkpoint_id: The ID of the parent checkpoint Returns: List of (type, blob) tuples representing pending sends Nr__pregel_tasksFrrCrArrhrirDrErOc0|d|d|dfS)Nrrrr+xs rNz6AsyncRedisSaver._aload_pending_sends..s1qtQqT@RrOrfc.g|]}|d|dfS)r+)ritems rN z8AsyncRedisSaver._aload_pending_sends..s%TTTta$q'*TTTrOrZr[r\)rD$.blobrArhridrrrcft|ddt|ddt|ddfSNrArrhrirr~ds rNrzz6AsyncRedisSaver._aload_pending_sends..s6;++9b))5!$$rOc g|]?}t|dt|ddx$|jf@S)rrEN)r~rDencode)rrrEs rNrz8AsyncRedisSaver._aload_pending_sends..sU   8WQ-E-EFFFSV]]__d #SSSrO)r5get_write_countrrrnr3rrrrsortrrrr+r,r2r sortedr )rLrZr[r write_countrtask_write_keysrfrrpending_sends_with_sort_keys write_dataparent_writes_queryresr rEs @rNrz$AsyncRedisSaver._aload_pending_sendss|$ I  9 8 $($6$F$F}.B%% !##I$(#5#D#D}.B$$ #%%44C||'73'>'>'..s333&I ;//E/BB*--CMMOO'',,,, ( 0 0 2 222222202,")  J! jnnY&?&?5&H&H4;; *{B ? ? *y" = = *ua 8 8 *vr : : *vs ; ; -116R6R1SSSUT7STTTT     *[!!%7 %B%BB''+>}+M+MMO''+=>R+S+SSUy>>U*, LKK    0778KLLLLLLLL H           s)F3AF3 D(F33 G?GList[PendingWrite]c K|gS|jre |j|||d{V}|dkrgS|j|||d{V}|jd}|D])}||*|d{V}i} |D]} | r| dd} t| dd} | | | dd| dd| d d d | | | f<tj |j | } | S#t$rYnwxYwttd t|ktd t!|kztdt|kzgdd}|j|d{V}t'|jd}i}|D]}tt+|dd} tt+|dd} t+|dd}t-|tr|d}| | tt+|ddtt+|dd|d || | f<tj |j |} | S)NrFrrhrrirCrDrErOrhrirCrDrErZr[r\)rhrirCrDrrCrc$t|ddS)Nrirrrxs rNrzz7AsyncRedisSaver._aload_pending_writes..Ys'!UTUBVBVrOr{rzutf-8)r5rrr3rrrrr/r& _load_writesrOrrrr+r,r2r rr r~rr)rLrZr[r\rrrrfr writes_dictrrhrir writes_querywrites_results sorted_writessearch_writes_dictr blob_datas rNrz%AsyncRedisSaver._aload_pending_writess  I  , + $($6$F$F}m%% !##I$(#5#D#D}m$$  ;//E/BB%--CMMOO'',,,, ( 0 0 2 2222222FH ")  J! ",..B"?"?!*..":":;;'.#&'1~~i'D'D$.NN62$>$>$.NN63$?$? 77 WcN3"0!}|||f}| |g}t|d }d |D||<?H|S) zBatch load pending sends for multiple parent checkpoints. Args: batch_keys: List of (thread_id, checkpoint_ns, parent_checkpoint_id) tuples Returns: Dict mapping batch_key -> list of (type, blob) tuples c,g|]}t|Sr+r+)rpids rNrz>AsyncRedisSaver._abatch_load_pending_sends..s.222,/"3''222rOrZr[rCr\)r\rDrrArhrirCrNcft|ddt|ddt|ddfSrrrxs rNrzz.s6;339b115!,,#rOr{c pg|]3}t|ddt|dt|ddf4S)rDrrrErOr)rrs rNrz>AsyncRedisSaver._abatch_load_pending_sends..sV***  VR00XwsFC/H/HII***rO)rnrr+r,rrrr2r r r)r\rr)rLr results_map grouped_keysrZr[r group_keyparent_checkpoint_idsrr"storage_safe_parent_checkpoint_ids thread_filter ns_filterchannel_filtercheckpoint_filter batch_query batch_resultswrites_by_checkpointrrr?rs rNrz*AsyncRedisSaver._abatch_load_pending_sendsos I :< >H A A :I}&:"M2I ,,*, Y'  # * *+? @ @ @ @BNASASAUAUB B  = &Y (=%7 %B%B ")<])K)K &223H222 .  ,,0FFMO,,0JJI ^^u4NO$$(JJ &"/###$!#!!K #'">"E"Ek"R"RRRRRRRM:< $) G G';C$%9:AA#FFFF)>  $& 7KL -112FKK!'!!! ** - *** I&&! 0rO.Dict[Tuple[str, str, str], List[PendingWrite]]c T K|siSi}|jr|jd}|D]7\}}}|j|||}||8|d{V}|jd}g} t |D]k\} \}}}|| dkrL| |||f|j|||}||dda|||f} g|| <l| r|d{V} |jd}i} d}t | D]Y\} \}}}| | }|rFd|D}|D]7}| |||||f| |<|dz }8Z|dkr |d{V}i}t |D]\}}|r| |\}}}}|||f} | |vri|| <| dd }t| d d}||| d d | d d | d dd|| ||f<| D]"\} }tj|j||| <#ni}|D].\}}}||f}||vrg||<|||/| D]\\}}}t!|}t#|}d|D}t%d|k}t%d|k} t%d|k}!t'|| z|!zgdd}"|j|"d{V}#i}$|#jD]}%t/|%j}||$vri|$|<t3|%dd }tt3|%d d}&t3|%dt3|%d d}'||&t3|%d d t3|%d d |'d|$|||&f<|D]:}|||f} |$ |i}tj|j||| <;|S)zBatch load pending writes for multiple checkpoints. Args: batch_keys: List of (thread_id, checkpoint_ns, checkpoint_id) tuples Returns: Dict mapping batch_key -> list of PendingWrite objects FrNrrcdg|]-}t|tr|n|.Sr+)rbytesdecode)rrfs rNrz?AsyncRedisSaver._abatch_load_pending_writes.. sC((( #-7sE,B,BKCJJLLL(((rOrrhrrirCrDrErOrc,g|]}t|Sr+r)rcids rNrz?AsyncRedisSaver._abatch_load_pending_writes..Ks.///03&s++///rOrZr[r\)r\rhrirCrDrirr)r5r3rrRzcardrrNrnzrangerrr/rr&rrOr+r,rrr2r r r)r\r~)(rLrrrrZr[r\r[ write_countscheckpoints_with_writesirall_write_keys_resultswrite_key_mappingpipeline_indexr decoded_keysrfall_writes_datarrirrhidx_valrrrcheckpoint_idsrrstorage_safe_checkpoint_idsrrrrrfallback_writes_by_checkpointridx_strrEs( rNrz+AsyncRedisSaver._abatch_load_pending_writessv IFH   _ {+++>>H>H&( #@I*@U@U 0 0<,>&>&>&>&>&>&>O ),5_+E+EZ% 1# 6II}mS*3M=(QI(0DDDBD 4Y ?&0nnY&C&CG&)*..*B*B&C&CG+2'.+5>>)R+H+H(2vr(B(B(2vs(C(C SS0;Wg@L;E > >7 =-& 6 L00.0L+Y'..}====?K>P>P>R>R6 6 :*MN);I)F)F&-@-O-O*//7E///+ !$K 0 04J J 004NN $'$8$8#> 'VR 8 8 $ XX1-@'7ASTT&4M!*M= II"?"C"CMSU"V"VK-;-H K..K **rOcKt|}ttd|kddgd}|j|d{V}g}t }|jD]e}t|dd}t|dd} ||| ||| } | | f|D],}d|d t|} | | -ttd|kgd d} |j | d{V} | jD]t}t|dd}t|dd} t|d d}t|d d }| ||| ||}| |u|jrt }|jD]s}t|dd}t|dd} ||| f} | |vrF|| |j||| }| |t|jr'|D]"}|j|d{V#dS|j}|D]}|||d{VdS)zDelete all checkpoints and writes associated with a specific thread ID. Args: thread_id: The thread ID whose checkpoints should be deleted. rZr[r\rlrNrrr^r[r\rhrirhrir)r+rrr1r r8r r~addrrnr,r2!_make_redis_checkpoint_writes_keyr5rRr4r3deleterr)rLrZrcheckpoint_querycheckpoint_resultskeys_to_deletecheckpoint_namespacesrr[r\rrrrrhri write_keyprocessed_checkpointsr[rfrs rNadelete_threadzAsyncRedisSaver.adelete_threads "4I!>!>'!+..2HH*O<    $(#9#@#@AQ#R#RRRRRRR #%* 2 2C#C"==M#C"==M " % %m 4 4 4"<<& }N  ! !. 1 1 1 13 6 6M!s6L!s!sObcpOqOq!s!s   ! !"4 5 5 5 5 #!+..2HHNNN   $;BB<PPPPPPPP!& - -C#C"==M#C"==Mc9b11G#ua((C>>& }gsI  ! !) , , , ,   4$'EE !). 4 4 '_b A A '_b A A "+]M!J!)>>>)--n===#1JJ!=-  H#))(333   %% . .k((---------- . .{++--H% % %$$$$""$$ $ $ $ $ $ $ $ $ $rO keep_latestrl)strategy keep_last max_results thread_ids Sequence[str]rrrrc&K| |dkrd}nd}|std|dkrtd||dkrtd||D]<}t|}ttd|kd d g| }|j|d{V}|js_tt} |jD].} t| d d } | |  | /g} t} | D]l\} }t|d d}||d}| |t|t|kr| | m| s>g}| D]]} t| d d }t| d d }| ||||ttd|ktd |kzgd| }|j|d{V}|jD]w}| ||t|d d t|d d t|dd t)t|ddx|jr/| |j|||_| D]} | d|d| |jr'|D]"}|j|d{V#|j}|D]}|||d{V>dS)uPrune old checkpoints for the given threads per namespace. Retains the most-recent checkpoints **per checkpoint namespace** and removes the rest, along with their associated write keys and key-registry sorted sets. Each namespace (root ``""`` and any subgraph namespaces) is treated as an independent checkpoint chain. Channel values are stored inline within each checkpoint document, so they are automatically removed when the checkpoint document is deleted. Args: thread_ids: Thread IDs whose old checkpoints should be pruned. strategy: Pruning strategy. ``"keep_latest"`` retains only the most recent checkpoint per namespace (default). ``"delete"`` removes all checkpoints for the thread. keep_last: Optional override — number of recent checkpoints to retain per namespace. When provided, takes precedence over ``strategy``. Use ``keep_last=0`` to remove all checkpoints. max_results: Maximum number of checkpoints fetched from the index per thread in a single query. Defaults to 10 000. Nrrrz+``thread_ids`` must be a non-empty sequencez ``keep_last`` must be >= 0, got z"``max_results`` must be >= 1, got rZr[r\rrc$t|ddS)Nr\rrrs rNrzz(AsyncRedisSaver.aprune..'s'!_b"A"ArOT)rfreverserrhrirr^)r r+rrr1r r rrr~rnr8rrextendrarrr2rrr5rRr4r3rrr)rLrrrrrZrrrby_nsrnsto_evictfully_evicted_nsns_docs ns_sorted ns_evictedrr[r\rrwdocrfrs rNaprunezAsyncRedisSaver.aprunes>  8##   LJKK K q==K KKLL L ??O+OOPP P#^ )^ )I%7 %B%B " +"%k"2"26L"L.@'     (,'='D'DEU'V'V!V!V!V!V!V!V %* &1%6%6E). & &S/266b   %%%% H$'EE ${{}} - - G"AA  'yzz2  +++z??c'll22$((,,, N& &  '_b A A '_b A A %%33. } +[))-CC//=@B#W"V"V +     (,'C'J'J<'X'X!X!X!X!X!X!X*/  D"))>>2#D/2>>#D/2>>#D)R88eQ 7 788 %"))*CC%}m '  %%F)?FF"FF  ))22C+,,S11111111112 ;//11)))COOC((((&&((((((((((}^ )^ )rOr*)r;r<r6r=r7r>r8r>r9r/r:r/r?r@)NNN)r;r<r6r=r7r>r?r@)r?r@)rZr/r[r/r\r/r?r/) rZr/r[r/r\r/rhr/rirjr?r/)r?r.)rurvrwrxryrzr?r@)NN)rr/rrrrr?r)rrr?r) rrrrrrrrjr?r)r) rrrrrrr$rr%r/r?r)r) rrr?r@rhr/rAr/r?r@)rr)rZr/r[r/r\r/r?rd) rrrrrrr$rr?r)r;r<r6r=r7r>r8r>r9r/r:r/r?rn)rrN) rZr/r[r/r\r/rr>r?rd)rZr/r[r/rr/r?rt)rZr/r[r/r\r/r?r)rrr?r)rrr?r)rZr/r?r@) rrrr/rrjrrr?r@)'__name__ __module__ __qualname____doc____annotations__r4r5r#r$rErSrYrgrprtrrrrrrrr#r>r`rcrhrkrm classmethodrrrrrrrrrr __classcell__)rMs@rNr.r.@s OOO''''----$(L''''04M4444 $(.HL48(,!2'>........8$(GK48 '''''(    @((((T>>>>( ; ; ; ;    &&&&&-1'+ 11111fO O O O j,0+/# xxxxxx@$ SSSSSt [[[[[D (MO,6""""H$(HL48(,!2'>[. 59 44444<TVh h h h h Z  YYYYYv]]]]~rrrrhX%X%X%X%|&#' K)K)K)K)K)K)K)K)K)K)rOr.)Gr __future__rrFrlogging collectionsr contextlibrtypesrtypingrrr r r r r rrrrlangchain_core.runnablesrlanggraph.checkpoint.baserrrrrrrlanggraph.constantsr redis.asyncior AsyncRedisredis.asyncio.clusterrr redisvl.indexr redisvl.queryrredisvl.query.filterrrredisvl.redis.connectionr r2r"langgraph.checkpoint.redis.baser#r$r%r&'langgraph.checkpoint.redis.key_registryr'rlanggraph.checkpoint.redis.utilr(r)r*r+r, getLoggerrrr.r+rOrNrs55"""""" ######******                         333333&%%%%%------CCCCCC******%%%%%%))))))));;;;;;   8 $ $g )g )g )g )g )5%6679IIJg )g )g )g )g )rO