a `g;e@sdZddlmZddlZddlZddlZddlZddlZddlZddl 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.e/e0Z1Gd d d e2Z3Gd d d e3Z4Gddde3Z5Gddde3Z6Gddde3Z7Gddde3Z8Gddde3Z9Gddde3Z:Gddde3Z;Gddde<Z=Gddde=Z>dd d!d"d#d$Z?d%d&d'd(Z@d)d*d+d,d-ZAd.d/d0d1d2ZBd3d4d5d6d7ZCejDd8d9d:d4d/d;dd4d?d@dAZFd>d>d?dBdCZGd>dDd?dEdFZHd>dGdHdIdJZId>dDdKdLdMZJd>d4dHdNdOZKd>d4dKdPdQZLejDd8d9dRd&dSdTZMdUdVd?dWdXZNdYd%dZd[d\ZOejDd]d9dd^d_d4d`dad`dbdcddZPejDd8d9dd`d&dfdgZQGdhdidiejRZSGdjdkdkejRZTGdldmdme+ZUeVZWejXdndodpdqdrdsZYd`d`dtdudvZZejXddwdxdpdydzd{Z[ejXdd|dxdpdyd}d~Z\d4d&ddZ]e dZ^dddd:d:ddddZ_dddddZ`d4d4d%dddZad4ddddZbe)dZcGdddeZdd`d4dddZed`d`dddZfd4d%dddZgejDdd9d`d4dddZhdd*d:d4dddZidd%dddZjdS)zGeneric utility functions.) annotationsN)FutureThreadPoolExecutor)AnyCallableDict GeneratorIterableIteratorListLiteralMappingOptionalSequenceTupleTypeVarUnioncast)parse) ParamSpec)Retry)schemasc@seZdZdZdS)LangSmithErrorz=An error occurred while communicating with the LangSmith API.N__name__ __module__ __qualname____doc__rr]/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langsmith/utils.pyr0src@seZdZdZdS)LangSmithAPIErrorz9Internal server error while communicating with LangSmith.Nrrrrrr 4sr c@seZdZdZdS)LangSmithRequestTimeoutz*Client took too long to send request body.Nrrrrrr!8sr!c@seZdZdZdS)LangSmithUserErrorzAUser error caused an exception when communicating with LangSmith.Nrrrrrr"<sr"c@seZdZdZdS)LangSmithRateLimitErrorz7You have exceeded the rate limit for the LangSmith API.Nrrrrrr#@sr#c@seZdZdZdS)LangSmithAuthErrorz-Couldn't authenticate with the LangSmith API.Nrrrrrr$Dsr$c@seZdZdZdS)LangSmithNotFoundErrorz%Couldn't find the requested resource.Nrrrrrr%Hsr%c@seZdZdZdS)LangSmithConflictErrorzThe resource already exists.Nrrrrrr&Lsr&c@seZdZdZdS)LangSmithConnectionErrorz&Couldn't connect to the LangSmith API.Nrrrrrr'Psr'c@seZdZdZdS)LangSmithWarningzBase class for warnings.Nrrrrrr(Wsr(c@seZdZdZdS)LangSmithMissingAPIKeyWarningzWarning for missing API key.Nrrrrrr)[sr)zOptional[dict]zUnion[bool, Literal['local']])ctxreturncCsTddlm}m}|p|}|ddur.|dS|r8dStdtddd d }|d kS) z"Return True if tracing is enabled.r)get_current_run_treeget_tracing_contextenabledNTZ TRACING_V2ZTRACINGdefaulttrue)Zlangsmith.run_helpersr,r- get_env_var)r*r,r-ZtcZ var_resultrrrtracing_is_enabled_s  r4boolr+cCstddddkS)z"Return True if testing is enabled.Z TEST_TRACKINGr/r0falser3rrrrtest_tracking_is_disabledrsr9zTuple[str, ...]r) arg_groupsr+csdddfdd }|S)z7Validate specified keyword args are mutually exclusive.r)funcr+cs&tddddfdd }|S)Nr)argskwargsr+cs\fddD}ddt|D}|rNfdd|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)r@Z arg_grouprBrr ~sz@xor_args..decorator..wrapper..cSsg|]\}}|dkr|qSr>r)r@icountrrrrFrDcsg|]}d|qS), )join)r@rHr:rrrFrDzFExactly one argument in each of the following groups must be defined: rJ) enumerate ValueErrorrK)r<r=countsZinvalid_groupsZinvalid_group_names)r:r;rBrwrapper{s z,xor_args..decorator..wrapper) functoolswraps)r;rPrL)r;r decoratorzszxor_args..decoratorr)r:rSrrLrxor_argswsrTz(Union[requests.Response, httpx.Response]None)responser+c CsLz |Wn:tjyF}z tt||j|WYd}~n d}~00dS)z&Raise an error with the response text.N)raise_for_statusrequests HTTPErrorstrtext)rVerrrraise_for_status_with_texts r]zUnion[enum.Enum, str]rZ)enur+cCst|tjr|jS|S)zGet the value of a string enum.) isinstanceenumEnumvalue)r^rrrget_enum_values rcr>)maxsizeint)levelmessager+cCst||dS)z4Log a message at the specified level, but only once.N)_LOGGERlog)rfrgrrrlog_oncesrjzMapping[str, Any])rgr+cCsh|s tdd|vrDd|vr,td|d|ddddSd |vr\td |d |d SdS) 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.)rNreplacelowerrgrrr_get_message_types  rvcCsX|s tdd|vr4d|vr,td|d|dSd|vrLtd|d|dSdS) Nrkrlr=rnz Message does not have kwargs.datarrz Message does not have data.)rNrurrr_get_message_fieldss  rxzDict[str, Any]cCst|}t|}||dS)z&Extract message from a message object.rqrw)rvrx)rgZ message_typeZ message_datarrr_convert_messagesrzzList[Dict[str, Any]])inputsr+cCsDd|vrdd|dDSd|vr0t|dgStd|ddS)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)rz)r@rgrrrrFrDz,get_messages_from_inputs..rgz-Could not find message(s) in run with inputs .N)rzrN)r{rrrget_messages_from_inputss r~)outputsr+cCsrd|vrtd|d|d}t|dkrFtdt|d|d|d}d|vrftd |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: r}r>z3Chat examples expect exactly one generation. Found z generations: rrgz"Unexpected format for generation: z%. Generation does not have a message.)rNlenrzrrZfirst_generationrrr#get_message_generation_from_outputss"   rcCsXd|vr|dSd|vrD|d}t|dkr4|dStd|dtd|dd S) 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}N)rrN)r{rrrrget_prompt_from_inputss   rcCs`d|vrtd|d|d}t|dkr:td||d}d|vrXtd||dS) z(Get the LLM generation from the outputs.rrr}r>zMultiple generations in run: rr[zNo text in generation: )rNrrrrrget_llm_generation_from_outputss rz List[str]c Csz"tjgdtjtjdddgWStjtfyz$tjddgtjtjddgWYStjtfyztdYn0Yn0dS) 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/N) subprocess check_callDEVNULLCalledProcessErrorFileNotFoundErrorrNrrrrget_docker_compose_command,s&  rzls_schemas.BaseMessageLikedictcCs<|jd|jid}|jr8t|jdkr8i|j|dd<|S)z*Convert a LangChain message to an example.contentryrrwadditional_kwargs)rqrrr)rgZ convertedrrrconvert_langchain_messageGs robject)objr+cCs@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. rNrrq)allr_getattrrZrhasattr)rrrris_base_message_likeSs rd)Z LANGSMITHZ LANGCHAIN) namespacesz Optional[str]r)namer1rr+cs<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)r@ namespacerrrrFyrDzget_env_var..N)osenvironr?)rr1rnamesrbrrrr3es   r3Tc Cs&tjdtdtd|rdndddS)z,Get the project name for a LangSmith tracer.ZHOSTED_LANGSERVE_PROJECT_NAMEZPROJECTZSESSIONr1Nr0)rrr?r3)Zreturn_default_valuerrrget_tracer_projects rcs:eZdZdZd ddddfdd Zdd d d ZZS) FilterPoolFullWarningzFFilter urrllib3 warnings logged when the connection pool isn't reused.r/rZrU)rhostr+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)super__init___host)selfrr __class__rrrs zFilterPoolFullWarning.__init__r5r6cCs|}d|vrdS|j|vS)zJurllib3.connectionpool:Connection pool is full, discarding connection: ...z.Connection pool is full, discarding connectionT) getMessagerrrecordmsgrrrfilterszFilterPoolFullWarning.filter)r/r/)rrrrrr __classcell__rrrrrs rc@seZdZdZddddZdS)FilterLangSmithRetryz!Filter for retries from this lib.r5r6cCs|}d|vS)z!Filter retries from this library.LangSmithRetry)rrrrrrszFilterLangSmithRetry.filterN)rrrrrrrrrrsrc@seZdZdZdS)rz&Wrapper to filter logs with this name.Nrrrrrrsrzlogging.LoggerzSequence[logging.Filter]zGenerator[None, None, None])loggerfiltersr+ccst$|D]}||q Wdn1s.0YzddVWtF|D]0}z||WqLtyztdYqL0qLWdq1s0Yn\tF|D]0}z||WqtytdYq0qWdn1s0Y0dS)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 BaseExceptionrhwarning)rrrrrr filter_logss" * 2 r)cacher+cCs|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_CACHEr0r8)rrrr get_cache_dirs rzUnion[str, pathlib.Path]zOptional[Sequence[str]])path ignore_hostsr+c#sz ddl}Wnty&tdYn0ddlm}|dddfdd }tj|\}}|j| d sx| d r|d nd |d gdddg|d}| |dVWdn1s0YdS)Use a cache for requests.rNzNvcrpy is required to use caching. Install with:pip install -U "langsmith[vcr]")_patchr)requestr+cs(rtfddDrdSi_S)Nc3s|]}j|VqdSN)url startswith)r@rrrrrCrDz>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) vcr ImportErrorZlangsmith._internalr patch_urllib3rrsplitZVCRendswithZ use_cassette)rrrrr cache_dirZ cache_fileZls_vcrrrr with_caches0     rz"Optional[Union[str, pathlib.Path]]ccsD|dur:t||dVWdq@1s.0YndVdS)rN)r)rrrrrwith_optional_caches &rcCs&tjt}dd|D}d|S)NcSsg|]}d|vr|qS)z langsmith/r)r@linerrrrFrDz_format_exc..r/) tracebackformat_exceptionsysexc_inforK)Ztb_linesZfiltered_linesrrr _format_excsrTzDict[int, Any])valmemo max_depth_depthr+cst|}t|dd}|dur:z |WSty8Yn0krF|St|trjfdd|DSt|trfdd|DSt|trtfdd|DSt|tr·fdd |DS|S) N __deepcopy__c s2i|]*\}}t|dt|dqSrG _middle_copy)r@kvrrrrr 4s z _middle_copy..csg|]}t|dqSrGrr@itemrrrrF;rDz _middle_copy..c3s |]}t|dVqdS)r>NrrrrrrC=rDz_middle_copy..csh|]}t|dqSrGrrrrr ?rDz_middle_copy..) rqrrr_ritemslisttupleset)rrrrclscopierrrrr&s(       r)rr+c CsVi}zt||WStyP}z&tdt|t||WYd}~Sd}~00dS)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)copydeepcopyrrhdebugreprr)rrr\rrr deepish_copyDs r)current_versiontarget_versionr+cCs(ddlm}||}||}||kS)zGCheck if the current version is greater or equal to the target version.r)version) packagingrr)rrrcurrenttargetrrris_version_greater_or_equalYs   rzTuple[str, str, str]) identifierr+cCs|r&|ddks&|ds&|dr4td||dd}|d}t|dkr\|dnd}d|vr|dd\}}|r|std||||fS|std|d||fSdS) 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: :rZlatest-N)rIrrrNrr)rpartsZ owner_namecommitownerrrrrparse_prompt_identifierbs(    r PcsNeZdZdZdddddfdd Zd d d d d ddddfddZZS)ContextThreadPoolExecutorz?ThreadPoolExecutor that copies the context to the child thread.zCallable[P, T]zP.argszP.kwargsz Future[T])r;r<r=r+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. .) rsubmitrrrrQpartial contextvars copy_contextrun)rr;r<r=rrrrs  z ContextThreadPoolExecutor.submitNr>timeout chunksizezCallable[..., T]z Iterable[Any]zOptional[float]rez Iterator[T])fn iterablesrrr+csLddtt|dDdddfdd }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)rr)r@rrrrrFrDz1ContextThreadPoolExecutor.map..rrr)r<r+csjg|RSr)popr)r<Zcontextsrrr _wrapped_fnsz2ContextThreadPoolExecutor.map.._wrapped_fnr)rangerrmap)rrrrrrrrrrszContextThreadPoolExecutor.map)rrrrrrrrrrrrs r)api_urlr+cCs@|ptttddd}|s&td|dddS)zBGet the LangSmith API URL from the environment or the given value.ZENDPOINTzhttps://api.smith.langchain.comr0z!LangSmith API URL cannot be empty"'r)rrZr3stripr"rstrip)rZ_api_urlrrr get_api_urlsr$)api_keyr+cCs@|dur |n tddd}|dus(|s,dS|ddS)z8Get the API key from the environment or the given value.NZAPI_KEYr0r r!)r3r")r%Zapi_key_rrr get_api_keysr&)rr+cCsZz>t|jdd}t|}|dkp<|dp<|dWStjyTYdS0dS)zCheck if the URL is localhost. Parameters ---------- url : str The URL to check. Returns: ------- bool True if the URL is localhost, False otherwise. rrz 127.0.0.1z0.0.0.0z::FN) urllib_parseurlsplitnetlocrsocket gethostbynamergaierror)rr)iprrr _is_localhosts  r.)web_urlrcCs|r|St|}t|r d}nt|jdrZt|jddd}t|j|d}nt|jdrt|jddd}t|j|d}nFt|j  drd}n0t|j  d rd }nt|j  d rd }nd }|S)z1Get the host URL based on the web URL or API URL.zhttp://localhostz/apir>r)rz/api/v1zeu.zhttps://eu.smith.langchain.comzdev.zhttps://dev.smith.langchain.comzbeta.z https://beta.smith.langchain.comzhttps://smith.langchain.com) r'urlparser.rZrrrsplit urlunparse_replacer)r)r0r parsed_urllinknew_pathrrr get_host_urls& r8)rdepthr+cCs|dkst|st|St|dr(|jSt|tjrDt|j|dSt|dr|t|drlt|j drl|j jSt|j |dSt|S)Nr/rr>__call__r) callablerZrrr_rQr_get_function_namer;rr:)rr9rrrr<s   r<rcCs(|dup&|dkp&|dkp&|dkp&|dkS)zCheck if the value is truish. Args: val (Any): The value to check. Returns: bool: True if the value is truish, False otherwise. Tr2TrueTRUE1r)rrrr is_truish%s r@)N)N)T)N)N)rr)r)kr __future__r contextlibrrr`rQloggingrpathlibr*rr threadingrconcurrent.futuresrrtypingrrrrr r r r r rrrrrrurllibrr'ZhttpxrXZtyping_extensionsrZ urllib3.utilrZ langsmithrZ ls_schemas getLoggerrrh Exceptionrr r!r"r#r$r%r&r' UserWarningr(r)r4r9rTr]rc lru_cacherjrvrxrzr~rrrrrrr3rFilterrrrRLockrcontextmanagerrrrrrrrrrr rrr$r&r.r8r<r@rrrrs D            *  #G