ܘh&dZddlZddlZddlZddlZddlZddlmZmZm Z m Z m Z m Z m Z mZddlZddlmZejeZGddeZGddeZGd d eZGd d eZGd deZGddeZGddeZGddeZdefdZ de e!dfdefdZ"dej#ddfdZ$deej%e!fde!fdZ&ej'dd e(d!e!ddfd"Z)d!e e!efde!fd#Z*d!e e!efde e!effd$Z+d!e e!efde e!effd%Z,d&e e!efde e e!effd'Z-d(e e!efde e!effd)Z.d&e e!efde!fd*Z/d(e e!efde!fd+Z0ej'dde e!fd,Z1d!ej2de3fd-Z4d.e5defd/Z6d4de e!fd1Z7Gd2d3ej8Z9dS)5zGeneric utility functions.N)AnyCallableDictListMappingOptionalTupleUnion)schemasceZdZdZdS)LangSmithErrorz=An error occurred while communicating with the LangSmith API.N__name__ __module__ __qualname____doc__W/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/langsmith/utils.pyr r sGGGGrr ceZdZdZdS)LangSmithAPIErrorz9Internal server error while communicating with LangSmith.NrrrrrrsCCCCrrceZdZdZdS)LangSmithUserErrorzAUser error caused an exception when communicating with LangSmith.NrrrrrrsKKKKrrceZdZdZdS)LangSmithRateLimitErrorz7You have exceeded the rate limit for the LangSmith API.NrrrrrrsAAAArrceZdZdZdS)LangSmithAuthErrorz-Couldn't authenticate with the LangSmith API.Nrrrrrr!s7777rrceZdZdZdS)LangSmithNotFoundErrorz%Couldn't find the requested resource.Nrrrrrr%s////rrceZdZdZdS)LangSmithConflictErrorzThe resource already exists.Nrrrrr!r!)s&&&&rr!ceZdZdZdS)LangSmithConnectionErrorz&Couldn't connect to the LangSmith API.Nrrrrr#r#-s0000rr#returnctjdtjdddkS)z"Return True if tracing is enabled.LANGCHAIN_TRACING_V2LANGCHAIN_TRACINGtrue)osenvirongetlowerrrrtracing_is_enabledr.1sC  "BJNN3F$K$K  %''  r arg_groups.c0dtdtffd }|S)z7Validate specified keyword args are mutually exclusive.funcr$cvtjdtdtdtffd }|S)Nargskwargsr$cfdD}dt|D}|r3fd|D}tdd||iS)z3Validate exactly one arg in each group is not None.cFg|]}tfd|DS)c3FK|]}|dVdS)N)r,).0argr4s r zJxor_args..decorator..wrapper...Cs2JJ#fjjoo.IA.I.I.I.IJJr)sum)r9 arg_groupr4s r z@xor_args..decorator..wrapper..BsFJJJJJJJJJrc$g|] \}}|dk |S)r8r)r9icounts rr>z@xor_args..decorator..wrapper..Fs!PPPHAuUaZZaZZZrcFg|]}d|S), )join)r9r@r/s rr>z@xor_args..decorator..wrapper..Hs)&X&X&XAtyyA'?'?&X&X&XrzFExactly one argument in each of the following groups must be defined: rC) enumerate ValueErrorrD)r3r4countsinvalid_groupsinvalid_group_namesr/r1s ` rwrapperz,xor_args..decorator..wrapper?s!+FQP &0A0APPPN &X&X&X&X&X&X&X# 9 "56699 4((( (r) functoolswrapsr)r1rJr/s` r decoratorzxor_args..decorator>sX    )3 )# )# ) ) ) ) ) )   ) r)r)r/rMs` rxor_argsrN;s7X( rresponsec |dS#tj$r-}tjt||j|d}~wwxYw)z&Raise an error with the response text.N)raise_for_statusrequests HTTPErrorstrtext)rOes rraise_for_status_with_textrWUs^?!!#####  ??? Q77Q>?sA(AAenucHt|tjr|jS|S)zGet the value of a string enum.) isinstanceenumEnumvalue)rXs rget_enum_valuer^]s##ty!!y Jrr8)maxsizelevelmessagec<t||dS)N)_LOGGERlog)r`ras rlog_oncereds KKwrc|stdd|vrKd|vrtd|d|ddddSd |vrtd |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.)rFreplacer-ras r_get_message_typerqis .,--- w w  0W000 t}R ((B77==???  1111 vrc|stdd|vrd|vrtd|d|dSd|vrtd|d|dS) Nrgrhr4rjz Message does not have kwargs.datarnz Message does not have data.)rFrps r_get_message_fieldsrt|s .,--- w 7 " "1W111 x   //// vrcHt|}t|}||dS)z&Extract message from a message object.rmrs)rqrt)ra message_type message_datas r_convert_messagerys*$W--L&w//L , 7 77rinputscd|vrd|dDSd|vrt|dgStd|d)Nmessagesc,g|]}t|Sr)ry)r9ras rr>z,get_messages_from_inputs..s!LLLg ))LLLrraz-Could not find message(s) in run with inputs .)ryrF)rzs rget_messages_from_inputsrsaVLL 9KLLLLF  !23344 NVNNN O OOroutputscd|vrtd|d|d}t|dkr#tdt|d|d|d}d|vrtd |d t|dS) N generations,No generations found in in run with output: r~r8z3Chat examples expect exactly one generation. Found z generations: rraz"Unexpected format for generation: z%. Generation does not have a message.)rFlenryrrfirst_generations r#get_message_generation_from_outputsrsG##RRRRSSS-(K ;1 E+&& E E6A E E E   #1~((( 31A 3 3 3    ,Y7 8 88rcd|vr|dSd|vr6|d}t|dkr|dStd|dtd|d) Npromptpromptsr8rz$Multiple prompts in run with inputs z!. Please create example manually.z)Could not find prompt in run with inputs r~)rrF)rzrs rget_prompt_from_inputsrs6hF# w<<1  1:  /6 / / /    JJJJ K KKrcd|vrtd|d|d}t|dkrtd||d}d|vrtd||dS) Nrrr~r8zMultiple generations in run: rrUzNo text in generation: )rFrrs rget_llm_generation_from_outputsrsG##RRRRSSS-(K ;1FFFGGG"1~ %%%E3CEEFFF F ##rcR tjgdtjtjddgS#tjtf$r] tjddgtjtjdgcYS#tjtf$rt dwxYwwxYw)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_callDEVNULLCalledProcessErrorFileNotFoundErrorrFrrrget_docker_compose_commandrs . . .%%    )$$  )+< =   !!;/!)!)     %% % % %-/@A   E  s!03B& /A<9B&<&B""B&c|jd|jid}|jr*t|jdkri|j|dd<|S)z*Convert a LangChain message to an example.contentrvrrsadditional_kwargs)rmrrr)ra converteds rconvert_langchain_messagers` GO,!!I  OS)B%C%Ca%G%G1NG4M1N &-. robjc ttt|ddttt|ddtt |do"tt|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. rNrrm)allrZgetattrrTdicthasattr)rs ris_base_message_likerst  wsIt44c : : ws$7>> E E C JZV0D0Dc%J%J   rTc tjdtjdtjd|rdndS)z,Get the project name for a LangSmith tracer.HOSTED_LANGSERVE_PROJECT_NAMELANGCHAIN_PROJECTLANGCHAIN_SESSIONdefaultN)r*r+r,)return_default_values rget_tracer_projectrsY :>> (   JNN$1; t       rceZdZdZdefdZdS)FilterPoolFullWarningzFFilter urrllib3 warnings logged when the connection pool isn't reused.r$c.d|vS)zJurllib3.connectionpool:Connection pool is full, discarding connection: ...z.Connection pool is full, discarding connection) getMessage)selfrecords rfilterzFilterPoolFullWarning.filters =FDUDUDWDW W rN)rrrrboolrrrrrrs5PP       rr)T):rr[rKloggingr*rtypingrrrrrrr r rR langsmithr ls_schemas getLoggerrrc Exceptionr rrrrrr!r#rr.rTrNResponserWr\r^ lru_cacheintrerqrtryrrrrrBaseMessageLikerrobjectrrFilterrrrrrs1  MMMMMMMMMMMMMMMMMMMM++++++ ' H % %HHHHHYHHHDDDDDDDDLLLLLLLLBBBBBnBBB8888888800000^000'''''^'''11111~111D%S/h4?):?t????din-#Q C # $     wsCx0S&c!2wsCx7H&8gc3h/8DcN8888PWS#X%6P4S#X;OPPPP9c1B9tCQTH~9999$ L738#4 L L L L L $WS#X-> $3 $ $ $ $QDI 4 z'A d    f&Xc](     GN     r