o !ir@sJdZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlmZmZmZmZmZddlmZmZddlmZmZmZmZmZmZm Z m!Z!ddl"m#Z$ddl%Z%ddl&Z&ddl'm(Z(dd l)m*Z*dd l+m,Z-e.e/Z0Gd d d e1Z2Gd dde2Z3Gddde2Z4Gddde2Z5Gddde2Z6Gddde2Z7Gddde2Z8Gddde2Z9Gddde2Z:Gddde2Z;dd"d#ZGd&d'd'e>Z?ddd+d,Z@dd.d/ZAdd3d4ZBdd8d9ZCddd?ddCdDZFddFdGZGddHdIZHddKdLZIddOdPZJddRdSZKddTdUZLddVdWZMejEd>d?ddYdZZNdd]d^ZOddadbZPddedfZQe!dgdhddmdnZRe! ddgdhddodnZRejEdpd? ddgdhddqdnZRejEd>d?dddsdtZSGdudvdvejTZUGdwdxdxejTZVGdydzdze*ZWeXZYejZdddZ[dddZ\ddddddZ]ejZ  ddddZ^ejZ  ddddZ_dddZ`edZa ddddZbdddZcdddZddddZee(dZfGdddeZgdddZhdddZidddZjdddZkejEdd?dddZlddddZmdddZndS)zGeneric utility functions.) annotationsN) GeneratorIterableIteratorMappingSequence)FutureThreadPoolExecutor)AnyCallableLiteralOptionalTypeVarUnioncastoverload)parse) ParamSpec)Retry)schemasc@eZdZdZdS)LangSmithErrorz=An error occurred while communicating with the LangSmith API.N__name__ __module__ __qualname____doc__rrT/var/www/html/crm_dreinet/venv_linux/lib/python3.10/site-packages/langsmith/utils.pyr*rc@r)LangSmithAPIErrorz9Internal server error while communicating with LangSmith.Nrrrrrr .rr c@r)LangSmithRequestTimeoutz*Client took too long to send request body.Nrrrrrr!2rr!c@r)LangSmithUserErrorzAUser error caused an exception when communicating with LangSmith.Nrrrrrr"6rr"c@r)LangSmithRateLimitErrorz7You have exceeded the rate limit for the LangSmith API.Nrrrrrr#:rr#c@r)LangSmithAuthErrorz-Couldn't authenticate with the LangSmith API.Nrrrrrr$>rr$c@r)LangSmithNotFoundErrorz%Couldn't find the requested resource.Nrrrrrr%Brr%c@r)LangSmithConflictErrorzThe resource already exists.Nrrrrrr&Frr&c@r)LangSmithConnectionErrorz&Couldn't connect to the LangSmith API.Nrrrrrr'Jrr'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__rrr0Qs z LangSmithExceptionGroup.__init__)r)r r*r+r,r r-r.)rrrrr0 __classcell__rrr2rr(Nsr( identifierstrr-cCs d|dS)a%Get the error message for an invalid prompt identifier. Used consistently across the codebase when parsing prompt identifiers fails. Args: identifier: The invalid identifier that was provided. Returns: A formatted error message explaining the valid formats. z#Invalid prompt identifier format: "z". Expected one of: - "prompt-name" (for private prompts) - "owner/prompt-name" (for prompts with explicit owner) - "prompt-name:commit-hash" (with commit reference) - "owner/prompt-name:commit-hash" (with owner and commit)r)r5rrr!get_invalid_prompt_identifier_msgYs r7c@r)LangSmithWarningzBase class for warnings.Nrrrrrr8qrr8c@r)LangSmithMissingAPIKeyWarningzWarning for missing API key.Nrrrrrr9urr9ctxOptional[dict]Union[bool, Literal['local']]cCsvddlmm}ddlm}m}|p|}|ddur |dS|r%dS|jdur-|jStdtddd d }|d kS) z"Return True if tracing is enabled.rN)get_current_run_treeget_tracing_contextenabledT TRACING_V2TRACINGdefaulttrue)langsmith._internal._context _internal_contextlangsmith.run_helpersr=r>_GLOBAL_TRACING_ENABLED get_env_var)r:rHr=r>tc var_resultrrrtracing_is_enabledys   rNboolcCstddddkS)z"Return True if testing is enabled. TEST_TRACKINGrBrCfalserKrrrrtest_tracking_is_disabledsrS 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)rY arg_groupr[rr sz@xor_args..decorator..wrapper..cSsg|] \}}|dkr|qSrWr)rYicountrrrr`scsg|] }d|qS), )join)rYrbrTrrr`zFExactly one argument in each of the following groups must be defined: rd) enumerate ValueErrorre)r)r,countsinvalid_groupsinvalid_group_names)rTrVr[rwrappers z,xor_args..decorator..wrapper)r)r r,r r-r ) functoolswraps)rVrmrf)rVr decoratorszxor_args..decoratorN)rVr r-r r)rTrprrfrxor_argssrqresponse(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: )requestrr)raise_for_statusrequests HTTPErrorr6texthttpxHTTPStatusErrorrt)rrerrrraise_for_status_with_textsr|enuUnion[enum.Enum, str]cCst|tjr |jS|S)zGet the value of a string enum.) isinstanceenumEnumvalue)r}rrrget_enum_values rrW)maxsizelevelintmessagecCst||dS)z4Log a message at the specified level, but only once.N)_LOGGERlog)rrrrrlog_oncesrMapping[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.MessagerBtype&Unexpected format for stored message: z Message does not have a type.)rireplacelowerrrrr_get_message_types  rcCsT|stdd|vrd|vrtd|d|dSd|vr&td|d|dS) Nrrr,rz Message does not have kwargs.datarz Message does not have data.)rirrrr_get_message_fieldss  rdict[str, Any]cCst|}t|}||dS)z&Extract message from a message object.rr)rr)r message_type message_datarrr_convert_messages rinputslist[dict[str, Any]]cCs@d|vr dd|dDSd|vrt|dgStd|d)aExtract messages from the given inputs dictionary. Args: inputs: The inputs dictionary. Returns: A list of dictionaries representing the extracted messages. Raises: ValueError: If no message(s) are found in the inputs dictionary. messagescSsg|]}t|qSr)r)rYrrrrr`z,get_messages_from_inputs..rz-Could not find message(s) in run with inputs .)rri)rrrrget_messages_from_inputss  routputscCsrd|vr td|d|d}t|dkr#tdt|d|d|d}d|vr3td |d t|dS) aRetrieve the message generation from the given outputs. Args: outputs: The outputs dictionary. Returns: 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: rrWz3Chat examples expect exactly one generation. Found z generations: rrz"Unexpected format for generation: z%. Generation does not have a message.)rilenrrrfirst_generationrrr#get_message_generation_from_outputs s"    rcCsTd|vr|dSd|vr"|d}t|dkr|dStd|dtd|d) zRetrieve the prompt from the given inputs. Args: inputs: The inputs dictionary. Returns: str: The prompt. Raises: ValueError: If the prompt is not found or if multiple prompts are present. promptpromptsrWrz$Multiple prompts in run with inputs z!. Please create example manually.z)Could not find prompt in run with inputs r)rri)rrrrrget_prompt_from_inputs*s   rcCs`d|vr td|d|d}t|dkrtd||d}d|vr,td||dS) z(Get the LLM generation from the outputs.rrrrWzMultiple generations in run: rrxzNo text in generation: )rirrrrrget_llm_generation_from_outputsCs 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_callDEVNULLCalledProcessErrorFileNotFoundErrorrirrrrget_docker_compose_commandPs*  rls_schemas.BaseMessageLikedictcCs<|jd|jid}|jrt|jdkri|j|dd<|S)z*Convert a LangChain message to an example.contentrrradditional_kwargs)rrrr)r convertedrrrconvert_langchain_messageks robjobjectcCs@ttt|ddttt|ddtt|dott|dtgS)zCheck if the given object is similar to `BaseMessage`. Args: obj: The object to check. Returns: bool: True if the object is similar to `BaseMessage`, `False` otherwise. rNrr)allrgetattrr6rhasattr)rrrris_base_message_likews  rr Optional[str]cCs|durdStt|S)z2Check if the given environment variable is truish.NF) is_truishrK)rrrris_env_var_truishs r) LANGSMITH LANGCHAIN) namespacesnamerDrtuplecCdSNrrrDrrrrrKrKcCrrrrrrrrKrdcsHfdd|D}|D]tj}|dur!|dkr!|Sq |S)aRetrieve an environment variable from a list of namespaces. Args: name: The name of the environment variable. default: The default value to return if the environment variable is not found. namespaces: A tuple of namespaces to search for the environment variable. Defaults to `('LANGSMITH', 'LANGCHAINs')`. Returns: The value of the environment variable if found, otherwise the default value. csg|] }|dqS)_r)rY namespacerrrr`rgzget_env_var..NrB)osenvironrXstrip)rrDrnamesrrrrrKs Tc Cs&tjdtdtd|r dndddS)z,Get the project name for a LangSmith tracer.HOSTED_LANGSERVE_PROJECT_NAMEPROJECTSESSIONrDNrC)rrrXrK)return_default_valuerrrget_tracer_projects rcs.eZdZdZd dfdd Zdd d ZZS)FilterPoolFullWarningzGFilter `urllib3` warnings logged when the connection pool isn't reused.rBrr6hostr-r.cst|||_dS)zInitialize the `FilterPoolFullWarning` filter. Args: name: The name of the filter. Defaults to `""`. host: The host to filter. Defaults to `""`. N)r/r0_host)r1rrr2rrr0s  zFilterPoolFullWarning.__init__rOcCs|}d|vr dS|j|vS)zJurllib3.connectionpool:Connection pool is full, discarding connection: ...z.Connection pool is full, discarding connectionT) getMessagerr1recordmsgrrrfilters zFilterPoolFullWarning.filter)rBrB)rr6rr6r-r.r-rO)rrrrr0rr4rrr2rrs rc@seZdZdZdddZdS)FilterLangSmithRetryz!Filter for retries from this lib.r-rOcCs|}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 BaseExceptionrwarning)rrrrrr filter_logss2   ( rcachecCs|dur|StdddS)zGet the testing cache directory. Args: cache: The cache path. Returns: The cache path if provided, otherwise the value from the `LANGSMITH_TEST_CACHE` environment variable. N TEST_CACHErCrR)rrrr get_cache_dirs  r ignore_hosts allow_hostsrtr rOptional[Sequence[str]]rcs||rtfdd|DrdS|r9ztj}Wn ty#YdSw|jp(dtfdd|D}|s9dSi_S)zAFilter request headers based on `ignore_hosts` and `allow_hosts`.c3s|] }j|VqdSr)url startswithrYrrtrrr\(sz)filter_request_headers..NrBc3s>|]}|drj|n |kpd|VqdS))zhttp://zhttps://rN)rrendswithrrt request_hostrrr\3s )any urllib_parseurlparser Exceptionhostnameheaders)rtrr parsed_url host_matchesrrrfilter_request_headers s    r pathUnion[str, pathlib.Path]c #szddl}Wn tytdwddlm}|tj|\}}|j| ds1| dr3dnd|d gd d d gfd dd}| | dVWddS1sYwYdS)Use a cache for requests.rNzNvcrpy is required to use caching. Install with:pip install -U "langsmith[vcr]")_patchz.yamlz.ymlyamljson new_episodes)urimethodrbody authorizationz Set-Cookiecst|dS)Nr)r rrrrreszwith_cache..) serializercassette_library_dir record_modematch_onfilter_headersbefore_record_request) vcr ImportErrorlangsmith._internalr patch_urllib3rrsplitVCRr use_cassette)rrrr!r$ cache_dir cache_filels_vcrrrr with_cacheDs4     "r+"Optional[Union[str, pathlib.Path]]ccsL|dur!t||| dVWddS1swYdSdVdS)rN)r+)rrrrrrwith_optional_cachems " r-cCs&tjt}dd|D}d|S)NcSsg|]}d|vr|qS)z langsmith/r)rYlinerrrr`~sz_format_exc..rB) tracebackformat_exceptionsysexc_infore)tb_linesfiltered_linesrrr _format_exc{s r5Tvalmemodict[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|dqSra _middle_copy)rYkvr<r;r9rr s  z _middle_copy..csg|] }t|dqSrar>rYitemrBrrr`z _middle_copy..c3s"|] }t|dVqdS)rWNr>rDrBrrr\r]z_middle_copy..csh|] }t|dqSrar>rDrBrr rFz_middle_copy..) rrrrritemslistrset)r8r9r;r<clscopierrrBrr?s*       r?c 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)copydeepcopyrrdebugreprr?)r8r9r{rrr deepish_copys rQcurrent_versiontarget_versioncCs(ddlm}||}||}||kS)zGCheck if the current version is greater or equal to the target version.r)version) packagingrTr)rRrSrTcurrenttargetrrris_version_greater_or_equals   rXtuple[str, str, str]cCs|r|ddks|ds|drtt||dd}|d}t|dkr-|dnd}d|vrJ|dd\}}|r?|sEtt||||fS|sRtt|d||fS)a3Parse a string in the format of `owner/name:hash`, `name:hash`, `owner/name`, or `name`. Args: identifier: The prompt identifier to parse. Returns: A tuple containing `(owner, name, hash)`. Raises: ValueError: If the identifier doesn't match the expected formats. /rW:rlatest-)rcrrrir7r%r)r5parts owner_namecommitownerrrrrparse_prompt_identifiers&       rbPcs8eZdZdZdfd d Zd d ddfddZZS)ContextThreadPoolExecutorz?ThreadPoolExecutor that copies the context to the child thread.rVCallable[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 r6rnpartial contextvars copy_contextrun)r1rVr)r,r2rrris  z ContextThreadPoolExecutor.submitNrWtimeout chunksizefnCallable[..., T] iterables Iterable[Any]roOptional[float]rpr 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)rkrl)rYrrrrr`!rz1ContextThreadPoolExecutor.map..rr)r r-r6csjg|RSr)poprm)r)contextsrqrr _wrapped_fn#sz2ContextThreadPoolExecutor.map.._wrapped_fnrnN)r)r r-r6)rangerr/map)r1rqrorprsrzr2rxrr|szContextThreadPoolExecutor.map)rVrer)rfr,rgr-rh) rqrrrsrtrorurprr-rv)rrrrrir|r4rrr2rrds rdapi_urlcCs@|p tttddd}|std|dddS)zBGet the LangSmith API URL from the environment or the given value.ENDPOINTzhttps://api.smith.langchain.comrCz!LangSmith API URL cannot be empty"'rZ)rr6rKrr"rstrip)r}_api_urlrrr get_api_url.srapi_keycC@|dur|ntddd}|dus|sdS|ddS)z8Get the API key from the environment or the given value.NAPI_KEYrCrrrKr)rapi_key_rrr get_api_key<sr workspace_idcCr)zGet workspace ID.N WORKSPACE_IDrCrrr)r workspace_id_rrrget_workspace_idDs rrcCsVzt|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)rurlsplitnetlocr%socket gethostbynamergaierror)rriprrr _is_localhostPs  rweb_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 }|St|j  d r}d}|Sd}|S)z1Get the host URL based on the web URL or API URL.zhttp://localhostz/apirWr)rz/api/v1zeu.zhttps://eu.smith.langchain.comzaws.zhttps://aws.smith.langchain.comzdev.zhttps://dev.smith.langchain.comzbeta.z https://beta.smith.langchain.comzhttps://smith.langchain.com) rrrr6rrrsplit urlunparse_replacerr)rr}r linknew_pathrrr get_host_urles8   rrqdepthcCs|dkst|s t|St|dr|jSt|tjr"t|j|dSt|dr>t|dr6t|j dr6|j jSt|j |dSt|S)NrrrW__call__r3) callabler6rrrrnrj_get_function_namerVr3r)rqrrrrrs   rcCs&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. rE1)rr6rrO)r8rrrrs r)r5r6r-r6r)r:r;r-r<r)rTrUr-r )rrrsr-r.)r}r~r-r6)rrrr6r-r.)rrr-r6)rrr-r)rrr-r)rrr-r)rrr-r)rrr-r6)rrr-r6)r-r)rrr-r)rrr-rO)rrr-rO)rr6rDr6rrr-r6)rr6rDr.rrr-r)rr6rDrrrr-r)T)r-r)rrrrr-r)rrr-r)rtr rrrrr-r )NN)rrrrrrr-r)rr,rrrrr-r)r-r6)r7r) r8r6r9r:r;rr<rr-r6)r8r6r-r6)rRr6rSr6r-rO)r5r6r-rY)r}rr-r6)rrr-r)rrr-r)rr6r-rO)rrr}r6)r)rqr rrr-r6)r8r r-rO)or __future__r contextlibrkrMrrnloggingrpathlibrrr1 threadingr/collections.abcrrrrrconcurrent.futuresrr typingr r r r rrrrurllibrrryrvtyping_extensionsr urllib3.utilr langsmithr ls_schemas getLoggerrrrrr r!r"r#r$r%r&r'r(r7 UserWarningr8r9rNrSrqr|r lru_cacherrrrrrrrrrrrrKrFilterrrrRLockrcontextmanagerrrr r+r-r5r6r?rQrXrbrcrdrrrrrrrrrrrs (                            $(    # G