[if| UddlmZddlZddlZddlZddlZddlZddlmZm Z m Z m Z m Z m Z mZddlmZmZddlmZmZmZddlmZmZddlmZmZmZmZmZddlm Z m!Z!m"Z"m#Z#m$Z$dd lm%Z&dd l'm(Z(m)Z)dd l*m+Z+m,Z,dd l-m.Z.dd l/m0Z0ddl1m2Z2m3Z3m4Z4m5Z5ddl6m7Z7m8Z8ddl9m:Z:ddl;mZ> d= d>dZ@d=d?dZAe d= d@dZBGddeCejZEeFZGejdk\ZIde!dee!dejjfdejjfdeYwxYw)F) annotationsN) AsyncIterator AwaitableCallable Coroutine GeneratorIteratorSequence)AsyncExitStackcontextmanager)ContextToken copy_context)partialwraps)AnyOptionalProtocol TypeGuardcast)RunnableRunnableConfigRunnableLambdaRunnableParallelRunnableSequence) RunnableLike)run_in_executorvar_child_runnable_config)InputOutput)LangChainTracer) BaseStore) ensure_config%get_async_callback_manager_for_configget_callback_manager_for_config patch_config)CONFCONFIG_KEY_RUNTIME)MISSING) StreamWriter)_StreamingCallbackHandlerconfigrcTtj|}|ddlm}|d|i|S)cSet the child Runnable config + tracing context. Args: config: The config to set. r_set_tracing_contextparent)rsetlangsmith.run_helpersr0)r,run config_tokenr0s n/var/www/html/userprofiledev.eatanceapp.com/venv/lib/python3.12/site-packages/langgraph/_internal/_runnable.py_set_config_contextr7As.-008L >h_- c ^tj||ddlm}|dddddddyy)zjSet the child Runnable config + tracing context. Args: token: The config token to reset. Nrr/)r1 project_nametagsmetadataenabledclient)rresetr3r0)tokenr4r0s r6_unset_config_contextrAQs> ##E* > $    r8c#Kt}|jt||} ||jt||y#|jt||wxYww)r.N)rr4r7rA)r,r4ctxr5s r6set_config_contextrDgsN .C77.rSs$r8storer"zOptional[BaseStore]previousruntimez1tuple[tuple[str, tuple[Any, ...], str, Any], ...]KWARGS_CONFIG_KEYSceZdZddZy)_RunnableWithWritercyrQrK)selfstaterNs r6__call__z_RunnableWithWriter.__call__sr8N)r\rrNr*returnr rGrHrIr]rKr8r6rYrYsLr8rYceZdZddZy)_RunnableWithStorecyrQrK)r[r\rTs r6r]z_RunnableWithStore.__call__sSr8N)r\rrTr"r^r r_rKr8r6rarasHr8rac$eZdZ ddZy)_RunnableWithWriterStorecyrQrK)r[r\rNrTs r6r]z!_RunnableWithWriterStore.__call__r8N)r\rrNr*rTr"r^r r_rKr8r6rdrds''3G r8rkc(eZdZ ddZy)_RunnableWithConfigWriterStorecyrQrK)r[r\r,rNrTs r6r]z'_RunnableWithConfigWriterStore.__call__sr8N) r\rr,rrNr*rTr"r^r r_rKr8r6rnrns7      r8rnceZdZdZ d dddddd d dZd dZ d d dZ d d d Zy)RunnableCallablezPA much simpler version of RunnableLambda that requires sync and async functions.NTF)namer;tracerecurse explode_argsc 6||_|j8|r" |jdk7r|j|_n|r |j|_||_||_||_||_||_||_||_ | | tdi|_ tjtt|xs|j } t"D]\} } } } | j%| }||j&t(vr.| t*fk7re|j,| vrW| dk(rQ|j,tj.j0k7r*t3j4d|j,dt6d| | f|j| <y#t$rY;wxYw#t$rYKwxYw)Nzz/At least one of func or afunc must be provided.r,z\The 'config' parameter should be typed as 'RunnableConfig' or 'RunnableConfig | None', not 'z'. ) stacklevel)rrrGAttributeErrorfuncafuncr;kwargsrsrtru ValueError func_acceptsinspect signaturerr parametersrWgetkind VALID_KINDSANY_TYPE annotation Parameteremptywarningswarn UserWarning)r[rzr{rrr;rsrtrur|paramskwtyp runtime_keydefaultps r6__init__zRunnableCallable.__init__s 99 }} 2$(MM  %DI      ( allg6G6G6M6M&MMM99:cK##$  &1'$:D  b !/.@+& &s" E;F ; FF FFc|jjDcic] \}}|dvr||}}}|jddjd|jDdScc}}w)N>rzrrr{rsr,r|(z, c30K|]\}}|d|yw)=NrK).0kvs r6 z,RunnableCallable.__repr__..Gs!-YGXtq!1QElGXs))__dict__itemsget_namejoin)r[rr repr_argss r6__repr__zRunnableCallable.__repr__As{ ++- -1NN qD-  --/"!DII-YyGX-Y$Y#ZZ[\\  sA,c |jtd|jd| t}|jr|\}}i|j ||}n|f}i|j |}|j tij t}|jjD]y\}\}} ||vrt} |dk(r|} n|r|dk(r|} n t||} | tur:| tjj urt#d|d|jd| } | ||<{|j$rt'||j(} | j+d||j dxs|j-|j/d d } t1|| j3 } | j4D]B}t7|t8s|j:j t=| j>}nd}tA| |5}|jB|jg|i|}ddd| jEn|j|i|}|jJr"t7|tLr|jO||S|S#t$rYwxYw#1swYmxYw#tF$r}| jI|d}~wwxYw) Nz%No synchronous function provided to "ze". Either initialize with a synchronous function or invoke via the async API (ainvoke, astream, etc.)r,rVMissing required config key '' for ''.run_namerun_idrrr callbacks)(rz TypeErrorrrr#rur|rr'r(r~rr)getattrryrrrr}rsr%r;on_chain_startrpopr& get_childhandlers isinstancer!run_mapstrrrDr4 on_chain_end BaseExceptionon_chain_errorrtrinvoke)r[inputr,r|args_kwargsrVrrrkw_valuecallback_manager run_manager child_confighr4contextretes r6rzRunnableCallable.invokeIs 99 7 {C>>  >"_F   !MD'9 9w9&9F8D. .v.F**T2&**+=>*.*;*;*A*A*C &B&gV|#HX~!?&H#*7K#@7"g//555$7 }GDII;VXY#!F2J/+D2 ::>vtyyQ *99ZZ +>t}}zz(D1 :K  .+Fk>S>S>UV $--A!!_5iimmC 0B0B,CD. C' c:g%'++diiA$A&AC; ((-$))T,V,C <s|j@j tC| jD}nd}tG| |5}tIjJ|| d{}dddn |d{}| jMd{n|j|i|d{}|jRr*t=|tTr|jW||d{S|S#t$rY wxYw7_7#1swYxYw77z#tN$r }| jQ|d{7d}~wwxYw77Yw) Nr,rVrrrrrrrr),r{rr#rur|rr'r(r~rr)rryrrrr}rrrsr$r;rrr&rrrrASYNCIO_ACCEPTS_CONTEXTrrr!rrrrDasyncio create_taskrrrrtrainvoke)r[rr,r|rrrVrrrrrrrcororr4rrrs r6rzRunnableCallable.ainvokeszz;;uf- - >"_F   !MD'9 9w9&9F8D. .v.F**T2&**+=>*.*;*;*A*A*C &B&gV|#HX~!?&H#*7K#@7"g//555$7 }GDII;VXY#!F2J-+D0 ::DVTYYW  0 ? ?ZZ +8tyyzz(D1 !@!K  4+Fk>S>S>UV IdD#o6  D8SF8ST*(11%a9"#))--K4F4F0G"HC!2 #+L#>'$+$7$7g$NN?>!%*C "..s333" D3F33C <% 4 ! !00333  43sC L/ KB1L/ K* L/A*K?;  %)">;2>;?>;  >; # >;>;>;>;>; >;@];?JJ"7JJMJ JZ;?GG"7GJMG Gr8rqctj|xs-t|dxrtj|jS)zCheck if a function is async.r])riscoroutinefunctionhasattrr]rzs r6is_async_callablers= ##D) 7 4 $ 7  ' ' 6r8ctj|xs-t|dxrtj|jS)z*Check if a function is an async generator.r])risasyncgenfunctionrr]rs r6is_async_generatorrs= ""4( 6 4 $ 6  & &t}} 5r8c t|tr|St|stj|r t ||St |rGt|rtd|||St|t|ttd|||St|tr t|Stdt|)zCoerce a runnable-like object into a Runnable. Args: thing: A runnable-like object. Returns: A Runnable. )rrNrrrszGExpected a Runnable, callable or dict.Instead got an unsupported type: )rrrrisgeneratorfunctionrcallablerrqrrrdictrrtype)thingrrrss r6coerce_to_runnablers%" E "g&A&A%&He$// % U ##D%d%H H#e W_dEBC   E4 &&004U } >  r8ceZdZdZddd d dZ d dZ d dZ d ddZ d ddZ d dd Z d dd Z y) RunnableSeqzSequence of `Runnable`, where the output of each is the input of the next. `RunnableSeq` is a simpler version of `RunnableSequence` that is internal to LangGraph. N)rr trace_inputsctg}|D]w}t|tr|j|j/t|tr|j|j[|j t |ddyt|dkrtdt|||_||_ ||_ y)zCreate a new RunnableSeq. Args: steps: The steps to include in the sequence. name: The name of the `Runnable`. Raises: ValueError: If the sequence has less than 2 steps. NTrz,RunnableSeq must have at least 2 steps, got ) rrextendstepsrappendrlenr}rrr)r[rrrr steps_flatsteps r6rzRunnableSeq.__init__s&( D$ 01!!$**-D+.!!$**-!!"4TD"QR  z?Q >s:>OP    (r8ct|trTtg|j|j|j |j d|jxs |jiSt|tr"_F:6B&55 (,(9(9(ED  e $5J':4==?::h- 6  $TZZ04%k&;&;iAw"_F@H,;; (,(9(9(ED  e $5J':4==?::h- <   $TZZ04%k&;&;iAwHF?HA(G/?G.-GGG GGGGG$H8G9HGG GGH H G;4G75G;;HHc +$K| t}t|}|jd|j|j|n||j dxs|j |j dd}|jD]B}t|ts|jj t|j}nd}t||jdd}t||5} t!|j"D]S\} } | dk(r| j$||fi|} t||jd| dz}| j' |} Ut(>|jD]/}t|t(s|j+|j } 1|j-t. } d|j1|  dddy#t2$r} |j5| d} ~ wwxYw#1swYyxYwwr)r#r%rrrrrrrr!rrrr&rrDrrstream transformr+tap_output_iterr4 _consume_iterrrr)r[rr,r|rrrr4ridxriteratoroutputrs r6rzRunnableSeq.streams >"_F:6B&55 (,(9(9(ED  e $5J':4==?::h- 6 %%A!_-iimmC (:(:$;<& C !++isO<    , 1 "+4::!6ICax#.4;;uf#G#G!-"&1&;&;iay>(F#C"7-8(11%a)BC'('8'89K9KX'VH2!]H= ((09- ,0! **1- 1- ,sPBHAH;H=BG!8G!H H! H*G<<HHH  Hc XK| t}t|}|jd|j|j|n||j dxs|j |j ddd{}tr|jD]B}t|ts|jj t|j}nd}t||jdd}t!||5} t#4d{} t%|j&D]{\} } | dk(r| j(||fi|} n4t||jd| dz}| j+ |} t-| dsa| j/| j0}t2>|jD]/}t|t2s|j5|j } 1t7j8t; | d{} ddddd{|j= d{ dddy t#4d{} t%|j&D]{\} } t||jd| dz}| dk(r| j(||fi|} n| j+ |} t-| dsa| j/| j0}t2>|jD]/}t|t2s|j5|j } 1t; d{} ddddd{|j= d{y77l7p7^#1d{7swYoxYw7^#t>$r }|jA|d{7d}~wwxYw#1swYyxYw7{77#1d{7swYxYw7~#t>$r }|jA|d{7d}~wwxYww) Nrrrrrrracloser)!r#r$rrrrrrrrr!rrrr&rrDr rrastream atransformrpush_async_callbackrr+tap_output_aiterrr_consume_aiterrrr)r[rr,r|rrrr4rstackrr aiteratorrrs r6rzRunnableSeq.astream s >"_F@H,;; (,(9(9(ED  e $5J':4==?::h- <   # ))a1))--K,>,>(?@C* !%//)A3@F $FC0G ;-//5)24::)>IC"ax,8DLL,Q&,Q )5$*.9.C.C*3C!G9(=/&*" -1OOIv,N &y(; % 9 9):J:J K*?5@%0%9%9#-a1J#K010B0B(3(:(:I1&I&: (/':':*95w("5 0/>&226:::C10F 7)++u%.tzz%: T!-"&1&;&;iay,>$/$6$6 -" "6 $2)#<; %%44Q777;10H,(=),+++67 ! !00333 sA>P*N*P*,AP* O N(NN(A6NANANN  N# N(.N /N(3ON%O P*O>%O &O>)A6O' AO'#,O'O# O' O>$O%%O>)P*=O<>P*N( N N(N" NN" N(%O( O1O OO  OOOP* O>#O'%O>'O9-O0 .O95O><P*> P'P"PP""P''P*)rrrrrrzCallable[[Any], Any] | Noner^r)rrr^rrQ)rrr,rr|rr^r)rrr,rr| Any | Noner^r)rrr,rr|r r^ Iterator[Any])rrr,rr|r r^AsyncIterator[Any]) rGrHrIrJrrrrrrrrKr8r6rrs+ 48 )))2 )  )> 2 4=A**$9*LO* *^)-33&3 3 3p)-:1:1&:1 :1  :1~)-b7b7&b7 b7  b7r8rcbd}d}|D]}||}|r ||z}|}|S#t$r|}d}Y&wxYw)zConsume an iterator.NFritr add_supportedchunks r6rrqsZFM >F  &% F M  & %  &s  ..cpKd}d}|23d{}|r ||z}|}7#t$r|}d}Y&wxYw6|Sw)zConsume an async iterator.NFrrs r6r r s^FM e  &% F   & %  &  Ms06336 63 0606rQ)r,rr4rr^Token[RunnableConfig | None])r@rr4rr^r)r,rr4rr^zGenerator[Context, None, None])rzrr^z#TypeGuard[Callable[..., Awaitable]])rzrr^z'TypeGuard[Callable[..., AsyncIterator]])rrrrrrsrr^r)rr r^r)rrr^r)^ __future__rrenumrsysrcollections.abcrrrrrr r contextlibr r contextvarsr rr functoolsrrtypingrrrrrlangchain_core.runnables.baserrrrrrLCRunnableLikelangchain_core.runnables.configrrlangchain_core.runnables.utilsrr langchain_core.tracers.langchainr!langgraph.store.baser"langgraph._internal._configr#r$r%r&langgraph._internal._constantsr'r(langgraph._internal._typingr)langgraph.typesr*!langchain_core.tracers._streamingr+ ImportErrorr7rArDrEnumrFobjectr version_inforrrrW__annotations__POSITIONAL_OR_KEYWORD KEYWORD_ONLYrrYrardrhrkrnrqrrrrrr rKr8r6r1s"  644$9<* 0(%K (,   !$ !   ,'+ :  :!$ :# : :"c499 8**g5    ^ $ &    # #      ~w'8'8'>'>?        # #    Y  !            [4IE4j$  668I8I8V8VW M(5&=1MI%-0Ixv 6  7 xv 6 XeVm%<%-()'(uf}-. v . / uf}- . %UF]3 4 ^x^B ( ,  ", 59   DW7(W7t &O% $%s'J::KK