NgbdZddlmZddlZddlZddlZddlZddlZddlZddl 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>ddd$Z?dd&Z@dd*ZAdd.ZBdd2ZCejDd34dd8ZEdd:ZFdd;ZGdd=ZHdd@ZIddBZJddCZKddDZLejDd34ddFZMddIZNddLZOejDdM4 ddNdOddUZPejDd34dddWZQGdXdYejRZSGdZd[ejRZTGd\d]e+ZUejVZWejXddcZYddeZZejX dddjZ[ejX dddlZ\ddmZ]e dnZ^ ddduZ_ddvZ`ddyZadd|Zbe)d}ZcGd~deZdddZeddZfddZgejDd4ddZhdddZidS)zGeneric utility functions.) annotationsN)FutureThreadPoolExecutor)AnyCallableDict GeneratorIterableIteratorListLiteralMappingOptionalSequenceTupleTypeVarUnioncast)parse) ParamSpec)Retry)schemasceZdZdZdS)LangSmithErrorz=An error occurred while communicating with the LangSmith API.N__name__ __module__ __qualname____doc__K/var/www/html/ai-engine/env/lib/python3.11/site-packages/langsmith/utils.pyrr0sGGGGr!rceZdZdZdS)LangSmithAPIErrorz9Internal server error while communicating with LangSmith.Nrr r!r"r$r$4sCCCCr!r$ceZdZdZdS)LangSmithRequestTimeoutz*Client took too long to send request body.Nrr r!r"r&r&8s4444r!r&ceZdZdZdS)LangSmithUserErrorzAUser error caused an exception when communicating with LangSmith.Nrr r!r"r(r(<sKKKKr!r(ceZdZdZdS)LangSmithRateLimitErrorz7You have exceeded the rate limit for the LangSmith API.Nrr r!r"r*r*@sAAAAr!r*ceZdZdZdS)LangSmithAuthErrorz-Couldn't authenticate with the LangSmith API.Nrr r!r"r,r,Ds7777r!r,ceZdZdZdS)LangSmithNotFoundErrorz%Couldn't find the requested resource.Nrr r!r"r.r.Hs////r!r.ceZdZdZdS)LangSmithConflictErrorzThe resource already exists.Nrr r!r"r0r0L&&&&r!r0ceZdZdZdS)LangSmithConnectionErrorz&Couldn't connect to the LangSmith API.Nrr r!r"r3r3P0000r!r3ceZdZdZdS)LangSmithWarningzBase class for warnings.Nrr r!r"r6r6Ws""""r!r6ceZdZdZdS)LangSmithMissingAPIKeyWarningzWarning for missing API key.Nrr r!r"r8r8[r1r!r8ctxOptional[dict]returnUnion[bool, Literal['local']]cddlm}m}|p |}|d|dS|rdStdtdd  }|d kS) z"Return True if tracing is enabled.r)get_current_run_treeget_tracing_contextenabledNT TRACING_V2TRACINGdefaulttrue)langsmith.run_helpersr>r? get_env_var)r9r>r?tc var_results r"tracing_is_enabledrK_sOOOOOOOO %##%%B  )} )}t\;yRT3U3U3UVVVJ  r!boolc,tdddkS)z"Return True if testing is enabled. TEST_TRACKINGrCrDfalserHr r!r"test_tracking_is_disabledrQrs  3 3 3w >>r! arg_groupsTuple[str, ...]rcdfd }|S)z7Validate specified keyword args are mutually exclusive.funcrr;cLtjdfd }|S)Nargsrkwargsr;cfdD}dt|D}|r3fd|D}tdd||iS)z3Validate exactly one arg in each group is not None.cFg|]}tfd|DS)c3FK|]}|dVdS)N)get).0argrXs r" zJxor_args..decorator..wrapper...s2JJ#fjjoo.IA.I.I.I.IJJr!)sum)r^ arg_grouprXs r" z@xor_args..decorator..wrapper..~sFJJJJJJJJJr!c$g|] \}}|dk |Sr\r )r^icounts r"rcz@xor_args..decorator..wrapper..s!PPPHAuUaZZaZZZr!cFg|]}d|S), )join)r^rfrRs r"rcz@xor_args..decorator..wrapper..s)&X&X&XAtyyA'?'?&X&X&Xr!zFExactly one argument in each of the following groups must be defined: ri) enumerate ValueErrorrj)rWrXcountsinvalid_groupsinvalid_group_namesrRrUs ` r"wrapperz,xor_args..decorator..wrapper{s!+FQP &0A0APPPN &X&X&X&X&X&X&X# 9 "56699 4((( (r!)rWrrXrr;r) functoolswraps)rUrprRs` r" decoratorzxor_args..decoratorzsC    ) ) ) ) ) )   ) r!)rUrr;rr )rRrss` r"xor_argsrtws)( r!response(Union[requests.Response, httpx.Response]Nonec |dS#tj$r-}tjt||j|d}~wwxYw)z&Raise an error with the response text.N)raise_for_statusrequests HTTPErrorstrtext)rues r"raise_for_status_with_textrs^?!!#####  ??? Q77Q>?sA(AAenuUnion[enum.Enum, str]r|cHt|tjr|jS|S)zGet the value of a string enum.) isinstanceenumEnumvalue)rs r"get_enum_valuers##ty!!y Jr!r\)maxsizelevelintmessagec<t||dS)z4Log a message at the specified level, but only once.N)_LOGGERlog)rrs r"log_oncers KKwr!Mapping[str, Any]c|stdd|vrKd|vrtd|d|ddddSd |vrtd |d |d S) NMessage is empty.lcid*Unexpected format for serialized message: z Message does not have an id.MessagerCtype&Unexpected format for stored message: z Message does not have a type.)rlreplacelowerrs r"_get_message_typers .,--- w w  0W000 t}R ((B77==???  1111 vr!c|stdd|vrd|vrtd|d|dSd|vrtd|d|dS) NrrrXrz Message does not have kwargs.datarz Message does not have data.)rlrs r"_get_message_fieldsrs .,--- w 7 " "1W111 x   //// vr!Dict[str, Any]cHt|}t|}||dS)z&Extract message from a message object.rr)rr)r message_type message_datas r"_convert_messagers*$W--L&w//L , 7 77r!inputsList[Dict[str, Any]]cd|vrd|dDSd|vrt|dgStd|d)aGExtract messages from the given inputs dictionary. Args: inputs (Mapping[str, Any]): The inputs dictionary. Returns: List[Dict[str, Any]]: A list of dictionaries representing the extracted messages. Raises: ValueError: If no message(s) are found in the inputs dictionary. messagesc,g|]}t|Sr )r)r^rs r"rcz,get_messages_from_inputs..s!LLLg ))LLLr!rz-Could not find message(s) in run with inputs .)rrl)rs r"get_messages_from_inputsrscVLL 9KLLLLF  !23344 NVNNN O OOr!outputscd|vrtd|d|d}t|dkr#tdt|d|d|d}d|vrtd |d t|dS) a'Retrieve the message generation from the given outputs. Args: outputs (Mapping[str, Any]): The outputs dictionary. Returns: Dict[str, Any]: The message generation. Raises: ValueError: If no generations are found or if multiple generations are present. generations,No generations found in in run with output: rr\z3Chat examples expect exactly one generation. Found z generations: rrz"Unexpected format for generation: z%. Generation does not have a message.)rllenrrrfirst_generations r"#get_message_generation_from_outputsrsG##RRRRSSS-(K ;1 E+&& E E6A E E E   #1~((( 31A 3 3 3    ,Y7 8 88r!cd|vr|dSd|vr6|d}t|dkr|dStd|dtd|d) zRetrieve the prompt from the given inputs. Args: inputs (Mapping[str, Any]): The inputs dictionary. Returns: str: The prompt. Raises: ValueError: If the prompt is not found or if multiple prompts are present. promptpromptsr\rz$Multiple prompts in run with inputs z!. Please create example manually.z)Could not find prompt in run with inputs r)rrl)rrs r"get_prompt_from_inputsrs6hF# w<<1  1:  /6 / / /    JJJJ K KKr!cd|vrtd|d|d}t|dkrtd||d}d|vrtd||dS) z(Get the LLM generation from the outputs.rrrr\zMultiple generations in run: rr}zNo text in generation: )rlrrs r"get_llm_generation_from_outputsrsG##RRRRSSS-(K ;1FFFGGG"1~ %%%E3CEEFFF F ##r! List[str]cR tjgdtjtjddgS#tjtf$r] tjddgtjtjdgcYS#tjtf$rt dwxYwwxYw)z7Get the correct docker compose command for this system.)dockercompose --version)stdoutstderrrrzdocker-composerzNeither 'docker compose' nor 'docker-compose' commands are available. Please install the Docker server following the instructions for your operating system at https://docs.docker.com/engine/install/) subprocess check_callDEVNULLCalledProcessErrorFileNotFoundErrorrlr r!r"get_docker_compose_commandr,s . . .%%    )$$  )+< =   !!;/!)!)     %% % % %-/@A   E  s!03B& /A<9B&<&B""B&ls_schemas.BaseMessageLikedictc|jd|jid}|jr*t|jdkri|j|dd<|S)z*Convert a LangChain message to an example.contentrrradditional_kwargs)rrrr)r converteds r"convert_langchain_messagerGs` GO,!!I  OS)B%C%Ca%G%G1NG4M1N &-. r!objobjectc ttt|ddttt|ddtt |do"tt|dtgS)zCheck if the given object is similar to BaseMessage. Args: obj (object): The object to check. Returns: bool: True if the object is similar to BaseMessage, False otherwise. rNrr)allrgetattrr|rhasattr)rs r"is_base_message_likerSst  wsIt44c : : ws$7>> E E C JZV0D0Dc%J%J   r!d) LANGSMITH LANGCHAIN) namespacesnamerE Optional[str]rrcxfd|D}|D]'tj}||cS(|S)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. cg|] }|d S)_r )r^ namespacers r"rczget_env_var..ys' ? ? ?y " "D " " ? ? ?r!)osenvironr])rrErnamesrs` r"rHrHesZ( @ ? ? ?J ? ? ?E t$$  LLL  Nr!Tc tjdtdtd|rdndS)z,Get the project name for a LangSmith tracer.HOSTED_LANGSERVE_PROJECT_NAMEPROJECTSESSIONrENrD)rrr]rH)return_default_values r"get_tracer_projectrsY :>> (  0D#N99$      r!c.eZdZdZd d fd Zd d ZxZS)FilterPoolFullWarningzFFilter urrllib3 warnings logged when the connection pool isn't reused.rCrr|hostr;rwcXt|||_dS)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)selfrr __class__s r"rzFilterPoolFullWarning.__init__s(  r!rLcH|}d|vrdS|j|vS)zJurllib3.connectionpool:Connection pool is full, discarding connection: ...z.Connection pool is full, discarding connectionT) getMessagerrrecordmsgs r"filterzFilterPoolFullWarning.filters0!! ;3 F F4z$$r!)rCrC)rr|rr|r;rwr;rL)rrrrrr __classcell__rs@r"rrs\PP%%%%%%%%r!rceZdZdZddZdS)FilterLangSmithRetryz!Filter for retries from this lib.r;rLc2|}d|vS)z!Filter retries from this library.LangSmithRetry)rrs r"rzFilterLangSmithRetry.filters !!s**r!Nr)rrrrrr r!r"rrs.++++++++r!rceZdZdZdS)rz&Wrapper to filter logs with this name.Nrr r!r"rrr4r!rloggerlogging.LoggerfiltersSequence[logging.Filter]Generator[None, None, None]c#(Kt5|D]}|| dddn #1swxYwY dVt5|D]B} ||#t$rtdY?wxYw ddddS#1swxYwYdS#t5|D]B} ||#t$rtdY?wxYw dddw#1swxYwYwxYw)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 BaseExceptionrwarning)rrrs r" filter_logsr s1 %% % %F   V $ $ $ $ %%%%%%%%%%%%%%%%?   ? ?! ? ??''////$???OO$=>>>>>? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ?\ ? ?! ? ??''////$???OO$=>>>>>? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ?s266B* BA%$B%$B  B B  BB!$B!*D2D8C D$C5 2D4C5 5D9 DD D D Dcachec,||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_CACHErDrP)r s r" get_cache_dirrs"  |T 2 2 22r!pathUnion[str, pathlib.Path] ignore_hostsOptional[Sequence[str]]c#K ddl}n#t$rtdwxYwddlm}|dfd }t j|\}}|| d s| d rd nd |d gdddg|}| |5dVddddS#1swxYwYdS)Use a cache for requests.rNzNvcrpy is required to use caching. Install with:pip install -U "langsmith[vcr]")_patchrequestrr;cVrtfdDrdSi_S)Nc3LK|]}j|VdSN)url startswith)r^rrs r"r`z>with_cache.._filter_request_headers..s3VV 6 6t < <VVVVVVr!)anyheaders)rrs`r"_filter_request_headersz+with_cache.._filter_request_headerssA  CVVVVVVVVV 4r!z.yamlz.ymlyamljson new_episodes)urimethodrbody authorizationz Set-Cookie) serializercassette_library_dir record_modematch_onfilter_headersbefore_record_request)rrr;r) vcr ImportErrorlangsmith._internalr patch_urllib3rrsplitVCRendswith use_cassette)rrr.r1r  cache_dir cache_filels_vcrs ` r" with_cacher9s       .    <;;;;;!!! GMM$//Iz WW""7++ /9/B/B6/J/J FF&#222'65  F   Z ( ( s $ CC"C"Optional[Union[str, pathlib.Path]]c#tK|/t||5dVddddS#1swxYwYdSdVdS)rN)r9)rrs r"with_optional_cacher<s   l + +   EEE                   s '++ctjtj}d|D}d|S)Ncg|]}d|v| S)z langsmith/r )r^lines r"rcz_format_exc..s"LLLt>:HLLxLLLN 77> " ""r!TvalmemoDict[int, Any] max_depth_depthct|}t|dd}| |S#t$rYnwxYwkr|St|tr"fd|DSt|t rfd|DSt|trtfd|DSt|trfd|DS|S)N __deepcopy__c hi|].\}}t|dzt|dz/Sre _middle_copy)r^kvrMrLrJs r" z _middle_copy..4sZ   1 D)VaZ 8 8,4FQJ;;   r!c :g|]}t|dzSrerQr^itemrMrLrJs r"rcz _middle_copy..;+PPPD T4FQJ??PPPr!c3BK|]}t|dzVdS)r\NrQrWs r"r`z_middle_copy..=s5UU\$i!DDUUUUUUr!c :h|]}t|dzSrerQrWs r" z_middle_copy..?rYr!) rrr rritemslisttupleset)rIrJrLrMclscopiers ``` r"rRrR&sg s))C S.$ / /F  6$<<     D   #t            #tQPPPPPPCPPPP#uVUUUUUUQTUUUUUU#sQPPPPPPCPPPP Js 2 ??ci} tj||S#t$rB}tdt |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)copydeepcopyr rdebugreprrR)rIrJr~s r" deepish_copyrhDs{D'}S$''' '''  4d1gg>>>C&&&&&&&& 's A%7A A% A%current_versiontarget_versioncnddlm}||}||}||kS)zGCheck if the current version is greater or equal to the target version.r)version) packagingrlr)rirjrlcurrenttargets r"is_version_greater_or_equalrpYsA!!!!!!mmO,,G ]]> * *F f r! identifierTuple[str, str, str]c|rC|ddks*|ds|drtd||dd}|d}t |dkr|dnd}d|vr4|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. /r\zInvalid identifier format: :rlatest-)rgrr4rlr2r)rqparts owner_namecommitownerrs r"parse_prompt_identifierr|bs* E   C 1 $ $   % % %   s # # %CzCCDDD   S! $ $EqJUaU1XXXF j &&sA.. t ID IG:GGHH HdF"" IG:GGHH HJ&&r!Pc8eZdZdZdfd Zd d d dfdZxZS)ContextThreadPoolExecutorz?ThreadPoolExecutor that copies the context to the child thread.rUCallable[P, T]rWP.argsrXP.kwargsr; Future[T]c tttdtft jtjj |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. .) rsubmitrrrGrqpartial contextvars copy_contextrun)rrUrWrXrs r"rz ContextThreadPoolExecutor.submitss ww~~ a !,..2D;?CI     r!Nr\timeout chunksizefnCallable[..., T] iterables Iterable[Any]rOptional[float]rr Iterator[T]cdtt|dDd fd }tj|g|R||dS) 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. c4g|]}tjSr )rr)r^rs r"rcz1ContextThreadPoolExecutor.map..s!QQQ1K,..QQQr!rrWrr;rGcDjg|RSr)popr)rWcontextsrs r" _wrapped_fnz2ContextThreadPoolExecutor.map.._wrapped_fns&%8<<>>%b04000 0r!r)rWrr;rG)rangerrmap)rrrrrrrrs ` @r"rzContextThreadPoolExecutor.maps<RQc)A,>O>O8P8PQQQ 1 1 1 1 1 1 1uww{         r!)rUrrWrrXrr;r) rrrrrrrrr;r)rrrrrrrrs@r"rrssII      :$( ( ( ( ( ( ( ( ( ( ( ( ( r!rapi_urlc.|p#tttdd}|st d|dddS)zBGet the LangSmith API URL from the environment or the given value.ENDPOINTzhttps://api.smith.langchain.comrDz!LangSmith API URL cannot be empty"'rt)rr|rHstripr(rstrip)r_api_urls r" get_api_urlrs$  5   H >>  F !DEEE >>   ! !# & & , ,S 1 1 8 8 = ==r!api_keyc||ntdd}||sdS|ddS)z8Get the API key from the environment or the given value.NAPI_KEYrDrr)rHr)rapi_key_s r" get_api_keyrsd!-ww;yRV3W3W3WHx~~//t >>   ! !# & & , ,S 1 11r!rc tj|jdd}t j|}|dkp)|dp|dS#tj$rYdSwxYw)zCheck if the URL is localhost. Parameters ---------- url : str The URL to check. Returns: ------- bool True if the URL is localhost, False otherwise. rurz 127.0.0.1z0.0.0.0z::F) urllib_parseurlsplitnetlocr2socket gethostbynamergaierror)rrips r" _is_localhostrs&s++288==a@  !& ) )[ SBMM)$<$<S d@S@SS ?uusA5A88B  B web_urlc|r|Stj|}t|rd}nt|jdrWt|jddd}tj||}nVt|j  drd}n,t|j  drd }nd }|S) z1Get the host URL based on the web URL or API URL.zhttp://localhostz/apir\r)rzeu.zhttps://eu.smith.langchain.comzdev.zhttps://dev.smith.langchain.comzhttps://smith.langchain.com) rurlparserr|rr4rsplit urlunparse_replacerr)rr parsed_urllinknew_paths r" get_host_urlrs&w//JW -! Z_   & &v . .-z''..vq99!<&z':':':'I'IJJ Z    * *5 1 1-/ Z    * *6 2 2-0, Kr!rdepthc|dkst|st|St|dr|jSt |t jrt|j|dzSt|drIt|dr!t|j dr |j jSt|j |dzSt|S)Nrrr\__call__r) callabler|rrrrqr_get_function_namerUrr)rrs r"rrs qyy y2wwr:{"i'((6!"'519555r:: 2{ # # ) j(I(I )<( (!"+uqy999 r77Nr!r)r9r:r;r<r)rRrSr;r)rurvr;rw)rrr;r|)rrrr|r;rw)rrr;r|)rrr;r)rrr;r)rrr;r)rrr;r)rrr;r|)rrr;r|)r;r)rrr;r)rrr;rL)rr|rErrrr;r)T)r;r)rrrrr;r)r rr;r)rrrrr;r)rr:rrr;r)r;r|)rHr) rIrGrJrKrLrrMrr;rG)rIrGr;rG)rir|rjr|r;rL)rqr|r;rr)rrr;r|)rrr;r)rr|r;rL)rrrr|)r)rrrrr;r|)jr __future__r contextlibrrdrrqloggingrpathlibrrrB threadingr@concurrent.futuresrrtypingrrrr r r r r rrrrrrrurllibrrhttpxrztyping_extensionsr urllib3.utilr langsmithr ls_schemas getLoggerrr Exceptionrr$r&r(r*r,r.r0r3 UserWarningr6r8rKrQrtrr lru_cacherrrrrrrrrrrrHrFilterrrrRLockrcontextmanagerr rr9r<rFrGrRrhrpr|r}rrrrrrr r!r"rsC """"""    99999999")((((( ''''''++++++ ' H % %HHHHHYHHHDDDDDDDD55555n555LLLLLLLLBBBBBnBBB8888888800000^000'''''^'''11111~111#####{###'''''$4'''     &???? 4????Q     &&8888PPPP(9999:LLLL2 $ $ $ $Q 4    $S!!!"3 "!6Q &%%%%%GN%%%*+++++7>+++11111U111y   ????2 3 3 3 3 LP'''''T -1     #### GCLLEF<''''* ' ' ' 'FIcNND D D D D 2D D D N > > > >2222*Q &r!