k i8 ddlZddlZddlmZddlZddlmZejeZ de e fdZ dddedede e effd Zdd de e efd edefd ZGd deZdS)N)Any)JsonPlusSerializerreturncJddlm}dtj|DS)zReturn the set of field names on the installed Interrupt dataclass. langgraph <=1.0.x has Interrupt(value, id). langgraph >=1.1.x has Interrupt(value, resumable, ns, when). r Interruptch|] }|j S)name).0fs C:\Users\Dell Inspiron 16\Desktop\tws\AgrotaPowerBi\back-agrota-powerbi\mcp-client-agrota\venv\Lib\site-packages\langgraph/checkpoint/redis/jsonplus_redis.py z$_interrupt_fields..s : : :qAF : : :)langgraph.typesr dataclassesfieldsrs r_interrupt_fieldsr s5 *))))) : :K.y99 : : ::r preprocessobjrct}ddi}|||j|d<n |j|d<dD]}||vrt||||<|S)z;Serialize an Interrupt object using whichever fields exist. __interrupt__TNvalueid resumablenswhen)rrgetattr)rrrresultfields r_serialize_interruptr#sp  F-t4F$*SY//w)w200 F??#C//F5M Mrreviver%cddlm}t}|||dn|d}d|i}dD]}||vr||vr ||||<|di|S)zIReconstruct an Interrupt from a serialized dict, tolerating both formats.rrNrrr )rrr)rr%rrrkwargsr"s r_deserialize_interruptr(%s))))))  F$*$6FF3w< CLE%u-F2'' F??u||JF5M 9  v  rceZdZdZgdZdedefdZdedefdZdedee e fffd Z dee e fdeffd Z de e efdefd Zdedefd ZxZS) JsonPlusRedisSerializera>Redis-optimized serializer using orjson for JSON processing. Redis requires JSON-serializable data (not msgpack), so this serializer: 1. Uses orjson for fast JSON serialization 2. Handles LangChain objects by encoding them in the LC constructor format 3. Handles Interrupt objects with custom serialization/deserialization 4. Applies parent's _reviver for security-checked object reconstruction In checkpoint 3.0, the serializer API uses only dumps_typed/loads_typed with tuple[str, bytes] signatures (changed from tuple[str, str] in 2.x). ) thread_id checkpoint_id checkpoint_nsparent_checkpoint_idrrcft|ttfrtdddlm}t||rt |Sddlm}t||rd|j|j dS t|dr/| }t|trd|vr|S| t|t|d r|jni S#t t"t$tf$r!td t|d wxYw) zCustom JSON encoder for objects that orjson can't serialize. This handles LangChain objects by delegating to the parent's _encode_constructor_args method which creates the LC format. z1bytes/bytearray in nested structure - use msgpackrrSendT__send__nodeargto_jsonlc__dict__r'zObject of type z is not JSON serializable) isinstancebytes bytearray TypeErrorrrr#r1r4r5hasattrr6dict_encode_constructor_argstyper8AttributeErrorKeyError ValueError)selfrrr1 json_dicts r_default_handlerz(JsonPlusRedisSerializer._default_handlerGst cE9- . . QOPP P .----- c9 % % -',, , )((((( c4   w  TsI&& %KKMM i..%493D3D$$00S '#z2J2J"R#,,PR1 *i@ T T TRd3iiRRRSS S Ts8>C27:C22>D0c*ddlm}m}t||rt |jSt||r#d|j|jdSt|trdddd gd fd |Did Stj |rkt|tsVfd tj | D}t||St|tr fd| DSt|t t"fr4fd|D}t|t"rt#|n|S|S)aERecursively add type markers to Interrupt and Send objects before serialization. This prevents false positives where user data with {value, id} fields could be incorrectly deserialized as Interrupt objects. Also handles dataclass instances to preserve type information during serialization. r)rr1rTr2 constructorbuiltinsset __set_items__c:g|]}|Sr _preprocess_interruptsr itemrEs r zBJsonPlusRedisSerializer._preprocess_interrupts..s'%X%X%XDd&A&A$&G&G%X%X%Xr)r7rArr'cBi|]\}}||Sr rOr kvrEs r zBJsonPlusRedisSerializer._preprocess_interrupts..s=Aq4..q11rr9cBi|]\}}||Sr rOrUs rrXzBJsonPlusRedisSerializer._preprocess_interrupts..s-NNN$!QAt22155NNNrc:g|]}|Sr rOrQs rrSzBJsonPlusRedisSerializer._preprocess_interrupts..s'KKKt44T::KKKr)rrr1r:r#rPr4r5rLr is_dataclassrAasdictitemsr@r?listtuple)rErrr1processed_dict processeds` rrPz.JsonPlusRedisSerializer._preprocess_interruptsus 43333333 c9 % %$ '8STTT T T " "" !2237;;  S ! ! %!5)#%X%X%X%XTW%X%X%X   %c * * :c43H3H '.s3399;;N00c>0RR R T " " NNNN#))++NNN N dE] + + KKKKsKKKI'1#u'='=L5###9 LJrcnt|trd|fSt|trdt|fS|dS ||}t j||j}d|fS#ttjf$r$t |cYSwxYw)zSerialize using orjson for JSON. Falls back to msgpack for structures containing bytes/bytearray. Returns: tuple[str, bytes]: Type identifier and serialized bytes r;r<N)nullr)defaultjson) r:r;r<rPorjsondumpsrGr=JSONEncodeErrorsuper dumps_typed)rEr processed_obj json_bytes __class__s rrjz#JsonPlusRedisSerializer.dumps_typeds c5 ! ! 0C<  Y ' ' 0c * * [; 0 $ ; ;C @ @ #\-AVWWW z))v56 0 0 0ww**3///// 0s3A::7B43B4datacF|\}}|dkrdS|dkr|S|dkrt|S|dkr)tj|}||S|dkr!t |St |S)zDeserialize with custom revival for LangChain/LangGraph objects. Args: data: Tuple of (type_str, data_bytes) Returns: Deserialized object with proper revival of LangChain/LangGraph types rcNr;r<remsgpack)r<rfloads_revive_if_neededri loads_typed)rErntype_ data_bytesparsedrms rrsz#JsonPlusRedisSerializer.loads_typeds!z F??4 g    k ! !Z(( ( f__\*--F))&11 1 i  77&&t,, ,77&&t,, ,rc B|dg}|rt|dkrtd|d|dd}|d} ddl}||}t ||}n2#ttf$r}td|d |d ||d}~wwxYw|d i} fd | D} |turd | vrt| d S|di| S)aReconstruct an object from LangChain constructor format. This handles objects that were serialized using _encode_constructor_args but are not LangChain objects (e.g., dataclasses, regular classes, sets). Args: obj: Dict with 'lc', 'type', 'id', and 'kwargs' keys Returns: Reconstructed object instance Raises: Exception: If object cannot be reconstructed rrIzInvalid constructor format: .NrzCannot import z from z: r'cBi|]\}}||Sr rrrUs rrXzIJsonPlusRedisSerializer._reconstruct_from_constructor..s-RRR41a!T33A66RRRrrMr ) getlenrDjoin importlib import_moduler ImportErrorrBr]rL) rErid_parts module_path class_namermoduleclser'revived_kwargss ` r_reconstruct_from_constructorz5JsonPlusRedisSerializer._reconstruct_from_constructorsk 774$$ C3x==1,,ACAABB Bhhx}-- b\      ,,[99F&*--CC^,   EEE;EE!EE   2&&RRRR6<<>>RRR #::/^;;~o677 7s$$^$$$s%)BB> B99B>ct|tr|ddvr|ddkri|}t|tr=|ddvr& |S#t $rYnwxYw|S|ddur\d|vrX t |jS#tttf$r'}t d |d Yd }~nd }~wwxYw|d durd |vrd|vrt|dkrt ddlm}||d |dS#tttf$r'}t d|d Yd }~nd }~wwxYwfd|DSt|t"rfd|DS|S)aERecursively apply reviver to handle LangChain and LangGraph serialized objects. This method is crucial for preventing MESSAGE_COERCION_FAILURE by ensuring that LangChain message objects stored in their serialized format are properly reconstructed. Without this, messages would remain as dictionaries with 'lc', 'type', and 'constructor' fields, causing errors when the application expects actual message objects with 'role' and 'content' attributes. It also handles LangGraph Interrupt objects (both old and new serialization formats) and must be reconstructed to prevent AttributeError when accessing Interrupt attributes. Additionally, it handles dataclass objects (like langmem's RunningSummary) that are serialized using the LangChain constructor format but need special reconstruction logic. Args: obj: The object to potentially revive, which may be a dict, list, or primitive. Returns: The revived object with LangChain/LangGraph objects properly reconstructed. r7)rIrArJrTrr$z*Failed to deserialize Interrupt object: %s)exc_infoNr3r4r5rr0)r4r5z%Failed to deserialize Send object: %scBi|]\}}||Sr r{rUs rrXz=JsonPlusRedisSerializer._revive_if_needed..Ts-IIITQAt--a00IIIrc:g|]}|Sr r{rQs rrSz=JsonPlusRedisSerializer._revive_if_needed..Ws'AAATD**400AAAr)r:r?r|_reviverr Exceptionr(rrrr=rDloggerdebugr}rr1r]r^)rErrevivedrr1s` rrrz)JsonPlusRedisSerializer._revive_if_neededs* c4 > wwt}}&&3776??m+K+K--,,gt,, #T1B1Bf1L1L#AA#FFF$ #N ww''4//GsNN1#d>TUUUU#Y ;LLDaRV! ##t++cMMSLLHHMM 4444444 [ 223u:>>$Y ;LL?T!JIIISYY[[III I T " " AAAASAAA AJsB B B,+B, C##D!:DD!1F G GG)__name__ __module__ __qualname____doc__SENTINEL_FIELDSrrGrPr_strr;rjrsr?rrr __classcell__)rms@rr*r*3sG  O,TC,TC,T,T,T,T\.#.#....`0s0uS%Z'80000002-c5j 1-c------8+%c3h+%C+%+%+%+%ZSSSSSSSSSSSSrr*)rloggingtypingrrf#langgraph.checkpoint.serde.jsonplusr getLoggerrrrLrrr?r#r(r*r rrrs9 BBBBBB  8 $ $;3s8;;;;9=   c # c3h    BF   S#X 3 #    ggggg0gggggr