o +in@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__rrL/var/www/html/psymed-ai/venv/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_contextenabledT TRACING_V2TRACINGdefaulttrue)langsmith._internal._context _internal_contextlangsmith.run_helpersr9r:_GLOBAL_TRACING_ENABLED get_env_var)r6rDr9r:tc var_resultrrrtracing_is_enabledcs   rJboolcCstddddkS)z"Return True if testing is enabled. TEST_TRACKINGr>r?falserGrrrrtest_tracking_is_disabled{srO 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)rU arg_grouprWrr sz@xor_args..decorator..wrapper..cSsg|] \}}|dkr|qSrSr)rUicountrrrr\scsg|] }d|qS), )join)rUr^rPrrr\zFExactly one argument in each of the following groups must be defined: r`) enumerate ValueErrorra)r(r+countsinvalid_groupsinvalid_group_names)rPrRrWrwrappers z,xor_args..decorator..wrapper)r(r r+r r,r ) functoolswraps)rRrirb)rRr decoratorszxor_args..decoratorN)rRr r,r r)rPrlrrbrxor_argssrmresponse(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: )requestrn)raise_for_statusrequests HTTPErrorstrtexthttpxHTTPStatusErrorrp)rnerrrraise_for_status_with_textsryenuUnion[enum.Enum, str]rtcCst|tjr |jS|S)zGet the value of a string enum.) isinstanceenumEnumvalue)rzrrrget_enum_values rrS)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.Messager>type&Unexpected format for stored message: z Message does not have a type.)rereplacelowerrrrr_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.)rerrrr_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)rUrrrrr\z,get_messages_from_inputs..rz-Could not find message(s) in run with inputs .)rre)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: rrSz3Chat examples expect exactly one generation. Found z generations: rrz"Unexpected format for generation: z%. Generation does not have a message.)relenrrrfirst_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: The inputs dictionary. Returns: str: The prompt. Raises: ValueError: If the prompt is not found or if multiple prompts are present. promptpromptsrSrz$Multiple prompts in run with inputs z!. Please create example manually.z)Could not find prompt in run with inputs r)rre)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.rrrrSzMultiple generations in run: rruzNo text in generation: )rerrrrrget_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_callDEVNULLCalledProcessErrorFileNotFoundErrorrerrrrget_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)r convertedrrrconvert_langchain_messageUs 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)allr|getattrrtrhasattr)rrrris_base_message_likeas  rr Optional[str]cCs tt|S)z2Check if the given environment variable is truish.) is_truishrG)rrrris_env_var_truishss rd) LANGSMITH LANGCHAIN) namespacesnamer@rtuplecs<fdd|D}|D]tj}|dur|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)rU namespacerrrr\rczget_env_var..N)osenvironrT)rr@rnamesrrrrrGxs rGTc Cs&tjdtdtd|r dndddS)z,Get the project name for a LangSmith tracer.HOSTED_LANGSERVE_PROJECT_NAMEPROJECTSESSIONr@Nr?)rrrTrG)return_default_valuerrrget_tracer_projects rcs.eZdZdZd dfdd Zdd d ZZS)FilterPoolFullWarningzGFilter `urllib3` warnings logged when the connection pool isn't reused.r>rrthostr,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.r/_host)r0rrr1rrr/s  zFilterPoolFullWarning.__init__rKcCs|}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>)rrtrrtr,r-r,rK)rrrrr/rr3rrr1rrs rc@seZdZdZdddZdS)FilterLangSmithRetryz!Filter for retries from this lib.r,rKcCs|}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_CACHEr?rN)rrrr get_cache_dirs  r ignore_hosts allow_hostsrpr 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|VqdSN)url startswithrUrrprrrXsz)filter_request_headers..Nr>c3s>|]}|drj|n |kpd|VqdS))zhttp://zhttps://rN)rrendswithrrp request_hostrrrX s )any urllib_parseurlparser Exceptionhostnameheaders)rprr parsed_url 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.ymlyamljson new_episodes)urimethodrbody authorizationz Set-Cookiecst|dS)Nr)rrrrrr;szwith_cache..) serializercassette_library_dir record_modematch_onfilter_headersbefore_record_request) vcr ImportErrorlangsmith._internalr  patch_urllib3rrsplitVCRr use_cassette)rrrrr cache_dir cache_filels_vcrrrr with_caches4     "r$"Optional[Union[str, pathlib.Path]]ccsL|dur!t||| dVWddS1swYdSdVdS)r N)r$)rrrrrrwith_optional_cacheCs " r&cCs&tjt}dd|D}d|S)NcSsg|]}d|vr|qS)z langsmith/r)rUlinerrrr\Tsz_format_exc..r>) tracebackformat_exceptionsysexc_infora)tb_linesfiltered_linesrrr _format_excQs 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|dqSr] _middle_copy)rUkvr5r4r2rr is  z _middle_copy..csg|] }t|dqSr]r7rUitemr;rrr\pz _middle_copy..c3s"|] }t|dVqdS)rSNr7r=r;rrrXrrYz_middle_copy..csh|] }t|dqSr]r7r=r;rr tr?z_middle_copy..) rrrr|ritemslistrset)r1r2r4r5clscopierrr;rr8[s*       r8c 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)copydeepcopyrrdebugreprr8)r1r2rxrrr deepish_copyys rJcurrent_versiontarget_versioncCs(ddlm}||}||}||kS)zGCheck if the current version is greater or equal to the target version.r)version) packagingrMr)rKrLrMcurrenttargetrrris_version_greater_or_equals   rQ identifiertuple[str, str, str]cCs|r|ddks|ds|drtd||dd}|d}t|dkr.|dnd}d|vrL|dd\}}|r@|sGtd||||fS|sUtd|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. /rSzInvalid identifier format: :rlatest-)r_rrrerr)rRparts owner_namecommitownerrrrrparse_prompt_identifiers&    r\Pcs8eZdZdZdfd d Zd d ddfddZZS)ContextThreadPoolExecutorz?ThreadPoolExecutor that copies the context to the child thread.rRCallable[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/rjpartial contextvars copy_contextrun)r0rRr(r+r1rrrcs  z ContextThreadPoolExecutor.submitNrStimeout chunksizefnCallable[..., T] iterables Iterable[Any]riOptional[float]rjr 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)rerf)rUrrrrr\rz1ContextThreadPoolExecutor.map..rr(r r,r/csjg|RSr)poprg)r(contextsrkrr _wrapped_fnsz2ContextThreadPoolExecutor.map.._wrapped_fnrhN)r(r r,r/)rangerr.map)r0rkrirjrmrtr1rrrrvszContextThreadPoolExecutor.map)rRr_r(r`r+rar,rb) rkrlrmrnrirorjrr,rp)rrrrrcrvr3rrr1rr^s r^api_urlcCs@|p tttddd}|std|dddS)zBGet the LangSmith API URL from the environment or the given value.ENDPOINTzhttps://api.smith.langchain.comr?z!LangSmith API URL cannot be empty"'rT)rrtrGstripr!rstrip)rw_api_urlrrr get_api_urlsr~api_keycC@|dur|ntddd}|dus|sdS|ddS)z8Get the API key from the environment or the given value.NAPI_KEYr?ryrzrGr{)rapi_key_rrr get_api_keysr workspace_idcCr)zGet workspace ID.N WORKSPACE_IDr?ryrzr)r workspace_id_rrrget_workspace_ids 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. rUrz 127.0.0.1z0.0.0.0z::F)rurlsplitnetlocrsocket gethostbynamergaierror)rriprrr _is_localhost&s  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 }|Sd }|S)z1Get the host URL based on the web URL or API URL.zhttp://localhostz/apirSr)rz/api/v1zeu.zhttps://eu.smith.langchain.comzdev.zhttps://dev.smith.langchain.comzbeta.z https://beta.smith.langchain.comzhttps://smith.langchain.com) rrrrtrrrsplit urlunparse_replacerr)rrwrlinknew_pathrrr get_host_url;s2   rrkdepthcCs|dkst|s t|St|dr|jSt|tjr"t|j|dSt|dr>t|dr6t|j dr6|j jSt|j |dSt|S)NrrrS__call__r2) callablertrrr|rjrd_get_function_namerRr2r)rkrrrrrTs   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. rA1)r|rtrrK)r1rrrrfs rr)r6r7r,r8r)rPrQr,r )rnror,r-)rzr{r,rt)rrrrtr,r-)rrr,rt)rrr,r)rrr,r)rrr,r)rrr,r)rrr,rt)rrr,rt)r,r)rrr,r)rrr,rK)rrr,rK)rrtr@rrrr,r)T)r,r)rrrrr,r)rrr,r)rpr rrrrr,r )NN)rrrrrrr,r)rr%rrrrr,r)r,rt)r0r) r1r/r2r3r4rr5rr,r/)r1r/r,r/)rKrtrLrtr,rK)rRrtr,rS)rwrr,rt)rrr,r)rrr,r)rrtr,rK)rrrwrt)r)rkr rrr,rt)r1r r,rK)mr __future__r contextlibrerFr}rjloggingrpathlibrrr* threadingr(collections.abcrrrrrconcurrent.futuresrr typingr r r r rrrurllibrrrvrrtyping_extensionsr urllib3.utilr langsmithr ls_schemas getLoggerrrrrrr r!r"r#r$r%r&r' UserWarningr4r5rJrOrmryr lru_cacherrrrrrrrrrrrrGrFilterrrrRLockrcontextmanagerrrrr$r&r.r/r8rJrQr\r]r^r~rrrrrrrrrrs $                           $(    # G