o Zheg@sdZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlmZmZmZmZmZddlmZmZddlmZmZmZmZmZmZm Z ddl!m"Z#ddl$Z$ddl%Z%ddl&m'Z'dd l(m)Z)dd l*m+Z,e-e.Z/Gd d d e0Z1Gd dde1Z2Gddde1Z3Gddde1Z4Gddde1Z5Gddde1Z6Gddde1Z7Gddde1Z8Gddde1Z9Gddde1Z:Gdd d e;Zdd*d+Z?dd/d0Z@dd4d5ZAdd9d:ZBejCd;d<dd@dAZDddCdDZEddEdFZFddHdIZGddLdMZHddOdPZIddQdRZJddSdTZKejCd;d<ddVdWZLddZd[ZMdd^d_ZNejCd`d< ddadbddhdiZOejCd;d<dddkdlZPGdmdndnejQZRGdodpdpejQZSGdqdrdre)ZTeUZVejWddxdyZXdd{d|ZYejW ddddZZejW ddddZ[dddZ\edZ] ddddZ^dddZ_dddZ`dddZae'dZbGdddeZcdddZddddZedddZfejCdd<dddZgddddZhdddZidS)zGeneric utility functions.) annotationsN) GeneratorIterableIteratorMappingSequence)FutureThreadPoolExecutor)AnyCallableLiteralOptionalTypeVarUnioncast)parse) ParamSpec)Retry)schemasc@eZdZdZdS)LangSmithErrorz=An error occurred while communicating with the LangSmith API.N__name__ __module__ __qualname____doc__rrF/var/www/html/lang_env/lib/python3.10/site-packages/langsmith/utils.pyr)rc@r)LangSmithAPIErrorz9Internal server error while communicating with LangSmith.Nrrrrrr-rrc@r)LangSmithRequestTimeoutz*Client took too long to send request body.Nrrrrrr 1rr c@r)LangSmithUserErrorzAUser error caused an exception when communicating with LangSmith.Nrrrrrr!5rr!c@r)LangSmithRateLimitErrorz7You have exceeded the rate limit for the LangSmith API.Nrrrrrr"9rr"c@r)LangSmithAuthErrorz-Couldn't authenticate with the LangSmith API.Nrrrrrr#=rr#c@r)LangSmithNotFoundErrorz%Couldn't find the requested resource.Nrrrrrr$Arr$c@r)LangSmithConflictErrorzThe resource already exists.Nrrrrrr%Err%c@r)LangSmithConnectionErrorz&Couldn't connect to the LangSmith API.Nrrrrrr&Irr&cs"eZdZdZd fd d ZZS) LangSmithExceptionGroupz%Port of ExceptionGroup for Py < 3.11.argsr exceptionsSequence[Exception]kwargsreturnNonecstj|i|||_dS)z Initialize.N)super__init__r))selfr)r(r+ __class__rrr/Ps z LangSmithExceptionGroup.__init__)r(r r)r*r+r r,r-)rrrrr/ __classcell__rrr1rr'Msr'c@r)LangSmithWarningzBase class for warnings.Nrrrrrr4[rr4c@r)LangSmithMissingAPIKeyWarningzWarning for missing API key.Nrrrrrr5_rr5ctxOptional[dict]r,Union[bool, Literal['local']]cCsTddlm}m}|p |}|ddur|dS|rdStdtddd d }|d kS) z"Return True if tracing is enabled.r)get_current_run_treeget_tracing_contextenabledNTZ TRACING_V2ZTRACINGdefaulttrue)Zlangsmith.run_helpersr9r: get_env_var)r6r9r:ZtcZ var_resultrrrtracing_is_enabledcs  rAboolcCstddddkS)z"Return True if testing is enabled.Z TEST_TRACKINGr<r=falser@rrrrtest_tracking_is_disabledvsrE arg_groupstuple[str, ...]r csdfdd }|S)z7Validate specified keyword args are mutually exclusive.funcr r,cstdfdd }|S)Nr(r r+r,cs\fddD}ddt|D}|r'fdd|D}tdd||iS)z3Validate exactly one arg in each group is not None.cs"g|] }tfdd|DqS)c3s"|] }|durdVqdS)N)get).0argr+rr  zJxor_args..decorator..wrapper...)sum)rKZ arg_grouprMrr sz@xor_args..decorator..wrapper..cSsg|] \}}|dkr|qSrIr)rKicountrrrrQscsg|] }d|qS), )join)rKrSrFrrrQzFExactly one argument in each of the following groups must be defined: rU) enumerate ValueErrorrV)r(r+countsZinvalid_groupsZinvalid_group_names)rFrHrMrwrappers z,xor_args..decorator..wrapper)r(r r+r r,r ) functoolswraps)rHr\rW)rHr decorator~szxor_args..decoratorN)rHr r,r r)rFr_rrWrxor_args{sr`response(Union[requests.Response, httpx.Response]r-c Cszz|WdStjy}z tt||j|d}~wtjy<}ztjt|d|j|j|d|d}~ww)z&Raise an error with the response text.Nz: )requestra)raise_for_statusrequests HTTPErrorstrtexthttpxZHTTPStatusErrorrc)raerrrraise_for_status_with_textsrkenuUnion[enum.Enum, str]rgcCst|tjr |jS|S)zGet the value of a string enum.) isinstanceenumEnumvalue)rlrrrget_enum_values rrrI)maxsizelevelintmessagecCst||dS)z4Log a message at the specified level, but only once.N)_LOGGERlog)rtrvrrrlog_oncesryMapping[str, Any]cCsd|stdd|vr"d|vrtd|d|ddddSd |vr.td |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.)rZreplacelowerrvrrr_get_message_types  rcCsT|stdd|vrd|vrtd|d|dSd|vr&td|d|dS) Nr{r|r+r~z Message does not have kwargs.datarz Message does not have data.)rZrrrr_get_message_fieldss  rdict[str, Any]cCst|}t|}||dS)z&Extract message from a message object.rr)rr)rvZ message_typeZ message_datarrr_convert_messages rinputslist[dict[str, Any]]cCs@d|vr dd|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. messagescSsg|]}t|qSr)r)rKrvrrrrQz,get_messages_from_inputs..rvz-Could not find message(s) in run with inputs .)rrZ)rrrrget_messages_from_inputss  routputscCsrd|vr td|d|d}t|dkr#tdt|d|d|d}d|vr3td |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: rrIz3Chat examples expect exactly one generation. Found z generations: rrvz"Unexpected format for generation: z%. Generation does not have a message.)rZlenrrrZfirst_generationrrr#get_message_generation_from_outputss"    rcCsTd|vr|dSd|vr"|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. promptpromptsrIrz$Multiple prompts in run with inputs z!. Please create example manually.z)Could not find prompt in run with inputs r)rrZ)rrrrrget_prompt_from_inputss   rcCs`d|vr td|d|d}t|dkrtd||d}d|vr,td||dS) z(Get the LLM generation from the outputs.rrrrIzMultiple generations in run: rrhzNo text in generation: )rZrrrrrget_llm_generation_from_outputs)s r list[str]c CszztjgdtjtjdddgWStjtfy<ztjddgtjtjddgWYStjtfy;tdww)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_callDEVNULLCalledProcessErrorFileNotFoundErrorrZrrrrget_docker_compose_command6s*  rls_schemas.BaseMessageLikedictcCs<|jd|jid}|jrt|jdkri|j|dd<|S)z*Convert a LangChain message to an example.contentrrradditional_kwargs)rrrr)rvZ convertedrrrconvert_langchain_messageQs robjobjectcCs@ttt|ddttt|ddtt|dott|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)allrngetattrrgrhasattr)rrrris_base_message_like]s  rd)Z LANGSMITHZ LANGCHAIN) namespacesnamer> Optional[str]rtuplecs<fdd|D}|D]tj}|dur|Sq |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. csg|] }|dqS)_r)rK namespacerrrrQrXzget_env_var..N)osenvironrJ)rr>rnamesrqrrrr@os r@Tc Cs&tjdtdtd|r dndddS)z,Get the project name for a LangSmith tracer.ZHOSTED_LANGSERVE_PROJECT_NAMEZPROJECTZSESSIONr>Nr=)rrrJr@)Zreturn_default_valuerrrget_tracer_projects rcs.eZdZdZd dfdd Zdd d ZZS)FilterPoolFullWarningzFFilter urrllib3 warnings logged when the connection pool isn't reused.r<rrghostr,r-cst|||_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)r.r/_host)r0rrr1rrr/s  zFilterPoolFullWarning.__init__rBcCs|}d|vr dS|j|vS)zJurllib3.connectionpool:Connection pool is full, discarding connection: ...z.Connection pool is full, discarding connectionT) getMessagerr0recordmsgrrrfilters zFilterPoolFullWarning.filter)r<r<)rrgrrgr,r-r,rB)rrrrr/rr3rrr1rrs rc@seZdZdZdddZdS)FilterLangSmithRetryz!Filter for retries from this lib.r,rBcCs|}d|vS)z!Filter retries from this library.LangSmithRetry)rrrrrrszFilterLangSmithRetry.filterNr)rrrrrrrrrrsrc@r)rz&Wrapper to filter logs with this name.Nrrrrrrrrloggerlogging.LoggerfiltersSequence[logging.Filter]Generator[None, None, None]ccst|D]}||qWdn1swYz3dVWt#|D]}z||Wq'ty>tdYq'wWddS1sJwYdSt"|D]}z||WqVtymtdYqVwWdw1sxwYw)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 BaseExceptionrwwarning)rrrrrr filter_logss2   ( rcachecCs|dur|StdddS)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. NZ TEST_CACHEr=rD)rrrr get_cache_dirs  rpathUnion[str, pathlib.Path] ignore_hostsOptional[Sequence[str]]c#szddl}Wn tytdwddlm}|dfdd }tj|\}}|j| d s8| d r:d nd |dgdddg|d}| | dVWddS1s[wYdS)Use a cache for requests.rNzNvcrpy is required to use caching. Install with:pip install -U "langsmith[vcr]")_patchrcr r,cs(rtfddDrdSi_S)Nc3s|] }j|VqdSN)url startswith)rKrrcrrrNsz>with_cache.._filter_request_headers..)anyheadersrrrr_filter_request_headerssz+with_cache.._filter_request_headersz.yamlz.ymlyamljsonZ new_episodes)urimethodrbody authorizationz Set-Cookie) serializerZcassette_library_dirZ record_modeZmatch_onZfilter_headersZbefore_record_request)rcr r,r ) vcr ImportErrorZlangsmith._internalr patch_urllib3rrsplitZVCRendswithZ use_cassette)rrrrr cache_dirZ cache_fileZls_vcrrrr with_caches6    "r"Optional[Union[str, pathlib.Path]]ccsJ|dur t|| dVWddS1swYdSdVdS)rN)r)rrrrrwith_optional_caches  " rcCs&tjt}dd|D}d|S)NcSsg|]}d|vr|qS)z langsmith/r)rKlinerrrrQ)sz_format_exc..r<) tracebackformat_exceptionsysexc_inforV)Ztb_linesZfiltered_linesrrr _format_exc&s rTvalmemodict[int, Any] max_depth_depthcst|}t|dd}|durz|WStyYnwkr#|St|tr5fdd|DSt|trEfdd|DSt|trWtfdd|DSt|trgfdd |DS|S) N __deepcopy__c s2i|]\}}t|dt|dqSrR _middle_copy)rKkvrrrrr >s  z _middle_copy..csg|] }t|dqSrRrrKitemrrrrQEz _middle_copy..c3s"|] }t|dVqdS)rINrr rrrrNGrOz_middle_copy..csh|] }t|dqSrRrr rrr Ir z_middle_copy..) rrrrnritemslistrset)rrrrclscopierrrrr0s*       rc CsRi}zt||WSty(}ztdt|t||WYd}~Sd}~ww)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)copydeepcopyrrwdebugreprr)rrrjrrr deepish_copyNs rcurrent_versiontarget_versioncCs(ddlm}||}||}||kS)zGCheck if the current version is greater or equal to the target version.r)version) packagingrr)rrrcurrenttargetrrris_version_greater_or_equalcs   r identifiertuple[str, str, str]cCs|r|ddks|ds|drtd||dd}|d}t|dkr.|dnd}d|vrL|dd\}}|r@|sGtd||||fS|sUtd|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. /rIzInvalid identifier format: :rZlatest-)rTrrrZrr)rpartsZ owner_namecommitownerrrrrparse_prompt_identifierls&    r&Pcs8eZdZdZdfd d Zd d ddfddZZS)ContextThreadPoolExecutorz?ThreadPoolExecutor that copies the context to the child thread.rHCallable[P, T]r(P.argsr+P.kwargsr, Future[T]cs6tttdtftjtj |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. .) r.submitrr rr]partial contextvars copy_contextrun)r0rHr(r+r1rrr-s  z ContextThreadPoolExecutor.submitNrItimeout chunksizefnCallable[..., T] iterables Iterable[Any]r3Optional[float]r4ru Iterator[T]csFddtt|dDd fdd }tj|g|R||d S) 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. cSsg|]}tqSr)r/r0)rKrrrrrQrz1ContextThreadPoolExecutor.map..rr(r r,rcsjg|RSr)popr1)r(Zcontextsr5rr _wrapped_fnsz2ContextThreadPoolExecutor.map.._wrapped_fnr2N)r(r r,r)rangerr.map)r0r5r3r4r7r=r1r<rr?szContextThreadPoolExecutor.map)rHr)r(r*r+r+r,r,) r5r6r7r8r3r9r4rur,r:)rrrrr-r?r3rrr1rr(s r(api_urlcCs@|p tttddd}|std|dddS)zBGet the LangSmith API URL from the environment or the given value.ZENDPOINTzhttps://api.smith.langchain.comr=z!LangSmith API URL cannot be empty"'r )rrgr@stripr!rstrip)r@Z_api_urlrrr get_api_urlsrEapi_keycCs@|dur|ntddd}|dus|sdS|ddS)z8Get the API key from the environment or the given value.NZAPI_KEYr=rArB)r@rC)rFZapi_key_rrr get_api_keysrGrcCsVzt|jdd}t|}|dkp|dp|dWStjy*YdSw)zCheck if the URL is localhost. Parameters ---------- url : str The URL to check. Returns: ------- bool True if the URL is localhost, False otherwise. r!rz 127.0.0.1z0.0.0.0z::F) urllib_parseurlsplitnetlocrsocket gethostbynamergaierror)rrJiprrr _is_localhosts  rOweb_urlcCs|r|St|}t|rd}|St|jdr/t|jddd}t|j|d}|St|jdrMt|jddd}t|j|d}|St|j  drYd}|St|j  d red }|St|j  d rqd }|Sd }|S)z1Get the host URL based on the web URL or API URL.zhttp://localhostz/apirIr)rz/api/v1zeu.zhttps://eu.smith.langchain.comzdev.zhttps://dev.smith.langchain.comzbeta.z https://beta.smith.langchain.comzhttps://smith.langchain.com) rHurlparserOrgrrrsplit urlunparse_replacerJr)rQr@ parsed_urllinknew_pathrrr get_host_urls2   rYr5depthcCs|dkst|s t|St|dr|jSt|tjr"t|j|dSt|dr>t|dr6t|j dr6|j jSt|j |dSt|S)NrPrrI__call__r2) callablergrrrnr]r._get_function_namerHr2r[)r5rZrrrr]s   r]r cCs&t|tr|dkp|dkSt|S)zCheck if the value is truish. Args: val (Any): The value to check. Returns: bool: True if the value is truish, False otherwise. r?1)rnrgrrB)rrrr is_truish/s r_r)r6r7r,r8r)rFrGr,r )rarbr,r-)rlrmr,rg)rtrurvrgr,r-)rvrzr,rg)rvrzr,rz)rvrzr,r)rrzr,r)rrzr,r)rrzr,rg)rrzr,rg)r,r)rvrr,r)rrr,rB)rrgr>rrrr,r)T)r,r)rrrrr,r)rrr,r)rrrrr,r)rrrrr,r)r,rg)rr) rrrrrrurrur,r)rrr,r)rrgrrgr,rB)rrgr,r)r@rr,rg)rFrr,r)rrgr,rB)rQrr@rg)r)r5r rZrur,rg)rr r,rB)jr __future__r contextlibr/rror]loggingrpathlibrKrr threadingrcollections.abcrrrrrconcurrent.futuresrr typingr r r r rrrurllibrrHriretyping_extensionsrZ urllib3.utilrZ langsmithrZ ls_schemas getLoggerrrw Exceptionrrr r!r"r#r$r%r&r' UserWarningr4r5rArEr`rkrr lru_cacheryrrrrrrrrrrr@rFilterrrrRLockrcontextmanagerrrrrrrrrrr&r'r(rErGrOrYr]r_rrrrs $                          *    # G