6|hddZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlmZmZddlmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"ddl#m$Z%ddl&Z&ddl'Z'ddl(m)Z)ddl*m+Z+dd l,m-Z.ej^e0Z1Gd d e2Z3Gd d e3Z4Gdde3Z5Gdde3Z6Gdde3Z7Gdde3Z8Gdde3Z9Gdde3Z:Gdde3Z;Gdde<Z=Gdde=Z>dQdRd Z?dSd!Z@dTd"ZA dUd#ZBdVd$ZCejd%&dWd'ZEdXd(ZFdYd)ZGdZd*ZHd[d+ZId\d,ZJd]d-ZKd^d.ZLejd%&d_d/ZMd`d0ZNdad1ZOejd2& dQd3d4 dbd5ZPejd%&dcddd6ZQGd7d8ejZSGd9d:ejZTGd;dZZej dQ dgd?Z[ej dQ dhd@Z\didAZ]e dBZ^ dj dkdCZ_dldDZ`dmdEZadndFZbe)dGZcGdHdIeZddodJZedpdKZfdqdLZgejdM&drdNZhdsdtdOZidudPZjy)vzGeneric utility functions.) annotationsN)FutureThreadPoolExecutor)AnyCallableDict GeneratorIterableIteratorListLiteralMappingOptionalSequenceTupleTypeVarUnioncast)parse) ParamSpec)Retry)schemasceZdZdZy)LangSmithErrorz=An error occurred while communicating with the LangSmith API.N__name__ __module__ __qualname____doc__N/var/www/html/test/engine/venv/lib/python3.12/site-packages/langsmith/utils.pyrr0sGr!rceZdZdZy)LangSmithAPIErrorz9Internal server error while communicating with LangSmith.Nrr r!r"r$r$4sCr!r$ceZdZdZy)LangSmithRequestTimeoutz*Client took too long to send request body.Nrr r!r"r&r&8s4r!r&ceZdZdZy)LangSmithUserErrorzAUser error caused an exception when communicating with LangSmith.Nrr r!r"r(r(<sKr!r(ceZdZdZy)LangSmithRateLimitErrorz7You have exceeded the rate limit for the LangSmith API.Nrr r!r"r*r*@sAr!r*ceZdZdZy)LangSmithAuthErrorz-Couldn't authenticate with the LangSmith API.Nrr r!r"r,r,Ds7r!r,ceZdZdZy)LangSmithNotFoundErrorz%Couldn't find the requested resource.Nrr r!r"r.r.Hs/r!r.ceZdZdZy)LangSmithConflictErrorzThe resource already exists.Nrr r!r"r0r0L&r!r0ceZdZdZy)LangSmithConnectionErrorz&Couldn't connect to the LangSmith API.Nrr r!r"r3r3P0r!r3ceZdZdZy)LangSmithWarningzBase class for warnings.Nrr r!r"r6r6Ws"r!r6ceZdZdZy)LangSmithMissingAPIKeyWarningzWarning for missing API key.Nrr r!r"r8r8[r1r!r8cddlm}m}|xs|}|d|dS|rytdtdd}|d k(S) z"Return True if tracing is enabled.r)get_current_run_treeget_tracing_contextenabledT TRACING_V2TRACINGdefaulttrue)langsmith.run_helpersr:r; get_env_var)ctxr:r;tc var_results r"tracing_is_enabledrH_sQO %#%B  )} )}\;yRT3UVJ  r!c"tdddk(S)z"Return True if testing is enabled. TEST_TRACKINGr?r@falserDr r!r"test_tracking_is_disabledrMrs  3w >>r!cdfd }|S)z7Validate specified keyword args are mutually exclusive.cHtjdfd }|S)Nc>Dcgc]}tfd|D}}t|Dcgc] \}}|dk7s |}}}|r>|Dcgc]}dj|}}tddj| |iScc}wcc}}wcc}w)z3Validate exactly one arg in each group is not None.c3FK|]}j|dyw)N)get).0argkwargss r" z?xor_args..decorator..wrapper..sJ#fjjo.IAJs!!rRz, zFExactly one argument in each of the following groups must be defined: )sum enumeratejoin ValueError) argsrV arg_groupcountsicountinvalid_groupsinvalid_group_names arg_groupsfuncs ` r"wrapperz,xor_args..decorator..wrapper{s ",JJJF1:&0APHAuUaZaPNPIW&XAtyyA'?&X#&X  "5679 (( (Q&XsB BB B)r\rrVrreturnr) functoolswraps)rdrercs` r" decoratorzxor_args..decoratorzs%   )  ) r!)rdrrfrr )rcris` r"xor_argsrjws( r!c |jy#tj$r/}tjt||j|d}~wwxYw)z&Raise an error with the response text.N)raise_for_statusrequests HTTPErrorstrtext)responsees r"raise_for_status_with_textrssF?!!#   ?  Q7Q>?sA*AAcRt|tjr |jS|S)zGet the value of a string enum.) isinstanceenumEnumvalue)enus r"get_enum_valuerzs#tyy!yy Jr!rR)maxsizec0tj||y)z4Log a message at the specified level, but only once.N)_LOGGERlog)levelmessages r"log_oncers KKwr!c|s tdd|vr9d|vrtd|d|ddjddjSd |vrtd |d |d S) NMessage is empty.lcid*Unexpected format for serialized message: z Message does not have an id.Messager?type&Unexpected format for stored message: z Message does not have a type.)r[replacelowerrs r"_get_message_typers ,-- w w  E C JZV0Dc%J  r!d) LANGSMITH LANGCHAIN) namespacesc|Dcgc] }|d| }}|D]'}tjj|}|%|cS|Scc}w)a+Retrieve an environment variable from a list of namespaces. Args: name (str): The name of the environment variable. default (Optional[str], optional): The default value to return if the environment variable is not found. Defaults to None. namespaces (Tuple, optional): A tuple of namespaces to search for the environment variable. Defaults to ("LANGSMITH", "LANGCHAINs"). Returns: Optional[str]: The value of the environment variable if found, otherwise the default value. _)osenvironrS)namerAr namespacenamesrxs r"rDrDes[(5? ?y {!D6 " ?E ? t$  L N @sAc vtjjdtdtd|rdndS)z,Get the project name for a LangSmith tracer.HOSTED_LANGSERVE_PROJECT_NAMEPROJECTSESSIONrANr@)rrrSrD)return_default_values r"get_tracer_projectrs= ::>> (  0D9$   r!c.eZdZdZddfd ZddZxZS)FilterPoolFullWarningzFFilter urrllib3 warnings logged when the connection pool isn't reused.c2t||||_y)zInitialize the FilterPoolFullWarning filter. Args: name (str, optional): The name of the filter. Defaults to "". host (str, optional): The host to filter. Defaults to "". N)super__init___host)selfrhost __class__s r"rzFilterPoolFullWarning.__init__s  r!cH|j}d|vry|j|vS)zJurllib3.connectionpool:Connection pool is full, discarding connection: ...z.Connection pool is full, discarding connectionT) getMessagerrrecordmsgs r"filterzFilterPoolFullWarning.filters*! ;3 Fzz$$r!)r?r?)rrorrorfNonerfbool)rrrrrr __classcell__rs@r"rrsP%r!rceZdZdZddZy)FilterLangSmithRetryz!Filter for retries from this lib.c*|j}d|vS)z!Filter retries from this library.LangSmithRetry)rrs r"rzFilterLangSmithRetry.filters!s**r!Nr)rrrrrr r!r"rrs ++r!rceZdZdZy)rz&Wrapper to filter logs with this name.Nrr r!r"rrr4r!rc#Kt5|D]}|j| ddd dt5|D]} |j| dddy#1swY8xYw#t$rtj dYMwxYw#1swYyxYw#t5|D]8} |j|#t$rtj dY6wxYw dddw#1swYwxYwxYww)zTemporarily adds specified filters to a logger. Parameters: - logger: The logger to which the filters will be added. - filters: A sequence of logging.Filter objects to be temporarily added to the logger. NzFailed to remove filter) _FILTER_LOCK addFilter removeFilter BaseExceptionr}warning)loggerfiltersrs r" filter_logsrs % %F   V $ %%?  ?! ??''/ ? ? ?%%%?OO$=>?  ? ?\ ?! ??''/$?OO$=>? ? ? ? ?sC4AC4BC4B A&B  C4A#C4&BB BB  BC4C1C%$B65C%6C C%C C% C1%C. *C11C4c$||StddS)zGet the testing cache directory. Args: cache (Optional[str]): The cache path. Returns: Optional[str]: The cache path if provided, otherwise the value from the LANGSMITH_TEST_CACHE environment variable. N TEST_CACHEr@rL)caches r" get_cache_dirrs  |T 22r!c#K ddl}ddlm}|j dfd }t j j|\}}|j|jds|jdrdnd |d gd d d g|}|j|5ddddy#t$r tdwxYw#1swYyxYww)Use a cache for requests.rNzNvcrpy is required to use caching. Install with:pip install -U "langsmith[vcr]")_patchcFrtfdDryi_S)Nc3TK|]}jj|!ywN)url startswith)rTrrequests r"rWz>with_cache.._filter_request_headers..s V 6 6t <Vs%()anyheaders)r ignore_hostss`r"_filter_request_headersz+with_cache.._filter_request_headerss" CVVVr!z.yamlz.ymlyamljson new_episodes)urimethodpathbody authorizationz Set-Cookie) serializercassette_library_dir record_modematch_onfilter_headersbefore_record_request)rrrfr) vcr ImportErrorlangsmith._internalr patch_urllib3rr splitVCRendswith use_cassette)r rrrr cache_dir cache_filels_vcrs ` r" with_cachers  <! GGMM$/Iz WW""7+z/B/B6/J &#2'65 F   Z ( A   .   @s4C B%BC B= C %B::C =CC c#dK|t||5ddddydy#1swYyxYww)rN)r)r rs r"with_optional_cacher!s9   l +       s0$0-0ctjtj}|Dcgc] }d|vs| }}dj |Scc}w)Nz langsmith/r?) tracebackformat_exceptionsysexc_inforZ)tb_lineslinefiltered_liness r" _format_excr*sG))3<<>:H'/Lt ""Ms A A Tc Jt|}t|dd}| |Sk\r|St|trB|j Dcic]&\}}t |dzt |dz(c}}St|tr|Dcgc]}t |dzc}St|trtfd|DSt|tr|Dchc]}t |dzc}S|S#t$rYwxYwcc}}wcc}wcc}w)N __deepcopy__rRc3@K|]}t|dzyw)rRN) _middle_copy)rTitem_depth max_depthmemos r"rWz_middle_copy..=s U\$i!DUs) rrrruritemsr/listtupleset) valr3r2r1clscopierkvr0s ``` r"r/r/&s6 s)C S.$ /F  $<  #t   1 D)VaZ 8,4FQJ;   #tLOPD T4FQJ?PP#uUQTUUU#sLOPD T4FQJ?PP J%     QQs#D+DD)D  DDci} tj||S#t$r5}tj dt |t ||cYd}~Sd}~wwxYw)zDeep copy a value with a compromise for uncopyable objects. Args: val: The value to be deep copied. Returns: The deep copied value. zFailed to deepcopy input: %sN)copydeepcopyrr}debugreprr/)r8r3rrs r" deepish_copyrBDsQD'}}S$'' '  4d1g>C&& 's A*A AAc\ddlm}|j|}|j|}||k\S)zGCheck if the current version is greater or equal to the target version.r)version) packagingrDr)current_versiontarget_versionrDcurrenttargets r"is_version_greater_or_equalrJYs,!mmO,G ]]> *F f r!cp|r6|jddkDs"|jds|jdrtd||j dd}|d}t |dkDr|dnd}d|vr,|j dd\}}|r|std||||fS|std|d||fS)aEParse a string in the format of owner/name:hash, name:hash, owner/name, or name. Args: identifier (str): The prompt identifier to parse. Returns: Tuple[str, str, str]: A tuple containing (owner, name, hash). Raises: ValueError: If the identifier doesn't match the expected formats. /rRzInvalid identifier format: :rlatest-)r`rrr[rr) identifierparts owner_namecommitownerrs r"parse_prompt_identifierrUbs    C 1 $   %   s #6zlCDD   S! $EqJUaU1XXF j &&sA. tD::,GH HdF""::,GH HJ&&r!PcZeZdZdZ dfd Zddd dfdZxZS) ContextThreadPoolExecutorz?ThreadPoolExecutor that copies the context to the child thread.c t|ttdtft j tjj|g|i|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. .) rsubmitrrr+rgpartial contextvars copy_contextrun)rrdr\rVrs r"rZz ContextThreadPoolExecutor.submits\ w~ a !!,,.22D;?CI   r!NrRtimeout chunksizectt|dDcgc]}tjc}dfd }t ||g|||dScc}w)aReturn an iterator equivalent to stdlib map. Each function will receive its own copy of the context from the parent thread. Args: fn: A callable that will take as many arguments as there are passed iterables. timeout: The maximum number of seconds to wait. If None, then there is no limit on the wait time. chunksize: The size of the chunks the iterable will be broken into before being passed to a child process. This argument is only used by ProcessPoolExecutor; it is ignored by ThreadPoolExecutor. Returns: An iterator equivalent to: map(func, *iterables) but the calls may be evaluated out-of-order. Raises: TimeoutError: If the entire result iterator could not be generated before the given timeout. Exception: If fn(*args) raises for any values. rcFjjg|Sr)popr^)r\contextsfns r" _wrapped_fnz2ContextThreadPoolExecutor.map.._wrapped_fns!%8<<>%%b040 0r!r_)r\rrfr+)rangerr\r]rmap) rrfr`ra iterablesrrgrers ` @r"rizContextThreadPoolExecutor.mapsb<9>c)A,>O8PQ1K,,.Q 1w{      RsA)rdzCallable[P, T]r\zP.argsrVzP.kwargsrfz Future[T]) rfzCallable[..., T]rjz Iterable[Any]r`zOptional[float]raintrfz Iterator[T])rrrrrZrirrs@r"rXrXssI       :$( ( ( "( ! (  (  ( ( r!rXc|xstttdd}|js t d|jjdjdj dS)zBGet the LangSmith API URL from the environment or the given value.ENDPOINTzhttps://api.smith.langchain.comr@z!LangSmith API URL cannot be empty"'rL)rrorDstripr(rstrip)api_url_api_urls r" get_api_urlrtsj$  5 H >>  !DEE >>  ! !# & , ,S 1 8 8 ==r!c||n tdd}||jsy|jjdjdS)z8Get the API key from the environment or the given value.NAPI_KEYr@rnro)rDrp)api_keyapi_key_s r" get_api_keyrysK!-w;yRV3WHx~~/ >>  ! !# & , ,S 11r!c tj|jjdd}t j |}|dk(xs$|j dxs|j dS#tj$rYywxYw)zCheck if the URL is localhost. Parameters ---------- url : str The URL to check. Returns: ------- bool True if the URL is localhost, False otherwise. rMrz 127.0.0.1z0.0.0.0z::F) urllib_parseurlsplitnetlocrsocket gethostbynamergaierror)rr}ips r" _is_localhostrsz&&s+2288=a@  ! !& )[ SBMM)$<S d@SS ??sA0A33B B c|r|Stj|}t|rd}|St|jj drOt|jj ddd}tj|j|}|St|jjdrd}|St|jjdrd }|Sd }|S) z1Get the host URL based on the web URL or API URL.zhttp://localhostz/apirRr)r zeu.zhttps://eu.smith.langchain.comzdev.zhttps://dev.smith.langchain.comzhttps://smith.langchain.com) r{urlparserror rrsplit urlunparse_replacer}r)web_urlrr parsed_urllinknew_paths r" get_host_urlrs&&w/JW! K Z__  & &v .z'..vq9!<&&z':':':'IJ K Z    * *5 1/ K Z    * *6 20 K- Kr!c|dkDs t|s t|St|dr |jSt |t j rt|j|dzSt|drQt|dr,t|jdr|jjSt|j|dzSt|S)NrrrR__call__r) callablerorrrurgr[_get_function_namerdrr)rfdepths r"rrs qy 2wr:{{"i''(!"''51955r: 2{ # j(I<<(( (!"++uqy99 r7Nr!cB|duxs|dk(xs|dk(xs |dk(xs|dk(S)zCheck if the value is truish. Args: val (Any): The value to check. Returns: bool: True if the value is truish, False otherwise. TrBTrueTRUE1r )r8s r" is_truishr s3 $; W#- W3&= WC6M WSTWZWr!r)rEzOptional[dict]rfzUnion[bool, Literal['local']]r)rczTuple[str, ...]rfr)rqz(Union[requests.Response, httpx.Response]rfr)ryzUnion[enum.Enum, str]rfro)rrkrrorfr)rMapping[str, Any]rfro)rrrfr)rrrfDict[str, Any])rrrfzList[Dict[str, Any]])rrrfr)rrrfro)rrrfro)rfz List[str])rzls_schemas.BaseMessageLikerfr)robjectrfr)rrorA Optional[str]rrrfr)T)rfr)rzlogging.LoggerrzSequence[logging.Filter]rfGenerator[None, None, None])rrrfr)r zUnion[str, pathlib.Path]rOptional[Sequence[str]]rfr)r z"Optional[Union[str, pathlib.Path]]rrrfr)rfro)r) r8r+r3zDict[int, Any]r2rkr1rkrfr+)r8r+rfr+)rFrorGrorfr)rProrfzTuple[str, str, str])rrrrfro)rwrrfr)rrorfr)rrrrro)r)rfrrrkrfro)r8rrfr)kr __future__r contextlibr\r>rvrgloggingrpathlibr~rr% threadingr#concurrent.futuresrrtypingrrrr r r r r rrrrrrrurllibrr{httpxrmtyping_extensionsr urllib3.utilr langsmithr ls_schemas getLoggerrr} Exceptionrr$r&r(r*r,r.r0r3 UserWarningr6r8rHrMrjrsrz lru_cacherrrrrrrrrrrrDrFilterrrrRLockrcontextmanagerrrrr!r*r+r/rBrJrUrVrXrtryrrrrr r!r"rs "    9") '+ '  H %HYHDD5n5LLBnB880^0'^'1~1#{#'$4' &? 4?6? ?Q   &&8P(9:L2 $Q 4 $S!"3      "6Q &%GNN%*+7>>+1U1y  ? ?%=? ??2 3 LP' "'2I' ''T -1  , ) !  # CLEF  -0>A<'* 'FcND 2D N >2*Q &$ Xr!