ghq[dZddlmZddlZddlZddlZddlZddlmZm Z ddl m Z ddl m Z ddlmZddlmZdd lmZmZmZmZmZmZmZmZdd lmZmZdd lmZerdd lmZm Z m!Z!m"Z"m#Z#dd l$m%Z%eddZ&eddZ'dfdZ(dgdZ)dhd"Z*dhd#Z+dhd$Z,e d%&did'Z-d(d)djd.Z.Gd/d0ej/Z0Gd1d2ej/Z1Gd3d4ej/Z2Gd5d6ej/Z3Gd7d8ej/Z4dkd<Z5dld>Z6e d?&dmdAZ7dndEZ8GdFdGe9e:efZ;edHdZ<edIdZ=GdJdKee=eedLe>eefMZ?dodQZ@dpdSZAGdTdUeZBGdVdWeZCGdXdYeZDeeBeCeDfZEGdZd[eZFdqd_ZGGd`daZHdrdcZIdsdeZJdS)tzUtility code for runnables.) annotationsN)MappingSequence)Context) lru_cache) signature)groupby) TYPE_CHECKINGAnyCallable NamedTupleOptionalProtocolTypeVarUnion) TypeGuardoverride) create_model) AsyncIterable AsyncIterator Awaitable CoroutineIterable) StreamEventInputT) contravariantOutput) covariant semaphoreasyncio.Semaphorecororreturnr crK|4d{V|d{Vcdddd{VS#1d{VswxYwYdS)zRun a coroutine with a semaphore. Args: semaphore: The semaphore to use. coro: The coroutine to run. Returns: The result of the coroutine. N)rr!s [/var/www/FlaskApp/flask-venv/lib/python3.11/site-packages/langchain_core/runnables/utils.py gated_coror&.szzzzzzs & 00nUnion[int, None]coroslistcK|tj|d{VStj|tjfd|Dd{VS)zGather coroutines with a limit on the number of concurrent coroutines. Args: n: The number of coroutines to run concurrently. *coros: The coroutines to run. Returns: The results of the coroutines. Nc38K|]}t|VdSN)r&).0crs r% z*gather_with_concurrency..Ks-!J!Jq*Y":":!J!J!J!J!J!J)asynciogather Semaphore)r'r)rs @r%gather_with_concurrencyr5<sq y^U++++++++!!$$I!J!J!J!JE!J!J!JK K K K K K KKr1callableCallable[..., Any]boolcx t|jdduS#t$rYdSwxYw)zCheck if a callable accepts a run_manager argument. Args: callable: The callable to check. Returns: bool: True if the callable accepts a run_manager argument, False otherwise. run_managerNFr parametersget ValueErrorr6s r%accepts_run_managerr@NsM""-11-@@LL uu (+ 99cx t|jdduS#t$rYdSwxYw)zCheck if a callable accepts a config argument. Args: callable: The callable to check. Returns: bool: True if the callable accepts a config argument, False otherwise. configNFr;r?s r%accepts_configrD]sM""-11(;;4GG uurAcx t|jdduS#t$rYdSwxYw)zCheck if a callable accepts a context argument. Args: callable: The callable to check. Returns: bool: True if the callable accepts a context argument, False otherwise. contextNFr;r?s r%accepts_contextrGlsM""-11)<>$$$$ @ @ @X @ ; ; ;X ; ; ; ;X ; ; ;r1rwcJeZdZdZd dZed dZedd Zd S) NonLocalsz Get nonlocal variables accessed.r"rVcRt|_t|_dS)zCreate a NonLocals visitor.N)ryloadsstoresrzs r%rYzNonLocals.__init__s"uu # r1rZast.Namer ct|jtjr!|j|jdSt|jtjr!|j|jdSdS)zVisit a name node. Args: node: The node to visit. Returns: Any: The result of the visit. N) r]r^r_r`rrfrcStorerrgs r% visit_NamezNonLocals.visit_Name sr dh ) ) % JNN47 # # # # # #) , , % KOODG $ $ $ $ $ % %r1 ast.Attributect|jtjr|j}|j}t|tjr.|jdz|z}|j}t|tj.t|tjrF|j |j dz|z|j |j dSt|tj rt|j tjr&|j |j j dS|j }d}t|tjr8|r|jdz|z}n|j}|j}t|tj8t|tjr+|j |j dz|zdSdSdSdS)zVisit an attribute node. Args: node: The node to visit. Returns: Any: The result of the visit. .N)r]r^r_r`rarnrmrbrrfrcdiscardCallrl)rXrZparent attr_exprs r%visit_AttributezNonLocals.visit_Attributes dh ) ) DZF IVS]33 &"K#- 9 VS]33 &&#(++ D vy3:;;; ""69-----FCH-- Dfk3844 DJNN6;>22222#[F "I$VS];;.$4(. c(9I(EII(. I!' %VS];;. "&#(33D vy3'BCCCCC- D D D DDDr1Nr)rZrr"r )rZrr"r )rrrsrtrurYrrrr$r1r%rrst**&&&&  % % %X %DDDXDDDr1rcbeZdZdZddZeddZedd Zedd Zd S)FunctionNonLocalsz2Get the nonlocal variables accessed of a function.r"rVc,t|_dS)z#Create a FunctionNonLocals visitor.N)ry nonlocalsrzs r%rYzFunctionNonLocals.__init__?s#&55r1rZrr ct}|||j|j|jz dS)rNrrrupdaterrrXrZvisitors r%rz#FunctionNonLocals.visit_FunctionDefCD++ d gmgn<=====r1rct}|||j|j|jz dS)rNrrs r%rz(FunctionNonLocals.visit_AsyncFunctionDefQrr1r{ct}|||j|j|jz dS)r}Nrrs r%rzFunctionNonLocals.visit_Lambda_rr1Nrrrr) rrrsrtrurYrrrrr$r1r%rr<s<<)))) > > >X > > > >X > > > >X > > >r1rc2eZdZdZd dZed dZd S) GetLambdaSourcez)Get the source code of a lambda function.r"rVc"d|_d|_dS)zInitialize the visitor.Nr)sourcecountrzs r%rYzGetLambdaSource.__init__qs%)  r1rZr{r c|xjdz c_ttdrtj||_dSdS)r}rHunparseN)rhasattrr_rrrgs r%rzGetLambdaSource.visit_LambdavsD a 3 " " ,+d++DKKK , ,r1Nrr)rrrsrtrurYrrr$r1r%rrnsN33  , , ,X , , ,r1rrlr Optional[list[str]]c@ tj|}tjt j|}t }|||jrt|jndS#ttttf$rYdSwxYw)zGet the keys of the first argument of a function if it is a dict. Args: func: The function to check. Returns: Optional[list[str]]: The keys of the first argument if it is a dict, None otherwise. N)inspect getsourcer_parsetextwrapdedentrwrrTsorted SyntaxError TypeErrorOSError SystemError)rlcodetreers r% get_function_first_arg_dict_keysrs &&y..//#%% d'.|=vgl###= G[ 9ttsA9A<<BB Optional[str]c |jdkr|jnd}n#t$rd}YnwxYw tj|}t jt j|}t}| |n%#ttttf$r|cYSwxYw|jdkr|jn|S)zGet the source code of a lambda function. Args: func: a Callable that can be a lambda function. Returns: str: the source code of the lambda function. zNrH)rrAttributeErrorrrr_rrrrrrrrrrr)rlrRrrrs r%get_lambda_sourcers $  ; ;t}}  &&y..//!## d G[ 9 $]a//7>>T9s &&ABB*)B* list[Any]cD tj|}tjt j|}t }||g}t|dr-t|j rtj |j ntj |}i|j |j }|D]\}}||j vr|||j D]x} d| vrr| |r]|} | dddD](} | n8 t%| | } #t&$rYnwxYw|| yn%#t(t*t,t.f$rgcYSwxYw|S)zGet the nonlocal variables accessed by a function. Args: func: The function to check. Returns: list[Any]: The nonlocal variables accessed by the function. __wrapped__rrHN)rrr_rrrrrrr6rgetclosurevarsglobalsritemsappend startswithsplitgetattrrrrrr) rlrrrvaluesclosure candidateskvkkvvparts r%get_function_nonlocalsrs &&y..//#%% dt]++ .089I0J0J .G "4#3 4 4 4'--  >=7+<= $$&& * *DAqG%%% a   ' * *"99q!1!19B " abb 1**:!E"!(T!2!2BB-"""!EE" b))) * * G[ 9  Ms6D>E;EE; E E;E E;;FFtextrSprefixct|}d|z|}d|dgfd|ddDzS)zIndent all lines of text after the first line. Args: text: The text to indent. prefix: Used to determine the number of spaces to indent. Returns: str: The indented text.   rcg|]}|zSr$r$)r.linespacess r% z,indent_lines_after_first..s"G"G"GT6D="G"G"Gr1rHN)ro splitlinesjoin)rrn_spaceslinesrs @r%indent_lines_after_firstrsb6{{H 8^F OO  E 99eAhZ"G"G"G"GU122Y"G"G"GG H HHr1c"eZdZdZddZddZdS) AddableDictz3Dictionary that can be added to another dictionary.otherr"ct|}|D]R}||vs|| ||||<||0 ||||z}n#t$r ||}YnwxYw|||<S|S)zlAdd a dictionary to this dictionary. Args: other: The other dictionary to add. rrrXrchunkkeyaddeds r%__add__zAddableDict.__add__s D!! # #C%5:#5"3Zc s''!#Js3EE '''!#JEEE'"c  AAAct|}|D]R}||vs|| ||||<||0 ||||z}n#t$r ||}YnwxYw|||<S|S)zzAdd this dictionary to another dictionary. Args: other: The other dictionary to be added to. rrs r%__radd__zAddableDict.__radd__s E"" # #C%5:#5!#Yc c&&!#Jc2EE &&& IEEE&"c  rN)rrr"r)rrrsrtrurrr$r1r%rrsB==$r1r_T_co _T_contraceZdZdZddZdS) SupportsAddz+Protocol for objects that support addition.xrr"rcdS)z!Add the object to another object.Nr$)rXrs r%rzSupportsAdd.__add__sr1N)rrr"r)rrrsrtrurr$r1r%rrs.55000000r1rAddable)boundaddablesIterable[Addable]Optional[Addable]c&d}|D] }||n||z} |S)zAdd a sequence of addable objects together. Args: addables: The addable objects to add. Returns: Optional[Addable]: The result of adding the addable objects. Nr$rfinalrs r%rfrf!s0 $E::EEM Lr1AsyncIterable[Addable]c6Kd}|23d{V}||n||z}6|S)zAsynchronously add a sequence of addable objects together. Args: addables: The addable objects to add. Returns: Optional[Addable]: The result of adding the addable objects. Nr$rs r%aaddr 0sR $E:::::::eEEM LscneZdZUdZded<dZded<dZded<dZded <d Zd ed <e ddZ dS)ConfigurableFieldaqField that can be configured by the user. Parameters: id: The unique identifier of the field. name: The name of the field. Defaults to None. description: The description of the field. Defaults to None. annotation: The annotation of the field. Defaults to None. is_shared: Whether the field is shared. Defaults to False. rSrcNrrR descriptionz Optional[Any] annotationFr8 is_sharedr"intc8t|j|jfSr-)hashrcrrzs r%__hash__zConfigurableField.__hash__QsTWdo.///r1r"r) rrrsrtru__annotations__rRr rrrrr$r1r%r r ?s GGGD!%K%%%% $J$$$$I 000X000r1r cteZdZUdZded<ded<ded<dZded <dZded <d Zd ed <eddZ dS)ConfigurableFieldSingleOptionaField that can be configured by the user with a default value. Parameters: id: The unique identifier of the field. options: The options for the field. default: The default value for the field. name: The name of the field. Defaults to None. description: The description of the field. Defaults to None. is_shared: Whether the field is shared. Defaults to False. rSrcMapping[str, Any]optionsdefaultNrrRr Fr8rr"rct|jt|j|jfSr-rrctuplerrTrrzs r%rz&ConfigurableFieldSingleOption.__hash__js1TWeDL$5$5$7$788$,GHHHr1r rrrsrtrurrRr rrrr$r1r%rrVs   GGGLLLD!%K%%%%I IIIXIIIr1rcteZdZUdZded<ded<ded<dZd ed <dZd ed <d Zd ed<eddZ dS)ConfigurableFieldMultiOptionaField that can be configured by the user with multiple default values. Parameters: id: The unique identifier of the field. options: The options for the field. default: The default values for the field. name: The name of the field. Defaults to None. description: The description of the field. Defaults to None. is_shared: Whether the field is shared. Defaults to False. rSrcrrz Sequence[str]rNrrRr Fr8rr"rct|jt|jt|jfSr-rrzs r%rz%ConfigurableFieldMultiOption.__hash__s9TWeDL$5$5$7$788% :M:MNOOOr1rrr$r1r%r r os   GGGD!%K%%%%I PPPXPPPr1r cneZdZUdZded<ded<dZded<dZded <dZded <d Zd ed <dZ ded<dS)ConfigurableFieldSpeca Field that can be configured by the user. It is a specification of a field. Parameters: id: The unique identifier of the field. annotation: The annotation of the field. name: The name of the field. Defaults to None. description: The description of the field. Defaults to None. default: The default value for the field. Defaults to None. is_shared: Whether the field is shared. Defaults to False. dependencies: The dependencies of the field. Defaults to None. rSrcr rNrrRr rFr8rr dependencies) rrrsrtrurrRr rrr$r$r1r%r#r#s   GGGOOOD!%K%%%%GI(,L,,,,,,r1r#specsIterable[ConfigurableFieldSpec]list[ConfigurableFieldSpec]cVtt|dd}g}|D]\}}t|t|}t |dkst fd|Dr|gd|dg|z}t||S)aGet the unique config specs from a sequence of config specs. Args: specs: The config specs. Returns: list[ConfigurableFieldSpec]: The unique config specs. Raises: ValueError: If the runnable sequence contains conflicting config specs. c&|jg|jpgRSr-)rcr$ss r%z)get_unique_config_specs..sQT$CQ^-Ar$C$Cr1)rc|jSr-)rcr*s r%r,z)get_unique_config_specs..s PQPTr1rc3$K|] }|kV dSr-r$)r.ofirsts r%r0z*get_unique_config_specs..s'">">!1:">">">">">">r1z6RunnableSequence contains conflicting config specsfor z: )r rnextr*roallrr>)r%groupeduniquercdupesothersmsgr0s @r%get_unique_config_specsr8suCCDDDnnG+-F " " EU e v;;!  s">">">">v">">">>>  MM% 000#Wv-00 S// ! Mr1c.eZdZddddddddd ZddZdS)_RootEventFilterN include_names include_types include_tags exclude_names exclude_types exclude_tagsr<Optional[Sequence[str]]r=r>r?r@rAr"rVcZ||_||_||_||_||_||_dS)zUtility to filter the root event in the astream_events implementation. This is simply binding the arguments to the namespace to make save on a bit of typing in the astream_events implementation. Nr;)rXr<r=r>r?r@rAs r%rYz_RootEventFilter.__init__s:+*(**(r1eventr root_typerSr8cjj jd}nd}|dpg}j|p|djv}j |p|jv}j|pt fd|D}j|o|djv}j |o|jv}j|otfd|D}|S)z&Determine whether to include an event.NTFtagsrRc3*K|] }|jvVdSr-)r>r.tagrXs r%r0z1_RootEventFilter.include_event..s+$T$T#SD,=%=$T$T$T$T$T$Tr1c3*K|] }|jvVdSr-)rArIs r%r0z1_RootEventFilter.include_event..s<&&144,,&&&&&&r1) r<r=r>r=anyr?r@rAr2)rXrDrEinclude event_tagss` r% include_eventz_RootEventFilter.include_events2   &"*!)GGGYYv&&,"   )Dv$2D!DG   )@d.@!@G   (T$T$T$T$T$T$T$T!T!TG   )I%-t7I"IG   )E)43E"EG   (#&&&&8B&&&##Gr1)r<rBr=rBr>rBr?rBr@rBrArBr"rV)rDrrErSr"r8)rrrsrtrYrOr$r1r%r:r:sZ261504151504)))))),r1r:'TypeGuard[Callable[..., AsyncIterator]]c|tj|p(t|dotj|jS)zCheck if a function is an async generator. Args: func: The function to check. Returns: TypeGuard[Callable[..., AsyncIterator]: True if the function is an async generator, False otherwise. __call__)risasyncgenfunctionrrRrls r%is_async_generatorrUs> "4(( 6 4 $ $ 6  &t} 5 5r1#TypeGuard[Callable[..., Awaitable]]c|tj|p(t|dotj|jS)zCheck if a function is async. Args: func: The function to check. Returns: TypeGuard[Callable[..., Awaitable]: True if the function is async, False otherwise. rR)r2iscoroutinefunctionrrRrTs r%is_async_callablerYs> #D)) 7 4 $ $ 7  ' 6 6r1)rr r!rr"r )r'r(r)rr"r*)r6r7r"r8)r"r8)r!rMrFrrKr8r"rM)rlr r"r)rlr r"r)rlr r"r)rrSrrSr"rS)rrr"r)rrr"r)r%r&r"r')rlr r"rP)rlr r"rV)Kru __future__rr_r2rrcollections.abcrr contextvarsr functoolsrr itertoolsr typingr r r r rrrrtyping_extensionsrrlangchain_core.utils.pydanticrrrrrrlangchain_core.runnables.schemarrrr&r5r@rDrGrLrO NodeVisitorrQrwrrrrrrrdictrSrrrrrrfr r rr AnyConfigurableFieldr#r8r:rUrYr$r1r%rfs!!"""""" --------                    21111111766666 <<;;;;;t,,, T * * *    LLLL$             10000 DI(5.5.5.5.5.#/5.5.5.p2;2;2;2;2;2;2;2;j7D7D7D7D7D7D7D7Dt/>/>/>/>/>/>/>/>d,,,,,co,,,.(::::. 3''''T I I I I %%%%%$sCx.%%%P 4((( GKt 4 4 4 00000(9e+,000 ');sCx#8 9 9 9        00000 000.IIIIIJIII2PPPPP:PPP246RR -----J---.@44444444n&r1