g3fi@? 6dZddlZddlZddlZddlZddlZddlZddlmZm Z m Z ddl m Z ddl mZmZddlmZddlmZddlmZdd lmZmZdd lmZdd lmZd eed fdefdZdeddfdZ ejBdejde e"fdZ#ddddededzdedzdefdZ$ d>dededzdedzdedzdedzddf dZ%dede&efdZ'd e(eefd!e&ede(eeffd"Z)d#e(eefd e(eefd!e&ede(eeffd$Z*d%eezdefd&Z+Gd'd(Z,e,Z-ed)edegeffd*Z.ed)ed+edegeffd,Z.ed)e ed+edegeffd-Z.ed)ed.edegeffd/Z.ed)ee ezd+ed.edzdegeffd0Z.ed)ed+dd.edzdegedzffd1Z.ed)ee ezd+ddegedzffd2Z.e-dd3d)ee ezd+ee,zdzd.edzdegefegedzfzfd4Z.ed)ee ezdegeffd5Z/ed)ed+edegeffd6Z/ed)ee ezd+ddegedzffd7Z/ed)ed.edegeffd8Z/e-dd3d)ee ezd+ee,zdzd.edzdegedzfegefzfd9Z/d:Z0 d;Z1 dz?xor_args..decorator..wrapper..(sJ#fjjo.IAJs!!rz, zFExactly one argument in each of the following groups must be defined: )sum enumeratejoin ValueError) rr arg_groupcountsicountinvalid_groupsinvalid_group_namesmsgrrs ` rwrapperz,xor_args..decorator..wrapper$s ",JJJF1:&0APHAuUaZaPNPIW&XAtyyA'?&X#&X "5679 !o%(( (Q&XsB BB B) functoolswrapsr)rr+rs` r decoratorzxor_args..decorator#s:   )3 )# )# )  )")r)rr.s` rxor_argsr0sX* r/responsect |jy#t$r}t|j|d}~wwxYw)zRaise an error with the response text. Args: response: The response to check for errors. Raises: ValueError: If the response has an error status code. N)raise_for_statusr r#text)r1es rraise_for_status_with_textr6;s4/!!# /'Q./s 727dt_valuec#KGfddtjtj}t_ tj|t_y#|t_wxYww)a{Context manager for mocking out datetime.now() in unit tests. Args: dt_value: The datetime value to use for datetime.now(). Yields: The mocked datetime class. Example: ```python with mock_now(datetime.datetime(2011, 2, 3, 10, 11)): assert datetime.datetime.now() == datetime.datetime(2011, 2, 3, 10, 11) ``` cXeZdZdZeeddejdzddffd Zy)mock_now..MockDateTimez3Mock datetime.datetime.now() with a fixed datetime.Ntzr MockDateTimec jjjjjj j jSN)yearmonthdayhourminutesecond microsecondtzinfo)clsr;r<r7s rnowz"mock_now..MockDateTime.now^sL    $$  r/r>) __name__ __module__ __qualname____doc__ classmethodrdatetimerFrH)r<r7srr<r:[s7A   4/ >    r/r<N)rN)r7 real_datetimer<s` @rmock_nowrPJsO"x(($%%M$H*)Ms8A+A A+ A((A+)pip_namepackage module_namerQrRc tj||}|S#ttf$rD}|xs&|j dddj dd}d|d|d }t||d }~wwxYw) a~Dynamically import a module. Raise an exception if the module is not installed. Args: module_name: The name of the module to import. pip_name: The name of the module to install with pip. package: The package to import the module from. Returns: The imported module. Raises: ImportError: If the module is not installed. .r)maxsplitr_-zCould not import z5 python package. Please install it with `pip install z`.N) importlib import_module ImportErrorModuleNotFoundErrorsplitreplace)rSrQrRmoduler5r*s r guard_importr`us$&((g> M , -&V{00q0A!DLLSRUV }-33;*B @ #A% &sA-?A((A- lt_version lte_version gt_version gte_versioncftt|}|%|t|k\rd|d|d|d}t||%|t|kDrd|d|d|d}t||%|t|krd|d|d|d}t||&|t|krd|d|d|d}t|yy) aCheck the version of a package. Args: package: The name of the package. lt_version: The version must be less than this. lte_version: The version must be less than or equal to this. gt_version: The version must be greater than this. gte_version: The version must be greater than or equal to this. Raises: ValueError: If the package version does not meet the requirements. Nz Expected z version to be < z . Received rUz version to be <= z version to be > z version to be >= )r rr#)rRrarbrcrdimported_versionr*s rcheck_package_versionrgs!(WW-."2eJ6G"Gy 1*[  # o#3eK6H#Hy 2;-{  # o"2eJ6G"Gy 1*[  # o#3eK6H#Hy 2;-{  # o $Ir/ pydantic_clsct}t|rd|jjD]E}|j |j |j s+|j |jG|S|jjD]>\}}|j ||js$|j |j@|S)zGet field names, including aliases, for a pydantic class. Args: pydantic_cls: Pydantic class. Returns: Field names. ) setr __fields__valuesaddname has_aliasalias model_fieldsitems)rhall_required_field_namesfieldrns rget_pydantic_field_namesrus #u|,!,,335 :E $ ( ( 4(,,U[[9 : $# (44::< :KD% $ ( ( .{{(,,U[[9 : $#r/rlrsc |jdi}t|D]Q}||vrd|d}t|||vstjd|d|d|dd |j |||<S|j |j}|r6tjd |d d |D]}|j |||<||d<|S) aBuild `model_kwargs` param from Pydantic constructor values. Args: values: All init args passed in by user. all_required_field_names: All required field names for the pydantic class. Returns: Extra kwargs. Raises: ValueError: If a field is specified in both `values` and `extra_kwargs`. ValueError: If a field is specified in `model_kwargs`. model_kwargsFound  supplied twice. WARNING! + is not default parameter. F was transferred to model_kwargs. Please confirm that  is what you intended. stacklevel Parameters a should be specified explicitly. Instead they were passed in as part of `model_kwargs` parameter.)rlistr#warningswarnpop intersectionkeys)rlrs extra_kwargs field_namer*invalid_model_kwargsks r_build_model_kwargsrs"::nb1L6l >  %:,&67CS/ ! 5 5 MMj\* %%/L0FJ   (.zz*'=L $ >4@@ARARATU ./0O P & ,A$((+F1I ,*F> Mr/rc (t|D]Q}||vrd|d}t|||vstjd|d|d|dd|j |||<S|j |j }|rd |d }t||S) a]Build extra kwargs from values and extra_kwargs. !!! danger "DON'T USE" Kept for backwards-compatibility but should never have been public. Use the internal `_build_model_kwargs` function instead. Args: extra_kwargs: Extra kwargs passed in by user. values: Values passed in by user. all_required_field_names: All required field names for the pydantic class. Returns: Extra kwargs. Raises: ValueError: If a field is specified in both `values` and `extra_kwargs`. ValueError: If a field is specified in `model_kwargs`. rxryrzr{r|r}r~rrr)rr#rrrrr)rrlrsrr*rs rbuild_extra_kwargsrs06l >  %:,&67CS/ ! 5 5 MMj\* %%/L0FJ   (.zz*'=L $ >4@@ARARATU./0O P o r/valuec<t|tr|St|S)zConvert a string to a SecretStr if needed. Args: value: The value to convert. Returns: The SecretStr value. ) isinstancer )rs rconvert_to_secret_strr9s%# U r/ceZdZdZy)_NoDefaultTypez.Type to indicate no default value is provided.N)rIrJrKrLr/rrrGs8r/rkeycyr>rrs rfrom_envrNs03r/defaultcyr>rrrs rrrRsADr/cyr>rrs rrrVsKNr/ error_messagecyr>rrrs rrrZsGJr/cyr>rrrrs rrr^sr/cyr>rrs rrrd #r/cyr>rrs rrrjrr/)rrc.dtdzffd }|S)aCreate a factory method that gets a value from an environment variable. Args: key: The environment variable to look up. If a list of keys is provided, the first key found in the environment will be used. If no key is found, the default value will be used if set, otherwise an error will be raised. default: The default value to return if the environment variable is not set. error_message: the error message which will be raised if the key is not found and no default value is provided. This will be raised as a ValueError. Returns: factory method that will look up the value from the environment. rNcxtttfr.D])}|tjvstj|cStt r%tjvrtjStt t dfrSr tdddd}t|)zGet a value from an environment variable. Raises: ValueError: If the environment variable is not set and no default is provided. Returns: The value from the environment. N Did not find &, please add an environment variable `` which contains it, or pass `` as a named parameter.)rrtupleosenvironstrtyper#rr*rrrs rget_from_env_fnz!from_env..get_from_env_fns cD%= ) ) ?::a=( ) c3 C2::$5::c? " gT$Z0 1N ]+ +C5!, . or/)r)rrrrs``` rrrps.S4Z8 r/cyr>rrs rsecret_from_envrsMPr/cyr>rrs rrrsNQr/cyr>rrs rrrs&)r/cyr>rrs rrrsTWr/c.dtdzffd }|S)aSecret from env. Args: key: The environment variable to look up. default: The default value to return if the environment variable is not set. error_message: the error message which will be raised if the key is not found and no default value is provided. This will be raised as a ValueError. Returns: factory method that will look up the secret from the environment. rNctttfr7D]2}|tjvst tj|cStt r.tjvrt tjStt r t Syr tdddd}t|)zGet a value from an environment variable. Raises: ValueError: If the environment variable is not set and no default is provided. Returns: The secret from the environment. Nrrrr)rrrrrr rr#rs rget_secret_from_envz,secret_from_env..get_secret_from_envs cD%= ) 4 ?$RZZ]33 4 c3 C2::$5RZZ_- - gs #W% % ? ]+ +C5!, . or/r )rrrrs``` rrrs(T!1: r/lc_zlc_run-id_valc.|xsttS)aTEnsure the ID is a valid string, generating a new UUID if not provided. Auto-generated UUIDs are prefixed by `'lc_'` to indicate they are LangChain-generated IDs. Args: id_val: Optional string ID value to validate. Returns: A string ID, either the validated provided value or a newly generated UUID4. )LC_AUTO_PREFIXr )rs r ensure_idrs  1'y11r/)NNNN)3rL contextlibrNr,rYrrcollections.abcrrrimportlib.metadatartypingrruuidr packaging.versionr pydanticr requestsr rtyping_extensionsrlangchain_core.utils.pydanticrrrr0r6contextmanagerrrPr`rgrjrudictrrrr _NoDefaultrrr LC_ID_PREFIXrrr/rrs  88& #(& %S/ h F / /d / '*x(('*Xd^'*'*V15D#&:?BTz@""!" , ,d ,t,d , t ,  ,^$3$3s8$.* cN*!#h* #s(^*\/sCx./ cN/"#h/ #s(^ /d S Y 99   3#3Xb#g.3 3 D#DcDhr3w.?D D N(3-NNS8IN N J#JCJHRW4EJ J  x} -0ADt b#g   # #!#25*# b#*n# #  # x} #-1# b#*n# #,6 $ 3 x} 3> !D ( 3 : 3 b#g"cDj.11 3l Px},PHR]4KP P QQSQXb)m5LQ Q ) x} )-1) b)d""#) )  WW3W8B M;RW W,6 $ 1 x} 1> !D ( 1 : 1 b)d""#hr9}&== 1hI  2cDj 2S 2r/