ghOUdZddlmZddlZddlZddlZddlmZmZm Z m Z m Z ddl m Z mZmZddlmZddlmZmZmZmZddlmZdd lmZmZmZmZmZmZm Z dd l!m"Z"m#Z#dd l$m%Z%m&Z&m'Z'm(Z(erdd l)m*Z*m+Z+dd l,m-Z-m.Z.m/Z/m0Z0neee1efZ+Gdde#dZ2Gdde#dZ3gdZ4gdZ5dZ6eddZ7de8d<dTdZ9edUd Z:dVdWd"Z;dXd'Z dVd[d>Z? dVd\dBZ@d]dDZAd^dFZBe"dGZCedHZDGdIdJeZEed_dLZFd`dSZGdS)az&Configuration utilities for Runnables.) annotationsN) Awaitable GeneratorIterableIteratorSequence)ExecutorFutureThreadPoolExecutor)contextmanager)Context ContextVarToken copy_context)partial) TYPE_CHECKINGAnyCallableOptionalTypeVarUnioncast) ParamSpec TypedDict)InputOutputaccepts_configaccepts_run_manager)BaseCallbackManager Callbacks)AsyncCallbackManagerAsyncCallbackManagerForChainRunCallbackManagerCallbackManagerForChainRunceZdZdZdS) EmptyDictzEmpty dict type.N)__name__ __module__ __qualname____doc__\/var/www/FlaskApp/flask-venv/lib/python3.11/site-packages/langchain_core/runnables/config.pyr&r&&sr,r&F)totalcreZdZUdZded< ded< ded< ded < d ed < d ed < ded< ded<dS)RunnableConfigzConfiguration for a Runnable.z list[str]tagszdict[str, Any]metadatar callbacksstrrun_name Optional[int]max_concurrencyintrecursion_limit configurablezOptional[uuid.UUID]run_idN)r'r(r)r*__annotations__r+r,r-r0r0*s''OOO   MMM#""" !    r,r0)r1r2r3r5r7r9r:r;)r1r2r3r:child_runnable_config)defaultz!ContextVar[RunnableConfig | None]var_child_runnable_configconfigreturn@tuple[Token[Optional[RunnableConfig]], Optional[dict[str, Any]]]c ddlm t|}d}|dx}rt |ddx}rut fdt |dgDdx}rH|jt|x}rddl m }m }|}|d |i||fS) tSet the child Runnable config + tracing context. Args: config (RunnableConfig): The config to set. r)LangChainTracerNr3 parent_run_idc3<K|]}t||VdSN) isinstance).0handlerrFs r- z&_set_config_context..sG!'?;;r,handlers)_set_tracing_contextget_tracing_contextparent) langchain_core.tracers.langchainrFr@setgetgetattrnextrun_mapr4langsmith.run_helpersrOrP) rA config_tokencurrent_contextr3rGtracerrunrOrPrFs @r-_set_config_contextr]ys&A@@@@@,0088LOjj-- -.%YFF FM. #*9j"#E#E   F .N&&s='9'9:: :S." TSSSSSSS--//h_---  ((r,Generator[Context, None, None]c #vKddlm}t}|t|\}} |V|t j|||ddddddddS#|t j|||dddddddwxYw)rEr)rON)rQ project_namer1r2enabledclient)rXrOrr\r]r@reset)rArOctxrY_s r-set_config_contextrfs;::::: ..Cgg16::OL!   )/>>>   $      )/>>>   $     s A99?B8Optional[RunnableConfig]ctgidti}tx}r?|t dd|D|?|t dd|D|0|D]\}}|tvr | ||d|<|diD]W\}}|ds=t|ttttfr||dvr ||d|<X|S) zEnsure that a config is a dict with all keys present. Args: config (Optional[RunnableConfig], optional): The config to ensure. Defaults to None. Returns: RunnableConfig: The ensured config. N)r1r2r3r9r:r0cXi|]'\}}|||tvr|n|(SrI) COPIABLE_KEYScopyrKkvs r- z!ensure_config..sA1}1 #5#5qvvxxx1$}}r,cji|]0\}}||tv||tvr|n|1SrI) CONFIG_KEYSrjrkrls r-roz!ensure_config..sM1}k)9)91 #5#5qvvxxx1)9)9)9r,r:__r2)r0DEFAULT_RECURSION_LIMITr@rTupdateritemsrq startswithrJr4r8floatbool)rAempty var_configrmrnkeyvalues r- ensure_configr}s  /    E/22444z     * 0 0 2 2      &    LLNN - -DAq ## +,n%a(ii3399;;++ Ut$$ +53UD"9:: +5,,,%*E* c " Lr,9Optional[Union[RunnableConfig, Sequence[RunnableConfig]]]lengthr8list[RunnableConfig]c|dkrd|}t|ttr8t|kr%dtd|d}t|ttr"t t t S|dkrttryddtj d td td d Dfdt|DSfdt|DS)aGet a list of configs from a single config or a list of configs. It is useful for subclasses overriding batch() or abatch(). Args: config (Optional[Union[RunnableConfig, list[RunnableConfig]]]): The config or list of configs. length (int): The length of the list. Returns: list[RunnableConfig]: 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 z.s#PPP1!x--q!---r,cPg|]"}|rtnt#Sr+r})rKirA subsequents r- z#get_config_list..sC   *+ EM* % % % f0E0E   r,c.g|]}tSr+r)rKrrAs r-rz#get_config_list..s! 9 9 9aM& ! ! 9 9 9r,) ValueErrorrJrlenlistmapr}dictrTwarningswarnRuntimeWarningrrurange)rArmsgrs` @r-get_config_listrs&zz6f66oo&(##F v(=(= A6{{ A A17 A A A oo&(##0C v../// zzj..z6::h3G3G3S N#     PP PPP       6]]     : 9 9 95== 9 9 99r,)r3r9r7r5r:r3Optional[BaseCallbackManager]r9r6r7r5 Optional[str]r:Optional[dict[str, Any]]ct|}|||d<d|vr|d=d|vr|d=|||d<|||d<|||d<|i|di||d<|S)aPatch a config with new values. Args: config (Optional[RunnableConfig]): The config to patch. callbacks (Optional[BaseCallbackManager], optional): The callbacks to set. Defaults to None. recursion_limit (Optional[int], optional): The recursion limit to set. Defaults to None. max_concurrency (Optional[int], optional): The max concurrency to set. Defaults to None. run_name (Optional[str], optional): The run name to set. Defaults to None. configurable (Optional[dict[str, Any]], optional): The configurable to set. Defaults to None. Returns: RunnableConfig: The patched config. Nr3r5r;r9r7r:)r}rT)rAr3r9r7r5r:s r- patch_configrs46 " "F({   z" v  x "$3 !"$3 !%z!SFJJ~r$B$B!Sl!S~ Mr,configsc i}d|DD]o}|D]h}|dkr3i|di|dpi|d<<|dkrLtt|dg|dpgz|d<|dkr3i|di|dpi|d<|dkr&|d}|d}t|trq|||d<t|tr ||z|d<:|}|D]}||d||d<q||||d<t|tr7|}|D]}||d||d<|||d<|d kr|d tkr |d |d <|tvr'||||||<H||p||||<jq|S) zMerge multiple configs into one. Args: *configs (Optional[RunnableConfig]): The configs to merge. Returns: RunnableConfig: The merged config. c38K|]}|t|VdSrIr)rKcs r-rMz merge_configs..Ws*FF =## FFr,r2r1r:r3NT)inheritr9) rTsortedrSrJrrk add_handlermergersrj)rbaserAr{base_callbacksthese_callbacksmngrcallbacks r- merge_configsrKsDGFWFFF29291 91 9Cj  $hhz2..$zz*--3$Z  %,, 60B0B0HbIJJ  V &&(hh~r22(zz.117R(^$$ ##!%+!6!6"("5ot44R%-,;,@,@,B,B[))#ND991,:_,L[)) .2244(7EEH ,,Xt,DDDD,0[))$0%-,;,@,@,B,B[))#ND99R.3355(6EEH ,,Xt,DDDD,0[))-;,@,@,Q,Q[))))+,0GGG.45F.GD*+ %%&+*A"3K,,..S "3K8488C==S c1 9d Kr,funcUnion[Callable[[Input], Output], Callable[[Input, RunnableConfig], Output], Callable[[Input, CallbackManagerForChainRun], Output], Callable[[Input, CallbackManagerForChainRun, RunnableConfig], Output]]inputr run_manager$Optional[CallbackManagerForChainRun]kwargsrrc t|r.|'t|||d<n||d<|t|r||d<||fi|S)aCall 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. Defaults to None. **kwargs: The keyword arguments to pass to the function. Returns: The output of the function. Nr3rArrr get_childrrrrArrs r-call_func_with_variable_argsrs0d&  "+Fk>S>S>U>UVVVF8  %F8 #6t#<#< +} 4   r,Union[Callable[[Input], Awaitable[Output]], Callable[[Input, RunnableConfig], Awaitable[Output]], Callable[[Input, AsyncCallbackManagerForChainRun], Awaitable[Output]], Callable[[Input, AsyncCallbackManagerForChainRun, RunnableConfig], Awaitable[Output]]])Optional[AsyncCallbackManagerForChainRun]Awaitable[Output]c t|r.|'t|||d<n||d<|t|r||d<||fi|S)aAsync 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. Defaults to None. **kwargs: The keyword arguments to pass to the function. Returns: The output of the function. NrrArrrs r-acall_func_with_variable_argsrs6d&  "+Fk>S>S>U>UVVVF8  %F8 #6t#<#< +} 4   r,r#cddlm}|j|d|d|dS)zGet a callback manager for a config. Args: config (RunnableConfig): The config. Returns: CallbackManager: The callback manager. r)r#r3r1r2inheritable_callbacksinheritable_tagsinheritable_metadata) langchain_core.callbacks.managerr# configurerT)rAr#s r-get_callback_manager_for_configrs`A@@@@@ $? $$jj55F++#ZZ 33   r,r!cddlm}|j|d|d|dS)zGet an async callback manager for a config. Args: config (RunnableConfig): The config. Returns: AsyncCallbackManager: The async callback manager. r)r!r3r1r2r)rr!rrT)rAr!s r-%get_async_callback_manager_for_configrsaFEEEEE )  )$jj55F++#ZZ 33   r,PTc8eZdZdZdfd Zd d d dfdZxZS)ContextThreadPoolExecutorz?ThreadPoolExecutor that copies the context to the child thread.rCallable[P, T]argsP.argsrP.kwargsrB Future[T]c ttdtt j|g|Ri|S)aCSubmit a function to the executor. Args: func (Callable[..., T]): The function to submit. *args (Any): The positional arguments to the function. **kwargs (Any): The keyword arguments to the function. Returns: Future[T]: The future for the function. Callable[..., T])supersubmitrrrr\)selfrrr __class__s r-rz ContextThreadPoolExecutor.submitsO ww~~ #W\^^-?%W%W%W%WPV%W%W X X   r,Nrtimeout chunksizefnr iterables Iterable[Any]r float | Nonerr8 Iterator[T]cdtt|dDd fd }tj|g|R||dS) aMap a function to multiple iterables. Args: fn (Callable[..., T]): The function to map. *iterables (Iterable[Any]): The iterables to map over. timeout (float | None, optional): The timeout for the map. Defaults to None. chunksize (int, optional): The chunksize for the map. Defaults to 1. Returns: Iterator[T]: The iterator for the mapped function. c*g|]}tSr+)r)rKres r-rz1ContextThreadPoolExecutor.map..(sEEEqLNNEEEr,rrrrBrcDjg|RSrI)popr\)rcontextsrs r- _wrapped_fnz2ContextThreadPoolExecutor.map.._wrapped_fn*s&%8<<>>%b04000 0r,r)rrrBr)rrrr)rrrrrrrrs ` @r-rzContextThreadPoolExecutor.maps&FEE#il2C2C,D,DEEE 1 1 1 1 1 1 1uww{         r,)rrrrrrrBr) rrrrrrrr8rBr)r'r(r)r*rr __classcell__)rs@r-rrssII      0!%             r,rGenerator[Executor, None, None]c#K|pi}t|d5}|VddddS#1swxYwYdS)zGet an executor for a config. Args: config (RunnableConfig): The config. Yields: Generator[Executor, None, None]: The executor. r7) max_workersN)rrT)rAexecutors r-get_executor_for_configr5s\rF "JJ011    s<AAexecutor_or_config)Optional[Union[Executor, RunnableConfig]]rrrrc XKdfd }|t|trZtjdt dt tj|d{VStj||d{VS)aRun a function in an executor. Args: executor_or_config: The executor or config to run in. func (Callable[P, Output]): The function. *args (Any): The positional arguments to the function. **kwargs (Any): The keyword arguments to the function. Returns: Output: The output of the function. Raises: RuntimeError: If the function raises a StopIteration. rBrcJ iS#t$r }t|d}~wwxYwrI) StopIteration RuntimeError)excrrrs r-wrapperz run_in_executor..wrapper]sF (4((( ( ( ( (C '  (s ""Nr)rBr) rJrasyncioget_running_looprun_in_executorrrrr\)rrrrrs ``` r-rrHs*((((((((!Z0BD%I%I!-//??  #W\^^-?%I%I J J         )++;;rs-,,,"""""" NNNNNNNNNNNNNNCCCCCCCCCC%%%%%%@@@@@@@@@@@@OOOOOOOOOOOOOOOOOO22222222 +LLLLLLLLtSy)*I 11111Ye1111h     @JzT@@@ !)!)!)!)H    622222j,:,:,:,:d04%)%)"-1++++++\????V9=!!!!!\>B"!"!"!"!"!J$(IcNN GCLL4 4 4 4 4 24 4 4 n$%Y%Y%Y%Y%Y%Yr,