6|haXUdZddlmZddlZddlZddlZddlmZmZddlm Z m Z m Z m Z m Z mZmZmZmZddlmZmZ ddlmZmZddlZddlZddlmZdd lm Z dd l!m"Z"m#Z#m$Z$m%Z%m&Z&ejNe(Z)d Z*ejVe*d Z,e,j[d Z.ejVe*dZ/ejVe*dZ0ejVe*dZ1da2de3d<ejhZ5ddZ6GddejnZ8GddZ9ddZ: ddZ;ddgZ dTd?Z$ dUd d d d d d d d d d d d@ dVdAZ%dBZ&dWdXdCZ'dYdDZ(dYdEZ)dZdFZ*e+ d[dGZ,e+ d\dHZ-e+ d]dIZ.d^dJZ/dKZ0xZ1S)_RunTreez1Run Schema with back-references for posting runs.strname)default_factoryridchain)defaultrun_typecHtjtjSN)rnowrutcr"r&r$zRunTree.8shll9Sr&r start_timeNT)r.excludeOptional[RunTree] parent_run__all__ parent_run_id)r+r6z List[RunTree] child_runsc2tjxsdS)Nr.)rget_tracer_projectr"r&r$r4zRunTree.?s 8 8 : Gir& project_name)r+alias session_name project_id)r.r?Optional[UUID] session_idrextraOptional[List[str]]rz List[Dict]eventsz Optional[Any] ls_clientz!The order of the run in the tree.)r. description dotted_orderzThe trace id of the run.trace_idFz3Whether to allow filesystem access for attachments.zOptional[bool]dangerously_allow_filesystemceZdZdZdZdZdZy)RunTree.ConfigzPydantic model configuration.TignoreN)__name__ __module__ __qualname____doc__arbitrary_types_allowedallow_population_by_field_namerDr"r&r$ConfigrNQs+"&)-&r&rV)prec|jd9|jd(d|dvr |dd|d<nd|dvr|ddd|d<|jdd|d<d|vr|jd|d<nd |vr|jd |d<|jdsd|d<|jd |d j|d <d|vr t|d<d |vrd |vr|d j|d <n|d|d <t t |jd i|jdg|d<|jdg|d<|jdi|d<|jdi|d<|S)zAssign name to the run.r*N serializedr,UnnamedclientrG_clientr8r:rKrDrFroutputs attachments)getpopr,rrKrdict setdefault)clsvaluess r$infer_defaultszRunTree.infer_defaultsXs ::f  %&**\*B*N --!' !5f!=v --!' !5d!;B!?v ::f  %&F6N v "(**X"6F;  & "(**Y"7F; zz+&"&F;  ::l # /&,\&:&=&=F? # v  7F4L V #v%%+L%9%B%Bz"%+D\z" T6$$Wb12 ::h  '!F8  ::f  %F6N ::i ( "F9  ::m $ ,$&F= ! r&c|jd}|r|jr|St|d|d}|dr|djdz|z|d<|S||d<|S)z#Ensure the dotted order of the run.rJr5r,r8.)r`strip_create_current_dotted_orderrJ)rdrecurrent_dotted_orders r$ensure_dotted_orderzRunTree.ensure_dotted_order|s &zz.9 $8$>$>$@M; < &,  , |$11C7:NN > "  &:F> " r&cP|jt|_|jS)zReturn the client.)rGr%selfs r$r\zRunTree.clients" >> !.0DN~~r&c|jSr1)rGrns r$r]zRunTree._clients~~r&c>|dk(r||_yt| ||S)zSet the _client specially.r]N)rGsuper __setattr__)ror*value __class__s r$rszRunTree.__setattr__s& 9 "DN7&tU3 3r&ct|tr|g}|jg|_|jj|y)zAdd tags to the run.N) isinstancer)rextend)rors r$add_tagszRunTree.add_tagss6 dC 6D 99 DI r&c|ji|_tt|jjdi}|j |y)zAdd metadata to the run.Nr)rDrrbrcupdate)ror metadata_s r$ add_metadatazRunTree.add_metadatas> :: DJtTZZ0;;JK "r&c`|ji|_|jj|yzUpsert the given outputs into the run. Args: outputs (Dict[str, Any]): A dictionary containing the outputs to be added. Returns: None N)r^r{)ror^s r$ add_outputszRunTree.add_outputss' << DL G$r&c`|ji|_|jj|yr)inputsr{)rors r$ add_inputszRunTree.add_inputss' ;; DK 6"r&cv|jg|_t|tr|jj|yt|trO|jjdt j tjj|dy|jj|y)a~Add an event to the list of events. Args: events (Union[ls_schemas.RunEvent, Sequence[ls_schemas.RunEvent], Sequence[dict], dict, str]): The event(s) to be added. It can be a single event, a sequence of events, a sequence of dictionaries, a dictionary, or a string. Returns: None Neventr*timemessage) rFrwrbappendr)rr2rr3 isoformatrx)rorFs r$ add_eventzRunTree.add_events* ;; DK fd # KK  v &  $ KK  #$LL6@@B%  KK  v &r&)r^errorend_timerFrc|xs#tjtj|_|/|j s||_n|j j ||||_||j|||j|yy)z/Set the end time of the run and all child runs.N) rr2rr3rr^r{rrr})ror^rrrFrs r$endz RunTree.ends{!>HLL$>  <<&  ##G,  DJ   NN6 "     h ' r&) run_idrYrr^rreference_example_idr5rrrDr_c ||xsd|i}tdid|dt|d|d|xsid|xsid|d|d|d | xs#tjtj d | d | xsid |d |j d|jd| d| xsid|j}|jj||S)z Add a child run to the run tree.r*r,rYrr^rr/rr5rrDr8r>rGrr_rLr") r(rrr2rr3r@rGrLr;r)ror*r/rrYrr^rrr5rrrDr_ serialized_runs r$ create_childzRunTree.create_childs$!2VTN   F# # rr^r;T)r6 exclude_nonerr^)rbrcopyr^)ro self_dicts r$_get_dicts_safezRunTree._get_dicts_safe0sdII7d  ;; ""&++"2"2"4Ih  << ##'<<#4#4#6Ii r&c|j}|jjdi||jdx}ro|Dcgc] }t |}}|j j dtjtjjt|d|s$|jD]}|jdyycc}w)z,Post the run tree to the API asynchronously.r_uploaded_attachmentrF)exclude_child_runsNr")rr\ create_runr`r)rFrrr2rr3rsetr;post)rorkwargsr_r*keys child_runs r$rz RunTree.post=s%%' (( **]3 3; 3*56$CI6D6 KK  1$LL6@@B"4y "!__ 9 %8 9"7sCc|js|j|jjDcic]\}}t |t s||}}} |rIt d|jDd}|r*|jDcic]\}}||dvr||}}}|jj|j|j|jr|jj!nd|j"r|j"j!nd|j$|j&|j(|j|j*|j,|j|j.|j0|ycc}}wcc}}w#t$r#}tjd|Yd}~d}~wwxYw)z5Patch the run tree to the API in a background thread.c3HK|]}|jddk(r|yw)r*rN)r`).0evs r$ z RunTree.patch..Ys)66&>-BBs "Nrz'Error filtering attachments to upload: )r*rrr^rr:rrrJrKrFrrDr_)rrr_itemsrwtuplenextrF Exceptionloggerwarningr\ update_runr*r,rrr^rr:rrJrKrrD)roavr_uploadedes r$patchz RunTree.patchNs}} HHJ!--335 QAu9MAqD   J"&++  %0$5$5$7# AqHY$771#K# 77)-4;;##%$+/<>Gcy)z"Wait for all _futures to complete.Nr"rns r$waitz RunTree.waitys r&c:|jj|S)zReturn the URL of the run.)r)r\ get_run_urlrns r$get_urlzRunTree.get_url}s{{&&4&00r&c Tt|i}tt|j|fi|S)zwCreate a new 'child' span from the provided dotted order. Returns: RunTree: The new span. )LANGSMITH_DOTTED_ORDERrr( from_headers)rdrJrheaderss r$from_dotted_orderzRunTree.from_dotted_orders2 #L G-S--g@@AAr&c  ddlm}m}ddlm}m}ddlm|%|t|tr t||nd}n t||}|jdx} rt| ||fr| jrtfd| jDdx} r| j jt#| jx} r| j$r| j$} | j&|d <| j(|d <| j*|d <| j,|d <| j.|d <t1t3| j4xsg|jdgz|d<| j6|d<|j9di} | j9di}|j;| j<nBt?| dr5| j| j@vr| j@| jd} ny| jB|d<| jD|d<tGjH| fi|Sy#t$r}td|d}~wwxYw)zCreate a new 'child' span from the provided runnable config. Requires langchain to be installed. Returns: Optional[RunTree]: The new span or None if no parent span information is found. r)AsyncCallbackManagerCallbackManager)RunnableConfig ensure_config)LangChainTracerz{RunTree.from_runnable_config requires langchain-core to be installed. You can install it with `pip install langchain-core`.N callbacksc3<K|]}t|s|ywr1)rw)rtrs r$rz/RunTree.from_runnable_config..sN1z!_/MQNsr/rr^r5rrr*rDr order_mapr\r>)% langchain_core.callbacks.managerrrlangchain_core.runnablesrr langchain_core.tracers.langchainr ImportErrorrwrbrr`r:rhandlersrun_mapr)rJr/rr^r5rsortedrrr*rcr{rhasattrrr\r>r(r)rdconfigrrrrrrconfig_cbtracerrrJextra_r|rs @r$from_runnable_configzRunTree.from_runnable_configs)   O H >#0:640H^V,dG>62G;;{+ +R +21EFG  N N ~~))#b.>.>*?@@@cFVFV"// %(\\z"#&::x $'KKy!'*~~|$%(\\z"!'CHH,SVZZPR=S8S(T!Uv!$v**7B7"--j"=   .-"2B2BfFVFV2V%//0@0@A!D %}}F8 %+%8%8F> ",,\DVD DU H  sH// I 8 II c |j}ttt|j t }|s>ttt |j t}|sy|jd}|j}t|}|dd}||d<|dd|d<||d<t|d k\r |d d|d <|j d xs#tjtj|d <|j d xsd|d <|j dxsd|d<t j#|} | j$s | j&r|j)di|d<|dj)di|dd<i| j$|dd} | |dd<t+t-| j&|j dgz} | |d<| j.r| j.|d<t1di|S)aCreate a new 'parent' span from the provided headers. Extracts parent span information from the headers and creates a new span. Metadata and tags are extracted from the baggage header. The dotted order and trace id are extracted from the trace header. Returns: Optional[RunTree]: The new span or None if no parent span information is found. NrrrrKrZr,rJr:r5r/r-r*parentrDrrr>r")rrr r)r`rbytesLANGSMITH_DOTTED_ORDER_BYTESdecoderi_parse_dotted_orderlenrr2rr3_Baggagerrrrcrrr>r() rdrr init_argslangsmith_tracelangsmith_trace_bytesparent_dotted_orderparsed_dotted_orderrKbaggagerrs r$rzRunTree.from_headerssKKM x}gkk:P.QR$(-I!J% !)3::7CO-33512EF&q)!, ( *-b1!4 $$7 .! " #q ()r) rKrJrrrDr`rr@ to_header)rorrs r$ to_headerszRunTree.to_headers sh ==373D3DG-. 0ZZ^^J3**  %..0 r&c pd|jd|jd|jd|jd S)z5Return a string representation of the RunTree object.z RunTree(id=z, name='z ', run_type='z', dotted_order='z'))r,r*r/rJrns r$__repr__zRunTree.__repr__sB$''(499+6'89J9J8K2 O r&)rerbreturnrb)rr)rr)rzUnion[Sequence[str], str]rNone)rDict[str, Any]rr)r^rrr)rrrr)rFzTUnion[ls_schemas.RunEvent, Sequence[ls_schemas.RunEvent], Sequence[dict], dict, str]rr) r^Optional[Dict]r Optional[str]rOptional[datetime]rFz'Optional[Sequence[ls_schemas.RunEvent]]rzOptional[Dict[str, Any]]rr)r-)r*r)r/rrzOptional[ID_TYPE]rYrrrr^rrrrrBr5rrrrrErDrr_z Optional[ls_schemas.Attachments]rr()T)rboolrr)rrrr))rJr)rrrr()rzOptional[dict]rrrr7)rz-Mapping[Union[str, bytes], Union[str, bytes]]rrrr7)rzDict[str, str])2rPrQrRrS__annotations__rrr,r/r5r8listr;r@rCrbrDrrFrGrJrKrLrVrrfrlpropertyr\r]rsryr}rrrrrrrrrr classmethodrrrrr __classcell__)rus@r$r(r(2s{; IU+B+'*Hc* 1STJT$)$$EJ!E %_-.!J GL#"'t-1( ( ( % ( 8 (+( (4 '' %)%)!%"&#/3)-'+$( $8<''' " ' # '' ''-'''%'"''6' !'R 9") V 1 B B B  B B>>>  >>@6$C6$OR6$ 6$6$p  r&r(cVeZdZdZ d ddZed dZed dZd dZy) rzBaggage header information.Nc>|xsi|_|xsg|_||_y)zInitialize the Baggage object.Nr)rorrr>s r$__init__z_Baggage.__init__#s"!B JB (r&c&|s|Si}g}d} |jdD]}|jdd\}}|tk(r3tjtj j |}T|tk(r/tj j |jd}|tk(stj j |} ||||S#t$r"}tjd|Yd}~1d}~wwxYw)z4Create a Baggage object from the given header value.N,=rzError parsing baggage header: r) splitLANGSMITH_METADATAjsonloadsurllibparseunquoteLANGSMITH_TAGSLANGSMITH_PROJECTrrr) rd header_valuerrr>itemkeyrtrs r$ from_headerz_Baggage.from_header.s5L  A$**3/ ?!ZZQ/ U,,#zz&,,*>*>u*EFHN*!<<//6<L ?H4lKK A NN;A3? @ @ AsB&C%8!C%% D.D  Dcd|vr|j|dSd|vr1|jtt|djdS|jdS)Nrsbaggager)r rrr)rdrs r$rz_Baggage.from_headersDs[  ??79#56 6 7 "??4wz/B#C#J#J7#ST T??4( (r&c.g}|jrLt|j}|jtdtj j ||jrRdj|j}|jtdtj j ||jrA|jtdtj j |jdj|S)z,Return the Baggage object as a header value.z metadata=rztags=zproject=) rrrLANGSMITH_PREFIXrrquoterjoinr>)rorserialized_metadataserialized_tagss r$rz_Baggage.to_headerMs =="-dmm"<  LL#$Ifll.@.@AT.U-VW  99!hhtyy1O LL#$E&,,*<*<_*M)NO     LL#$HV\\-?-?@Q@Q-R,ST xxr&)NNN)rzOptional[Dict[str, str]]rrEr>r)r rrr)rzMapping[Union[str, bytes], Any]rrr) rPrQrRrSrrr rrr"r&r$rr s`%.2$(&* )* )" )$ )LL*))r&rc |jd}|Dcgc])}tj|dddt|ddf+c}Scc}w)zParse the dotted order string.rhNi%Y%m%dT%H%M%S%fZ)rrstrptimer)rJpartsparts r$rrasX   s #E     4:'9 :Dcdr0sW$"  'SSS1+SS   8 $#'7&8$>?5</056CJJ"2!37;<     k j  k \>>B6"6,:66 i KsD.. D>=D>