A'h*ddlZddlZddlZddlZddlmZmZddlmZm Z m Z ddl m Z ddl m Z ddlmZddlmZmZmZmZmZmZmZmZmZddlmZdd lmZmZdd lm Z dd l!m"Z"dd l#m$Z$dd l%m&Z&m'Z'm(Z(m)Z)m*Z*ddl+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;mZ>m?Z?m@Z@mAZAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZIddlJmKZKddlLmMZMddlNmOZOmPZPddlQmRZRddlSmTZTddlUmVZVmWZWddlXmYZYddlZm[Z[m\Z\m]Z]m^Z^m_Z_m`Z`maZaddlbmcZcmdZdeee*e$ge*fZeejdk\ZgGddeZhGddeZiGddZjde(d ee[e ekfd!ee^d"ele^fd#Zm ddd%e eke$fd&eMd'ehd(eelekekfd)end"eeoekefeff d*Zpd+ee eqekefgdfd,eekd-e eqekefd"dfd.Zrd/eesd0e$d"esfd1Ztde(d%e eke$fd!eehd2eeed3e eke ekfd"eqeoekelefeuekff d4Zvd3e eke ekfd5euekde(d"enfd6Zwedddddd7dd8de(d9ele)d:e ekeWfd%e eke$fd&eMd;e dedd?edd@edd3ee eke ekfd5eeuekdAe eadBedd"eoeke`ff dCZxeddd7ddDde(d9ele)d:e ekeWfd%e eke$fd&eMd;e deeYd?ee&d@edeefd3ee eke ekfd5eeuekdAe eadBee]d"eoeke^ff dFZxdddddd7dd8de(d9ele)d:e ekeWfd%e eke$fd&eMd;e deeYd?ee&d@edeefd3ee eke ekfd5eeuekdAe eadBee]d"eeoeke`feoeke^fff dGZxeCfZydddddd7dHdIeqedJfdKeekde(dLezdMee*d9ele)d:e ekeWfd%e eke$fd&eMd;e deeYd?ee&d@edeefdNeeoeVefdBee]dAe ead"ede`e^ff&dOZ{de(d"ee*fdPZ|d%e eke$fdQe'dRee'dSe*dTeWd"e ekf dUZ}dVee_d9ele)dWekdXekdYeeoekefd"e_f dZZ~dTeWd&eMd%e eke$fd=endNeeoeVefd"ef d[Zd\ezd]eekezfd"ekfd^Zd\ezd]eekezfd"ekfd_Zd`eekeseqfd"ekfdaZejZGdbdcZy)eN) defaultdictdeque)IterableMappingSequence)copy)partial)sha1) AnyCallableLiteral NamedTupleOptionalProtocolUnioncastoverload) Callbacks)AsyncParentRunManagerParentRunManager)RunnableConfig)xxh3_128_hexdigest) BaseChannel)BaseCheckpointSaverChannelVersions Checkpoint PendingWriteV)CACHE_NS_WRITESCONFCONFIG_KEY_CHECKPOINT_IDCONFIG_KEY_CHECKPOINT_MAPCONFIG_KEY_CHECKPOINT_NSCONFIG_KEY_CHECKPOINTERCONFIG_KEY_PREVIOUSCONFIG_KEY_READCONFIG_KEY_RESUME_MAPCONFIG_KEY_SCRATCHPADCONFIG_KEY_SENDCONFIG_KEY_STORECONFIG_KEY_TASK_ID EMPTY_SEQERROR INTERRUPTMISSING NO_WRITESNS_ENDNS_SEP NULL_TASK_IDPREVIOUSPULLPUSHRESERVEDRESUMERETURN TAG_HIDDENTASKSSend)InvalidUpdateError)ManagedValueMapping)get_runnable_for_task identifier) read_channels)logger)INPUT_CACHE_KEY_TYPE PregelNode) BaseStore)AllCacheKey CachePolicyPregelExecutableTaskPregelScratchpad PregelTask RetryPolicy) merge_configs patch_config) ceZdZdZedeeeeefdffdZ edefdZ ede eee ffdZ ede efdZy) WritesProtocolzProtocol for objects containing writes to be applied to checkpoint. Implemented by PregelTaskWrites and PregelExecutableTask.return.cyNselfs \/home/kushmeetdev/Regenta/Chatbot/venv/lib/python3.12/site-packages/langgraph/pregel/algo.pypathzWritesProtocol.path]s:=cyrUrVrWs rYnamezWritesProtocol.name`sr[cyrUrVrWs rYwriteszWritesProtocol.writescs36r[cyrUrVrWs rYtriggerszWritesProtocol.triggersfs),r[N)__name__ __module__ __qualname____doc__propertytuplerstrintrZr]rr r_rarVr[rYrRrRYs|A=eE#sE/2C78== c 6sCx166 ,(3-,,r[rRcfeZdZUdZeeeeefdfed<eed<e eee fed<e eed<y)PregelTaskWriteszSimplest implementation of WritesProtocol, for usage with writes that don't originate from a runnable task, eg. graph input, update_state, etc..rZr]r_raN) rbrcrdrergrrhri__annotations__rr rVr[rYrkrkjsGQ c3o&+ ,, I U38_ %%smr[rkc eZdZUdZeed<eeedfee effed<e e e ed<e e ed<eed<dedeeedfee effde e e de e dedd f d Zy ) Callfuncinputretry cache_policy callbacksrp.rqrrrsrtrSNcJ||_||_||_||_||_yrUro)rXrprqrrrsrts rY__init__z Call.__init__}s(   ("r[)rbrcrd __slots__r rlrgr dictrhrrrLrHrrvrVr[rYrnrntsGI N sCx$sCx.0 11 H[) **;'' # #U38_d38n45 # -. # {+ # #  #r[rn checkpointinterrupt_nodestasksrScttt|djd}||dj t it fd|djD}|rV|Dcgc]J}|dk(r3|jr5t|jj dtvrn|j|vr|Lc}SgScc}w)z@Check if the graph should be interrupted based on current state.channel_versionsN versions_seenc3NK|]\}}|j|kDywrU)get).0chanversion null_versionseens rY z#should_interrupt..s-+ D' $((4..+s"%*tags) typenextitervaluesrr.anyitemsconfigr:r,r])ryrzr{ version_type any_updates_since_prev_interrupttaskrrs @@rYshould_interruptrs T*-?"@"G"G"IJDQRL>L o & * *9b 9D'*+'(:;AAC+($$ ,  #c) !)KKYY/1   s>ACFchannelsmanagedrselectfreshc ztt}t|tr1g}|jD]\}}||k(s ||j |!nU|D cgc] } | |vs|  }} |D cgc] } | |vs|  }} |jD]\}}||vs ||j | |rN|rLi} |D]8} | |vr(|| j } | j|| n|| } | | | <:t| |} n t||} |r(| j|D cic] } | || c} | Scc} wcc} wcc} w)zFunction injected under CONFIG_KEY_READ in task config, to read current state. Used by conditional edges to read a copy of the state with reflecting the writes from that node only.) rlist isinstancerhr_appendrupdaterA) rrrrrupdated managed_keyscvklocal_channelsccrs rY local_readrs[%0$5G&# KK %DAqF{ !!!$ %$*:aQ'\: :#8q'7!88KK %DAqF{ !!!$ % 13 #AG|a[%%' '!*%a[ "N1   #~v6x0 =1q*'!*,=> M+;8&>s D.D.) D33D3D8commit process_keysr_c|D]\\}}|ttfvs|t|tst d||j |vsEt d|j d||y)zFunction injected under CONFIG_KEY_SEND in task config, to write to channels. Validates writes and forwards them to `commit` function.NzExpected Send, got zInvalid node name z in packet)r6r;rr<r=node)rrr_rvalues rY local_writersuV e D%= U%6eT*(+>ug)FGGzz-(+=ejj\)TUU V  6Nr[currentchannelc||dzSdS)zHDefault channel versioning function, increments the current int version.rV)rrs rY incrementrs!-7Q;414r[get_next_versiontrigger_to_nodesc t|d}td|D}|D]V}|dj|jij |j Dcic]}||dvr ||d|c}X|drt |dj}nd}|Dchc]!}|j D]}|tvr||vr|#c}}D]+}||js||||||d|<-|dr|r|djtt} tt} |D]}|jD]s\}} |ttt t"t$t&fvr)|t(k(r|dj+| G||vr| |j+| `| |j+| u|drt |dj}nd}t-} | j/D]Y\}} ||vs ||j | s |#|||||d|<||j1sI| j3|[|rs|D]n}||j1s|| vs||j t4s5|8|||||d|<||j1s^| j3|p|rk|dsf| j7|rU|D]P}||j9s||||||d|<||j1s@| j3|R| | fScc}wcc}}w)a6Apply writes from a set of tasks (usually the tasks from a Pregel step) to the checkpoint and channels, and return managed values writes to be applied externally. Args: checkpoint: The checkpoint to update. channels: The channels to update. tasks: The tasks to apply writes from. get_next_version: Optional function to determine the next version of a channel. Returns: A tuple containing the managed values writes to be applied externally, and the set of channels that were updated in this step. c2t|jddS)NrO) task_path_strrZ)ts rYzapply_writes..s affRaj(Ar[)keyc34K|]}|jywrU)ra)rrs rYrzapply_writes..s.1AJJ.sr~r}N pending_sends)sortedr setdefaultr]rramaxrr7consumeclearrrr_r0r6r8r.r9r-r;rsetr is_availableaddr, isdisjointfinish)ryrr{rr bump_steprr max_versionpending_writes_by_channelpending_writes_by_managedvalupdated_channelsvalss rY apply_writesrs0 5A BE...I ?#..tyy"=DD!MM :&899j!34T::   $%*%78??AB   MM  x DH$4    D> ! ! #(8(D3C4J) *4 0 /"y?#))+7B$6G6A$6G < ..0$((. / 3D~**,=M1MD>((38H8T;K# <J1248  ~224(,,T2 3 ?+  ' '(8 9 /D~$$&+;+G7GTN8 -.t4 D>..0$((. / %&6 66{ s L> %&MrcFt|dxs|j| S)zACheck if there are any tasks that should be run in the next step.r)boolr)rrrys rYhas_next_tasksrls1  ?+ , 4D4O4O51r[rV)store checkpointermanagerrr retry_policyrspending_writes processesrstep for_executionrrrrrscyrUrVryrrrrrrrrrrrrrrss rYprepare_next_tasksrws$ r[)rrrrsTcyrUrVrs rYrrs$'*r[ci}tj|djdd}t|}g}t |dD]Z\}}t t |fdfid|d|d|d |d |d |d |d |d|d|d|d| d| d|d|d| x}sJ|j|\| rD| rBt}| D]'}| j|x}s|j|)t|}n|dsd}n|j}|D]W}t t|fdfid|d|d|d |d |d |d |d |d|d|d|d| d| d|d|d| x}sG|j|Y|Dcic]}|j|c}Scc}w)awPrepare the set of tasks that will make up the next Pregel step. Args: checkpoint: The current checkpoint. pending_writes: The list of pending writes. processes: The mapping of process names to PregelNode instances. channels: The mapping of channel names to BaseChannel instances. managed: The mapping of managed value names to functions. config: The runnable configuration. step: The current step. for_execution: Whether the tasks are being prepared for execution. store: An instance of BaseStore to make it available for usage within tasks. checkpointer: Checkpointer instance used for saving checkpoints. manager: The parent run manager to use for the tasks. trigger_to_nodes: Optional: Mapping of channel names to the set of nodes that are can be triggered by that channel. updated_channels: Optional. Set of channel names that have been updated during the previous step. Using in conjunction with trigger_to_nodes to speed up the process of determining which nodes should be triggered in the next step. Returns: A dictionary of tasks to be executed. The keys are the task ids and the values are the tasks themselves. This is the union of all PUSH tasks (Sends) and PULL tasks (nodes triggered by edges). id-rNrycheckpoint_id_bytescheckpoint_null_versionrrrrrrrrrr input_cachersrr}rV)binascii unhexlifyreplacer enumerateprepare_single_taskr6rrrrrkeysr5r)ryrrrrrrrrrrrrrrsrrrr{idx_rtriggered_nodesrnode_idscandidate_nodesr]rs rYrrsX46K",,Z-=-E-Ec2-NO*:6L;=EJ78Q& 3K  " !4  %1  *        (  &  $! "&# $&%  4 ( LL +<,$'E' 1G+//88x8&&x0 1*0)@ * +#..* & 4L  " !4  %1  *        (  &  $! "&# $&%  4 ( LL +,# #ADD!G ## #s!E8)rrrrrsr task_path.task_id_checksumrrrc%`| jti}|jtd}|ddkDrtnt}|dt k(r%t |dtrtttttttf|}|d}t|j}|j}| tdt }|r |t"|n|}|||t| |t t%|dt|d}|d |}g|dd d }| ||||d }|||k(s J|d || r-t'}|j(xs|}|r|j*|j,di|j,d}t/t0t3|jxsdft5t |tr|j7n||j8} nd} t;||j,||t=t?| d|i||j@xs|r|jCd| ndtD|tFtItJ|jL|jOtPtItR||tU||||tV| xs|jtVtX| xs|jtXtZi|jtZi||dit\dt|t^ta| tjt^||t5|j7| tjtbi ||jdxs|| || Stg|||S|dt k(rti|dk(rtt|d}!|!ti|dk\ry|d|!}"t |"tjs#tmjndtq|"dy|"jr|vr$tmjnd|"jrdy||"jr}#|#jr}$|$yt }|r|t"|"jrn |"jr}|||t| |"jrt t|!}ntmjnd|y|d |}g|dd d}| |"jr|||d }|||k(s J|d || r|#jtr|jw|#jtt'}|#j(xs|}|rz|j+|"jx}t/t0t3|#xsd|"jrft5t |tr|j7n||j8} nd} t;|"jr|"jx|$|t=t?| ||#jzd|"jr|r|jCd| ndtD|tFtItJ|jL|jOtPtItR||tU||"jr||tV| xs|jtVtX| xs|jtXtZi|jtZi||dit\dt|t^ta| tjt^||t5|j7| tjtbt||djt~di ||#jxs|| |||#j|#j Stg||"jr|S|dtk(r\tt|d}||vry||}#|yt||d|dj|||#rtt|#j} t|#||| |}%|%tury |r |t"|n|}|||t| |tg|}|t|}| |||dd |d }|||k(s J|d || rv|#jrx}'rf|#jtr|jw|#jtt'}|#j(xs|}|rf|j+|%}t/t0t3|#xsd|ft5t |tr|j7n||j8} nd} t;||%|'|t=t?| ||#jzd||r|jCd| ndtD|tFtItJ|jLt|jOtPtItR||tU|dd |||tV| xs|jtVtX| xs|jtXtZi|jtZi||dit\dt|t^ta| tjt^||t5|j7| tjtbt||djt~di ||#jxs|| ||dd |#j|#j Sytg|||dd Syy#t$r'}&tr|&jd|d |dd d!d}&~&wwxYw)"zPrepares a single task for the next Pregel step, given a task path, which uniquely identifies a PUSH or PULL task within the graph.rrrrNz1`call` functions must have a `__name__` attribute:rOT)langgraph_steplanggraph_nodelanggraph_triggerslanggraph_pathlanggraph_checkpoint_nsz != __dynamic__metadataz graph:step:r)run_namert configurablerzIgnoring invalid packet type z in pending sendszIgnoring unknown node name z Ignoring invalid PUSH task path F)rrchannel_values)writers subgraphsr}r~)rrzBefore task with name 'z ' and path '')Mrr r# _xxhash_str _uuid5_strr6rrnrrgrhrir?rpr] ValueError PUSH_TRIGGERr2rrrskey_funcrqrGrr@rencodettlrIrNrMrt get_childr+r)r rextendrr&rrkr*r$r"r!r( _scratchpadr'rrrKlenr<rBwarningrrrrargrr%r4r flat_writersrr5 _triggersrra _proc_inputr/ ExceptionSUPPORTS_EXC_NOTESadd_noter1)(rrryrrrrrrrrrrrrrrsrr parent_ns task_id_func task_path_tcallproc_r]ra checkpoint_nstask_idtask_checkpoint_nsrr_args_key cache_keyrpacketproc proc_noderexcrs( rYrrs .::dB'L  !92>I",S/A"5;:L|t 9R=$ ?5eS#t!;?PT*G''"MM%NN,* (&$#,YfhO * )5+VL4D4DEU4V/(UL,<,<=T,U14*../H"M4%z$'741$02D-{"4L,,-BC*#./A/H/H/JK"4L,,-BC 01" %L *l_00 dgtY7 7 1  y>Q sIaL)Cc*_56605Ffd+3DL>ARS{{)+1&++>OPV[[)D I #H7@9+fXfkk]3fkk ##D  C G NN=i[I J  -ay9,im+U+ "$kk"*''9    '.. R7)4@P?Q0R R. }} .WF,,< L'00<$'#D):]  '-7#-F)H!$$  ! '  !&x*ST#[[CJ))Kv*>?PT+G''"MM%NN,* (&$#,Y VXV * )5+VL4D4DEU4V/(UL,<,<=T,U14*../H"M4%z$'741$02D-{"4L,,-BC*#./A/H/H/JK"4L,,-BC 0,Z8H-I-M-M$d.?"" )T!!1\))..k66 pgv{{I> > 1 C1& y  " *    ) *  ' + +D 1 #   VDMM23H !"/ + '>"=Fyk&$84M"#D    G%2?6(7)!D "&"&&."+BQ-+= H +"22VwitDTCU4VV299$4$}}  6"WF#'#4#4#D L##/#8#8#=$, /!+D!1!B] $ /#-h#<!) 1%- ),, % %) /$) &Xtyy(Q&*$+!( 1 1Kv2F G%)!3G /$/$*MM$))..*:$;2" !0$.$,$+$4(1"1 (,(.(0 %& 2"!1$)$O\-=-=>N-O 7$0%Q'3'7'78O'P 9<"&2&6&67PRT&U<"$-z$/?<"!9$ 8:L 5{$*4L$4$45J$K$2$+$67I7P7P7R$S$*4L$4$45J$K 8"!4Z@P5Q5U5U$,d6"O**5l!))9\!!"1 $ 1 1"&..CBB-%t"'42A??Q  6 %LL1$|IbqM?RST  sk== l-"l((l-cV|djD]}t|cSy)z6Get the null version for the checkpoint, if available.r}N)rr)ryrs rYrrs30188:tG} r[versionsrrrc|/|jD]}||js|fcStS|jD]@}||js|j|||j||kDs=|fcStSrU)rarrr,)rr"rrrrs rYrrs |MM D~**,w   MM D~**,l2|,2-w   r[parent_scratchpadrnamespace_hash resume_mapc tdkDr|D]}|dtk(s|dtk(s| nd D]1}|d|k(s |dtk(s|d}t|ts|g}ng}~|r||vr||}|j |nd g}ddt dtf fd }ttt||tS) NrrrrrScj|Sy|r jdSdS#t$rYywxYw)Nr)get_null_resumeremover)rnull_resume_writer$rs rYr)z$_scratchpad..get_null_resumesf  $ ,(88AA  %%&78(++!##  s6 AA) call_counterinterrupt_counterresumer)subgraph_counterF) r r3r8rrrrr rJLazyAtomicCounter) r$rrr%r&wtask_resume_writemapped_resume_writer)r+s `` @rYr r s >Q %At|#!$%! %!%   #Atw1Q46>$%aD!!"3T:):(;%  #!#   .J6",^"<   $ $%8 9! $ $# $ &(+- '*,  r[c|&|j|vrt||jSt|jtrai}|jj D]A\}}||vr+||j s||j||<5||||<Cnt|jtrO|jD]:}||vr)||j s||j}n+||}ntStd|j|r|j|j|}||||j<|S)zLPrepare input for a PULL task, based on the process's channels and triggers.z2Invalid channels type, expected list or dict, got ) input_cache_keyrrrrxrrrrr/ RuntimeErrormapper)rrrrrrrrs rYrr s\4#7#7;#FK 4 4566$--& }}**, &GAtxD>..0%d^//1CF#A  & DMM4 (MM DxD>..0"4.,,.C#gdmo N@ P  0kk#,/ D(() Jr[ namespacepartsc t|d}|jdjd|D|j}|ddd|ddd|dd d|d d d|d d  S) zAGenerate a UUID from the SHA-1 hash of a namespace and str parts.F)usedforsecurityr[c3`K|]&}t|tr|jn|(ywrUrrhrrps rYrz_uuid5_str..Bs$OAjC&8 a?O,.Nr  )r rjoin hexdigest)r9r:shahexs rYrr>sw y% 0CJJsxxOOOP --/C"1gYaAb {!C2Jr?s rYrz_xxhash_str..Js$T:a+=QXXZ1DTrANrBrrCrDrErF)rrG)r9r:rJs rYrrGsf CHHTeTTT C"1gYaAb {!C2J.Rs41mA&4s010d)rrgrrGrirh)rMs rYrrOs[ cE4= ) DII44 4 56 c3 Dz X r[c@eZdZUdZeegefed<ddZdefdZ y)r1_counterrUrSNcd|_yrUrTrWs rYrvzLazyAtomicCounter.__init__bs  r[c|jOt5|j$tjdj|_ddd|jS|jS#1swY|jSxYw)Nr)rULAZY_ATOMIC_COUNTER_LOCK itertoolscount__next__rWs rY__call__zLazyAtomicCounter.__call__esi == ) @==($-OOA$6$?$?DM @}}t}} @}}s 1A,,B)rSN) rbrcrdrwrr rirlrvr\rVr[rYr1r1]s,IxC())#r[r1r0)rrYsys threading collectionsrrcollections.abcrrrr functoolsr hashlibr typingr r r rrrrrrlangchain_core.callbacksr langchain_core.callbacks.managerrrlangchain_core.runnables.configrxxhashrlanggraph.channels.baserlanggraph.checkpoint.baserrrrrlanggraph.constantsrr r!r"r#r$r%r&r'r(r)r*r+r,r-r.r/r0r1r2r3r4r5r6r7r8r9r:r;r<langgraph.errorsr=langgraph.managed.baser>langgraph.pregel.callr?r@langgraph.pregel.iorAlanggraph.pregel.logrBlanggraph.pregel.readrCrDlanggraph.store.baserElanggraph.typesrFrGrHrIrJrKrLlanggraph.utils.configrMrNGetNextVersion version_inforrRrkrnrhrrrrxrrgrrirrrrrrbytesrrrr rrrrLockrXr1rVr[rYrxsW  *77   /T:%/@06C-'B*?8A; 4a78%%0-X-"z##23 -. ( )  L &c;&'& & & $s)S. ! &  &  4S>3  &R  huS#X/0$6 7 3-  U38_ %   5x}5{5s5 }7}7c;&'}7 N #}7~. }7 c8C=01 }7  4T#Y S )* }7@c8C=01#h   "&!>B+/*,"&!  & sJ' c;&'  !      5>  4= $- T] wsHSM'9:; s3x( ;' $-! " #z/#   ( ?C+/*,*.!**&*sJ'*c;&' * ! *  * *4=* I *./*4)+@@ A*wsHSM'9:;*s3x(*;'* ;'!*" ## #$#* *<"&26DH>B+/*,*.!t$t$&t$sJ't$c;&' t$ ! t$  t$ t$t$ I t$./t$4)+@@ At$wsHSM'9:;t$s3x(t$;'t$ ;'!t$" 4Z $s,@'@"A AB#t$nw ""&26DH=A*.*,'R@S#XR@smR@ R@  R@ &a[ R@&R@sJ'R@c;&'R@!R@ R@ R@R@ I R@./R@ 4)+@@ A!R@"$3S89:#R@$;'%R@&;''R@( 41 12)R@j  a[c;&' ? #    c] (= 01=&== = c3h( =  =@. . .c;&'.  . $3S89: . .bK%KsEz):KsKK5K%U *;KKuS#u_-#*9>>+  r[