D_iK%SrSSKJr SSKrSSKrSSKrSSKJrJrJ r J r J r J r SSK JrJrJr SSKJr SSKJrJrJrJr SSKJr SS KJrJrJrJrJr SS K J!r!J"r" SS K#J$r$ SS K%J&r&J'r' SS K(J)r)J*r*J+r+J,r, SSK-J.r. \(aSSK/J0r0J1r1 SSK%J2r2J3r3 O\4\-S-r1"SS\$SS9r5"SS\$SS9r6/SQr7/SQr8Sr9\"SSS9r:S\;S'S/Sjr<\S0Sj5r=S1S2S jjr>S3S!jr?SSSSSS".S4S#jjr@S5S$jrAS1S6S%jjrBS1S7S&jjrCS8S'jrDS9S(jrE\"S)5rF\"S*5rG"S+S,\5rH\S:S-j5rIS;S.jrJg)r?r@)r6r7r8r?child_runnable_config)defaultz!ContextVar[RunnableConfig | None]var_child_runnable_configc\[RU5nSnURS5=n(a{[USS5=n(ag[ S[US/55S5=n(aBUR R[ U55=n(a[5n[SU05 X4$)zSet the child Runnable config + tracing context. Args: config: The config to set. Returns: The token to reset the config and the previous tracing context. Nr8 parent_run_idc3T# UHn[U[5(dMUv M g7fN) isinstancer!).0handlers r1 &_set_config_context..s##E!'?;G#Es( (handlersparent) rEsetgetgetattrnextrun_mapr9rr)config config_tokencurrent_contextr8rGtracerruns r1_set_config_contextr[s-008LOjj- - -$YF FM F#*9j"#E   F NN&&s='9: :S :-/h_-  ((r0c #X# [5nUR[U5up#Uv UR[RU5 UR[ SSSSSSS.5 g!UR[RU5 UR[ SSSSSSS.5 f=f7f)zxSet the child Runnable config + tracing context. Args: config: The config to set. Yields: The config context. N)rP project_namer6r7enabledclient)rrZr[rEresetr)rVctxrW_s r1set_config_contextrcs .Cgg16:OL   )//>   $     )//>   $    s#B*A(>B*(?B''B*c r[/0S[0S9n[R5=n(a]UR [ SUR 5VVs0sH&up4UcM X3[;aUR5OU_M( snn55 UbiUR [ SUR 5VVs0sH2up4UcM U[;dMX3[;aUR5OU_M4 snn55 Ub1UR 5Hup4U[;dMUcMXASU'M URS05R 5H]upVURS5(aM[U[[[[45(dMDXQS;dMNUS:wdMVXaSU'M_ U$s snnfs snnf)zEnsure that a config is a dict with all keys present. Args: config: The config to ensure. Returns: The ensured config. N)r6r7r8r>r?r4r?__r7api_key)r4DEFAULT_RECURSION_LIMITrErRupdateritems COPIABLE_KEYScopy CONFIG_KEYS startswithrJr9r=floatbool)rVempty var_configkvkeyvalues r1 ensure_configrvs  /  E/2244z4  !+ 0 0 2 2=A #5qvvx1< 2    !'  .=)*k)9=A #5qvvx1< .  LLNDA # +,n%a(#ii399; t$$53UD"9::,,y %** c "< L=s F- F-1 F3>F3 F3c US:aSU3n[U5e[U[5(a,[U5U:waS[U5SUS3n[U5e[U[5(a[ [ [ U55$US:a[U[5(aURS5b[R"S[S S 9 [S UR5VVs0sHup4US:wdM X4_M snn5n[U5Vs/sH nU(a [ U5O [ U5PM" sn$[U5Vs/sHn[ U5PM sn$s snnfs snfs snf) aVGet a list of configs from a single config or a list of configs. It is useful for subclasses overriding batch() or abatch(). Args: config: The config or list of configs. length: The length of the list. Returns: The list of configs. Raises: ValueError: If the length of the list is not equal to the length of the inputs. rzlength must be >= 0, but got zr<r:r?c[U5nUbXS'SU;aUS SU;aUS UbX S'UbX0S'UbX@S'Ub0URS05EUEUS'U$)a?Patch a config with new values. Args: config: The config to patch. callbacks: The callbacks to set. recursion_limit: The recursion limit to set. max_concurrency: The max concurrency to set. run_name: The run name to set. configurable: The configurable to set. Returns: The patched config. r8r:r@r>r<r?)rvrR)rVr8r>r<r:r?s r1 patch_configr*s,6 "F({  z" v x "$3 !"$3 !%z!SFJJ~r$B!Sl!S~ Mr0c 0nSU5GHEnUGH:nUS:Xa30URS05EURS5=(d 0EUS'M=US:XaD[[URS/5URS5=(d /-55US'MUS:Xa30URS05EURS5=(d 0EUS'MUS:XGaURS5nUSn[U[5(ahUcUR 5US'GM [U[5(a XE-US'GM*UR 5nUHnUR USS9 M XaS'GMZUbuUcUR 5US'GMv[U[5(a0UR 5nUHnUR USS9 M XaS'GMURU5US'GMGMUS:XaUS[:wa USUS'GMGMU[;aX#bX#R 5X'GMX#=(d URU5X'GM= GMH U$) zmMerge multiple configs into one. Args: *configs: The configs to merge. Returns: The merged config. c3@# UHocM[U5v M g7frI)rv)rKcs r1rM merge_configs..`sFW#=##Wsr7r6r?r8T)inheritr>) rRsortedrQrJr~rk add_handlermergergrj)configsbaserVrtbase_callbacksthese_callbacksmngrcallbacks r1 merge_configsrTsdDGWFCj $hhz2.$zz*-3$Z %, 60B0HbIJ V &(hh~r2(zz.17R(^$ #!%+!6"("5ot44%-,;,@,@,B[)#ND99,:,L[) .224(7H ,,Xt,D)8,0[)$0%-,;,@,@,B[)#ND99.335(6H ,,Xt,D)7,0[)-;,@,@,Q[)1))+,0GG.45F.GD*+H %&+*A"K,,. "K8488C= cGf Kr0c [U5(a"Ub[X#R5S9US'OX$S'Ub[U5(aX4S'U"U40UD6$)adCall function that may optionally accept a run_manager and/or config. Args: func: The function to call. input: The input to the function. config: The config to pass to the function. run_manager: The run manager to pass to the function. **kwargs: The keyword arguments to pass to the function. Returns: The output of the function. r8rV run_managerrr get_childr funcinputrVrkwargss r1call_func_with_variable_argsrs^,d  "+F>S>S>UVF8 %8 #6t#<#< +}    r0c [U5(a"Ub[X#R5S9US'OX$S'Ub[U5(aX4S'U"U40UD6$)ajAsync call function that may optionally accept a run_manager and/or config. Args: func: The function to call. input: The input to the function. config: The config to pass to the function. run_manager: The run manager to pass to the function. **kwargs: The keyword arguments to pass to the function. Returns: The output of the function. rrVrrrs r1acall_func_with_variable_argsrs^0d  "+F>S>S>UVF8 %8 #6t#<#< +}    r0c[R"URS5URS5URS5S9$)zhGet a callback manager for a config. Args: config: The config. Returns: The callback manager. r8r6r7inheritable_callbacksinheritable_tagsinheritable_metadata)r configurerRrVs r1get_callback_manager_for_configrs;  $ $$jj5F+#ZZ 3 r0c[R"URS5URS5URS5S9$)zuGet an async callback manager for a config. Args: config: The config. Returns: The async callback manager. r8r6r7r)rrrRrs r1%get_async_callback_manager_for_configrs; ) )$jj5F+#ZZ 3 r0PTc`^\rSrSrSrSU4SjjrSU4SjjrSrU=r$)ContextThreadPoolExecutoriz?ThreadPoolExecutor that copies the context to the child thread.c v>[TU][S[[ 5R U/UQ70UD655$)zSubmit a function to the executor. Args: func: The function to submit. *args: The positional arguments to the function. **kwargs: The keyword arguments to the function. Returns: The future for the function. Callable[..., T])supersubmitrrrrZ)selfrargsr __class__s r1r ContextThreadPoolExecutor.submits; w~ #W\^-?-?%W%WPV%W X  r0c>^^[[US55Vs/sH n[5PM snmSUU4Sjjn[TU]"U/UQ70UD6$s snf)zMap a function to multiple iterables. Args: fn: The function to map. *iterables: The iterables to map over. timeout: The timeout for the map. chunksize: The chunksize for the map. Returns: The iterator for the mapped function. rcF>TR5R"T/UQ76$rI)poprZ)rcontextsfns r1 _wrapped_fn2ContextThreadPoolExecutor.map.._wrapped_fn(s<<>%%b040 0r0)rrreturnr)rr}rrr)rr iterablesrrbrrrs ` @r1rContextThreadPoolExecutor.mapsc"-2#il2C,DE,DqLN,DE 1 1w{     FsAr))rCallable[P, T]rP.argsrP.kwargsrz Future[T])rrrz Iterable[Any]rrrz Iterator[T]) r*r+r,r-r.rrr/ __classcell__)rs@r1rrsaI       (  "      r0rc## U=(d 0n[URS5S9nUv SSS5 g!,(df  g=f7f)zXGet an executor for a config. Args: config: The config. Yields: The executor. r<) max_workersN)rrR)rVexecutors r1get_executor_for_configr2s<\rF "JJ01     s$A4 A AAc Z^^^# SUUU4SjjnUb[U[5(aS[R"5R S[ S[ [5RU555IShvN $[R"5R X5IShvN $N0N7f)a Run a function in an executor. Args: executor_or_config: The executor or config to run in. func: The function. *args: The positional arguments to the function. **kwargs: The keyword arguments to the function. Returns: The output of the function. cH>T"T0TD6$![a n[UeSnAff=frI) StopIteration RuntimeError)excrrrs r1wrapper run_in_executor..wrapperWs2 ((( ( (C '  (s !!Nr)rr) rJrasyncioget_running_looprun_in_executorrrrrZ)executor_or_configrrrrs ``` r1rrEs$((!Z0BD%I%I--/??  #W\^-?-?%I J   ))+;;r;r<r;r:z str | Noner?zdict[str, Any] | Nonerr4)rrrr4) rzCallable[[Input], Output] | Callable[[Input, RunnableConfig], Output] | Callable[[Input, CallbackManagerForChainRun], Output] | Callable[[Input, CallbackManagerForChainRun, RunnableConfig], Output]rrrVr4rz!CallbackManagerForChainRun | Nonerrrr) rzCallable[[Input], Awaitable[Output]] | Callable[[Input, RunnableConfig], Awaitable[Output]] | Callable[[Input, AsyncCallbackManagerForChainRun], Awaitable[Output]] | Callable[[Input, AsyncCallbackManagerForChainRun, RunnableConfig], Awaitable[Output]]rrrVr4rz&AsyncCallbackManagerForChainRun | NonerrrzAwaitable[Output])rVr4rr)rVr4rr)rVrrzGenerator[Executor, None, None]) rz Executor | RunnableConfig | Nonerrrrrrrr)Kr. __future__rruuidrcollections.abcrrrrrr concurrent.futuresr r r contextlibr contextvarsrrrr functoolsrtypingrrrrrlangsmith.run_helpersrrtyping_extensionsr langchain_core.callbacks.managerrrlangchain_core.runnables.utilsrrrr langchain_core.tracers.langchainr!langchain_core.callbacks.baser"r#r$r%r~r'r4rlrjrgrErAr[rcrvrrrrrrrrrrrrr)r0r1rs," XXCC%@@L'R =Ls T!I 6Ye6r   @JT@< )  )? )F  82j+: <+:FI+:+:b-1"&"&*.' !'*' '  '  '(''T?R6:! L!  !  !3!! !R;?! ! ! !8!!!D   $cN CL1 21 h !$$"Y8"Y "Y "Y "Y  "Yr0