o +if&@sdZddlZddlZddlZddlZddlZddlmZmZm Z m Z m Z m Z m Z mZddlZddlmZeeZGdddeZGdddeZGd d d eZGd d d eZGd ddeZGdddeZGdddeZGdddeZdefddZ de e!dfdefddZ"dej#ddfddZ$deej%e!fde!fd d!Z&ej'd"d#d$e(d%e!ddfd&d'Z)d%e e!efde!fd(d)Z*d%e e!efde e!effd*d+Z+d%e e!efde e!effd,d-Z,d.e e!efde e e!effd/d0Z-d1e e!efde e!effd2d3Z.d.e e!efde!fd4d5Z/d1e e!efde!fd6d7Z0ej'd"d#de e!fd8d9Z1d%ej2de3fd:d;Z4dZ6dDde e!fd@dAZ7GdBdCdCej8Z9dS)EzGeneric utility functions.N)AnyCallableDictListMappingOptionalTupleUnion)schemasc@eZdZdZdS)LangSmithErrorz=An error occurred while communicating with the LangSmith API.N__name__ __module__ __qualname____doc__rrH/var/www/html/corbot_env/lib/python3.10/site-packages/langsmith/utils.pyr r c@r )LangSmithAPIErrorz9Internal server error while communicating with LangSmith.Nr rrrrrrrc@r )LangSmithUserErrorzAUser error caused an exception when communicating with LangSmith.Nr rrrrrrrc@r )LangSmithRateLimitErrorz7You have exceeded the rate limit for the LangSmith API.Nr rrrrrrrc@r )LangSmithAuthErrorz-Couldn't authenticate with the LangSmith API.Nr rrrrr!rrc@r )LangSmithNotFoundErrorz%Couldn't find the requested resource.Nr rrrrr%rrc@r )LangSmithConflictErrorzThe resource already exists.Nr rrrrr)rrc@r )LangSmithConnectionErrorz&Couldn't connect to the LangSmith API.Nr rrrrr-rrreturncCs tjdtjdddkS)z"Return True if tracing is enabled.LANGCHAIN_TRACING_V2LANGCHAIN_TRACINGtrue)osenvirongetlowerrrrrtracing_is_enabled1sr% arg_groups.csdtdtffdd }|S)z7Validate specified keyword args are mutually exclusive.funcrcs*tdtdtdtffdd }|S)Nargskwargsrcs\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)r#).0argr)rr Cs zJxor_args..decorator..wrapper...)sum)r+ arg_groupr-rr Bsz@xor_args..decorator..wrapper..cSsg|] \}}|dkr|qS)r*r)r+icountrrrr1Fscsg|] }d|qS), )join)r+r2r&rrr1HszFExactly one argument in each of the following groups must be defined: r4) enumerate ValueErrorr5)r(r)countsinvalid_groupsinvalid_group_names)r&r'r-rwrapper?s z,xor_args..decorator..wrapper) functoolswrapsr)r'r<r6)r'r decorator>szxor_args..decorator)r)r&r?rr6rxor_args;sr@responsec Cs>z|WdStjy}z tt||j|d}~ww)z&Raise an error with the response text.N)raise_for_statusrequests HTTPErrorstrtext)rAerrrraise_for_status_with_textUs rHenucCst|tjr |jS|S)zGet the value of a string enum.) isinstanceenumEnumvalue)rIrrrget_enum_value]s rNr*)maxsizelevelmessagecCst||dS)N)_LOGGERlog)rPrQrrrlog_oncedsrTcCsd|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.Messagertype&Unexpected format for stored message: z Message does not have a type.)r8replacer$rQrrr_get_message_typeis  r_cCsT|stdd|vrd|vrtd|d|dSd|vr&td|d|dS) NrUrVr)rXz Message does not have kwargs.datar\z Message does not have data.)r8r^rrr_get_message_fields|s  racCst|}t|}||dS)z&Extract message from a message object.r[r`)r_ra)rQ message_type message_datarrr_convert_messages reinputscCs@d|vr dd|dDSd|vrt|dgStd|d)NmessagescSsg|]}t|qSr)re)r+rQrrrr1sz,get_messages_from_inputs..rQz-Could not find message(s) in run with inputs .)rer8)rfrrrget_messages_from_inputss rioutputscCsrd|vr td|d|d}t|dkr#tdt|d|d|d}d|vr3td |d t|dS) N generations,No generations found in in run with output: rhr*z3Chat examples expect exactly one generation. Found z generations: rrQz"Unexpected format for generation: z%. Generation does not have a message.)r8lenrerjrkfirst_generationrrr#get_message_generation_from_outputss"   rpcCsTd|vr|dSd|vr"|d}t|dkr|dStd|dtd|d) Npromptpromptsr*rz$Multiple prompts in run with inputs z!. Please create example manually.z)Could not find prompt in run with inputs rh)rmr8)rfrrrrrget_prompt_from_inputss  rscCs`d|vr td|d|d}t|dkrtd||d}d|vr,td||dS) Nrkrlrhr*zMultiple generations in run: rrFzNo text in generation: )r8rmrnrrrget_llm_generation_from_outputss rtc CszztjgdtjtjdddgWStjtfy<ztjddgtjtjddgWYStjtfy;tdww)z7Get the correct docker compose command for this system.)dockercompose --version)stdoutstderrrurvzdocker-composerwzNeither '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_callDEVNULLCalledProcessErrorFileNotFoundErrorr8rrrrget_docker_compose_commands*  rcCs<|jd|jid}|jrt|jdkri|j|dd<|S)z*Convert a LangChain message to an example.contentrbrr`additional_kwargs)r[rrrm)rQ convertedrrrconvert_langchain_messages robjcCs@ttt|ddttt|ddtt|dott|dtgS)z Check 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[)allrJgetattrrEdicthasattr)rrrris_base_message_likes  rTc Cs0tjdtjdtjd|rdSdS)z,Get the project name for a LangSmith tracer.HOSTED_LANGSERVE_PROJECT_NAMELANGCHAIN_PROJECTLANGCHAIN_SESSIONdefaultN)r!r"r#)return_default_valuerrrget_tracer_projects rc@seZdZdZdefddZdS)FilterPoolFullWarningzFFilter urrllib3 warnings logged when the connection pool isn't reused.rcCs d|vS)zJurllib3.connectionpool:Connection pool is full, discarding connection: ...z.Connection pool is full, discarding connection) getMessage)selfrecordrrrfilters zFilterPoolFullWarning.filterN)rrrrboolrrrrrrsr)T):rrKr=loggingr!rztypingrrrrrrrr rC langsmithr ls_schemas getLoggerrrR Exceptionr rrrrrrrrr%rEr@ResponserHrLrN lru_cacheintrTr_rarerirprsrtrBaseMessageLikerrobjectrrFilterrrrrrsJ(    ""&"