o yq io@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_ZNddbdcZOejCddd< ddedfddkdlZPejCd;d<dddndoZQGdpdqdqejRZSGdrdsdsejRZTGdtdudue)ZUeVZWejXdd{d|ZYdd~dZZddddddZ[ejX  ddddZ\ejX  ddddZ]dddZ^edZ_ ddddZ`dddZadddZbdddZce'dZdGdddeZedddZfdddZgdddZhdddZiejCdd<dddZjddddZkdddZldS)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__rrV/home/app/PaddleOCR-VL/.venv_paddleocr/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']]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_contextenabledTZ TRACING_V2ZTRACINGdefaulttrue)Zlangsmith._internal._context _internal_contextZlangsmith.run_helpersr9r:Z_GLOBAL_TRACING_ENABLED get_env_var)r6rAr9r:ZtcZ var_resultrrrtracing_is_enabledcs   rCboolcCstddddkS)z"Return True if testing is enabled.Z TEST_TRACKINGr<r=falserBrrrrtest_tracking_is_disabled{srG 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)rMZ arg_grouprOrr sz@xor_args..decorator..wrapper..cSsg|] \}}|dkr|qSrKr)rMicountrrrrSscsg|] }d|qS), )join)rMrUrHrrrSzFExactly one argument in each of the following groups must be defined: rW) enumerate ValueErrorrX)r(r+countsZinvalid_groupsZinvalid_group_names)rHrJrOrwrappers z,xor_args..decorator..wrapper)r(r r+r r,r ) functoolswraps)rJr^rY)rJr decoratorszxor_args..decoratorN)rJr r,r r)rHrarrYrxor_argssrbresponse(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: )requestrc)raise_for_statusrequests HTTPErrorstrtexthttpxZHTTPStatusErrorre)rcerrrraise_for_status_with_textsrmenuUnion[enum.Enum, str]ricCst|tjr |jS|S)zGet the value of a string enum.) isinstanceenumEnumvalue)rnrrrget_enum_values rtrK)maxsizelevelintmessagecCst||dS)z4Log a message at the specified level, but only once.N)_LOGGERlog)rvrxrrrlog_oncesr{Mapping[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.)r\replacelowerrxrrr_get_message_types  rcCsT|stdd|vrd|vrtd|d|dSd|vr&td|d|dS) Nr}r~r+rz Message does not have kwargs.datarz Message does not have data.)r\rrrr_get_message_fieldss  rdict[str, Any]cCst|}t|}||dS)z&Extract message from a message object.rr)rr)rxZ 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)rMrxrrrrSz,get_messages_from_inputs..rxz-Could not find message(s) in run with inputs .)rr\)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: rrKz3Chat examples expect exactly one generation. Found z generations: rrxz"Unexpected format for generation: z%. Generation does not have a message.)r\lenrrrZfirst_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. promptpromptsrKrz$Multiple prompts in run with inputs z!. Please create example manually.z)Could not find prompt in run with inputs r)rr\)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.rrrrKzMultiple generations in run: rrjzNo text in generation: )r\rrrrrget_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_callDEVNULLCalledProcessErrorFileNotFoundErrorr\rrrrget_docker_compose_command;s*  rls_schemas.BaseMessageLikedictcCs<|jd|jid}|jrt|jdkri|j|dd<|S)z*Convert a LangChain message to an example.contentrrradditional_kwargs)rrrr)rxZ convertedrrrconvert_langchain_messageVs 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)allrpgetattrrirhasattr)rrrris_base_message_likebs  rrs Optional[str]cCs tt|S)z2Check if the given environment variable is truish.) is_truishrB)rsrrris_env_var_truishts rd)Z LANGSMITHZ LANGCHAIN) namespacesnamer>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)rM namespacerrrrSrZzget_env_var..N)osenvironrL)rr>rnamesrsrrrrBys rBTc Cs&tjdtdtd|r dndddS)z,Get the project name for a LangSmith tracer.ZHOSTED_LANGSERVE_PROJECT_NAMEZPROJECTZSESSIONr>Nr=)rrrLrB)Zreturn_default_valuerrrget_tracer_projects rcs.eZdZdZd dfdd Zdd d ZZS)FilterPoolFullWarningzFFilter urrllib3 warnings logged when the connection pool isn't reused.r<rrihostr,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__rDcCs|}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<)rrirrir,r-r,rD)rrrrr/rr3rrr1rrs rc@seZdZdZdddZdS)FilterLangSmithRetryz!Filter for retries from this lib.r,rDcCs|}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 BaseExceptionrywarning)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=rF)rrrr get_cache_dirs  r ignore_hosts allow_hostsrer rOptional[Sequence[str]]rcs||rtfdd|DrdS|r9ztj}Wn ty#YdSw|jp(dtfdd|D}|s9dSi_S)z=Filter request headers based on ignore_hosts and allow_hosts.c3s|] }j|VqdSN)url startswithrMrrerrrPsz)filter_request_headers..Nr<c3s>|]}|drj|n |kpd|VqdS))zhttp://zhttps://rN)rrendswithrre request_hostrrrP s )any urllib_parseurlparser Exceptionhostnameheaders)rerr parsed_urlZ host_matchesrrrfilter_request_headerss    rpathUnion[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.ymlyamljsonZ new_episodes)urimethodrbody authorizationz Set-Cookiecst|dS)Nr)rrrrrr=szwith_cache..) serializerZcassette_library_dirZ record_modeZmatch_onZfilter_headersZbefore_record_request) vcr ImportErrorZlangsmith._internalr patch_urllib3rrsplitZVCRrZ use_cassette)rrrrr cache_dirZ cache_fileZls_vcrrrr with_caches4     "r"Optional[Union[str, pathlib.Path]]ccsL|dur!t||| dVWddS1swYdSdVdS)rN)r)rrrrrrwith_optional_cacheEs " rcCs&tjt}dd|D}d|S)NcSsg|]}d|vr|qS)z langsmith/r)rMlinerrrrSVsz_format_exc..r<) tracebackformat_exceptionsysexc_inforX)Ztb_linesZfiltered_linesrrr _format_excSs r Tvalmemodict[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|dqSrT _middle_copy)rMkvrrr rr ks  z _middle_copy..csg|] }t|dqSrTrrMitemrrrrSrz _middle_copy..c3s"|] }t|dVqdS)rKNrrrrrrPtrQz_middle_copy..csh|] }t|dqSrTrrrrr vrz_middle_copy..) rrrrpritemslistrset)r r rrclscopierrrrr]s*       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)copydeepcopyrrydebugreprr)r r rlrrr deepish_copy{s r%current_versiontarget_versioncCs(ddlm}||}||}||kS)zGCheck if the current version is greater or equal to the target version.r)version) packagingr(r)r&r'r(currenttargetrrris_version_greater_or_equals   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. /rKzInvalid identifier format: :rZlatest-)rVrrr\rr)r-partsZ owner_namecommitownerrrrrparse_prompt_identifiers&    r5Pcs8eZdZdZdfd d Zd d ddfddZZS)ContextThreadPoolExecutorz?ThreadPoolExecutor that copies the context to the child thread.rJCallable[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 r r_partial contextvars copy_contextrun)r0rJr(r+r1rrr<s  z ContextThreadPoolExecutor.submitNrKtimeout chunksizefnCallable[..., T] iterables Iterable[Any]rBOptional[float]rCrw 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>r?)rMrrrrrSrz1ContextThreadPoolExecutor.map..rr(r r,r csjg|RSr)popr@)r(ZcontextsrDrr _wrapped_fnsz2ContextThreadPoolExecutor.map.._wrapped_fnrAN)r(r r,r )rangerr.map)r0rDrBrCrFrLr1rKrrNszContextThreadPoolExecutor.map)rJr8r(r9r+r:r,r;) rDrErFrGrBrHrCrwr,rI)rrrrr<rNr3rrr1rr7s r7api_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/)rrirBstripr!rstrip)rOZ_api_urlrrr get_api_urlsrTapi_keycC@|dur|ntddd}|dus|sdS|ddS)z8Get the API key from the environment or the given value.NZAPI_KEYr=rPrQrBrR)rUZapi_key_rrr get_api_keysrX workspace_idcCrV)zGet workspace ID.NZ WORKSPACE_IDr=rPrQrW)rYZ workspace_id_rrrget_workspace_ids rZrcCsVzt|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. r0rz 127.0.0.1z0.0.0.0z::F)rurlsplitnetlocrsocket gethostbynamergaierror)rr\iprrr _is_localhost(s  raweb_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/apirKr)rz/api/v1zeu.zhttps://eu.smith.langchain.comzdev.zhttps://dev.smith.langchain.comzbeta.z https://beta.smith.langchain.comzhttps://smith.langchain.com) rrrarirrrsplit urlunparse_replacer\r)rcrOrlinknew_pathrrr get_host_url=s2   rirDdepthcCs|dkst|s t|St|dr|jSt|tjr"t|j|dSt|dr>t|dr6t|j dr6|j jSt|j |dSt|S)NrbrrK__call__r2) callablerirrrpr_r=_get_function_namerJr2rk)rDrjrrrrmVs   rmcCs&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)rprirrD)r rrrrhs rr)r6r7r,r8r)rHrIr,r )rcrdr,r-)rnror,ri)rvrwrxrir,r-)rxr|r,ri)rxr|r,r|)rxr|r,r)rr|r,r)rr|r,r)rr|r,ri)rr|r,ri)r,r)rxrr,r)rrr,rD)rsrr,rD)rrir>rrrr,r)T)r,r)rrrrr,r)rrr,r)rer rrrrr,r )NN)rrrrrrr,r)rrrrrrr,r)r,ri)r r) r r r rrrwrrwr,r )r r r,r )r&rir'rir,rD)r-rir,r.)rOrr,ri)rUrr,r)rYrr,r)rrir,rD)rcrrOri)r)rDr rjrwr,ri)r r r,rD)mr __future__r contextlibr>r!rqr_loggingrpathlibr]rr threadingrcollections.abcrrrrrconcurrent.futuresrr typingr r r r rrrurllibrrrkrgZtyping_extensionsrZ urllib3.utilrZ langsmithrZ ls_schemas getLoggerrryrrrr r!r"r#r$r%r&r' UserWarningr4r5rCrGrbrmrt lru_cacher{rrrrrrrrrrrrBrFilterrrrRLockrcontextmanagerrrrrrr r rr%r,r5r6r7rTrXrZrarirmrrrrrs $                           $(    # G