&`iJddlZddlZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z m Z mZmZmZmZmZmZmZmZddlZddlZddlmZmZmZddlmZejeZ Gd d Z!d a"da#d e$fd Z%dZ&deefdZ'de defdZ(Gdde)Z*dee e+fd e fdZ,GddZ-eddd edfdZ.de+fdZ/d e d!e fd e+fd"Z0d e d!e fd e+fd#Z1d$ee+e d!e ffd%ee+e d!e fffd&Z2d$ee+e d!e ffd%ee+e d!e ffd e+fd'Z3d$ee+e d!e ffd%ee+e d!e ffd e+fd(Z4d)Z5d*Z6d+Z7d,Z8d-Z9dS).N)contextmanager)wraps) Parameter) ModuleType) AnyCallableDict GeneratorListMutableMappingOptionalSequenceUnioncast)is_class_methodis_function_or_methodis_static_method)get_runtime_contextcDeZdZdZhdZdZdZdZdZdZ dZ d Z d S) _OpenTelemetryProxya This proxy makes it possible for tracing to be disabled when opentelemetry is not installed on the cluster, but is installed locally. The check for `opentelemetry`'s existence must happen where the functions are executed because `opentelemetry` may be present where the functions are pickled. This can happen when `ray[full]` is installed locally by `ray` (no extra dependencies) is installed on the cluster. >traceContextcontext propagatecx|tjvrt|d|Std|)N_zAttribute does not exist: )rallowed_functionsgetattrAttributeError)selfnames s/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/util/tracing/tracing_helper.py __getattr__z_OpenTelemetryProxy.__getattr__/sF &8 8 8,74T,,.. . !Dd!D!DEE Ec,|dS)Nzopentelemetry.trace _try_importr s r"_tracez_OpenTelemetryProxy._trace5s 5666r$c,|dS)Nzopentelemetry.contextr&r(s r"_contextz_OpenTelemetryProxy._context8s 7888r$c,|dS)Nzopentelemetry.propagater&r(s r" _propagatez_OpenTelemetryProxy._propagate;s 9:::r$cJ|}|r |jjSdSN)r+rr)r rs r"_Contextz_OpenTelemetryProxy._Context>s'--//  ?* *4r$c||||dSr/)r)r+r-r0r(s r"try_allz_OpenTelemetryProxy.try_allEs=    r$c tj|S#t$r!trtdYdSwxYw)NaInstall OpenTelemetry with 'pip install opentelemetry-api==1.34.1 opentelemetry-sdk==1.34.1 opentelemetry-exporter-otlp==1.34.1' to enable tracing. See the Ray documentation for details: https://docs.ray.io/en/latest/ray-observability/user-guides/ray-tracing.html#installation) importlib import_module ImportError_is_tracing_enabled)r modules r"r'z_OpenTelemetryProxy._try_importKsh *622 2   "$$ !p    s'AAN) __name__ __module__ __qualname____doc__rr#r)r+r-r0r2r'r$r"rr"sEDDFFF 777999;;;     r$rFreturnctS)zgChecks environment variable feature flag to see if tracing is turned on. Tracing is off by default.)_global_is_tracing_enabledr=r$r"r7r7\s  &%r$cXdatatdS)NT)r@r_opentelemetryr2r=r$r"_enable_tracingrCbs+!%(**Nr$ params_listct|D]D\}}|jtjkr*|||nE|S)z]Given a list of Parameters, if a kwargs Parameter exists, move it to the end of the list.) enumeratekindr VAR_KEYWORDappendpop)rDiparams r"_sort_params_listrMisak**5 :. . .   {q11 2 2 2 E / r$function new_paramctj|}t|j}t fd|Dr|St |gz}||}|S)z/Add additional Parameter to function signature.c38K|]}|jjkVdSr/)r!).0rLrOs r" z*_add_param_to_signature..xs, G GE5: ' G G G G G Gr$) parameters)inspect signaturelistrTvaluesanyrMreplace)rNrOold_sigold_sig_list_repr new_paramsnew_sigs ` r"_add_param_to_signaturer_ss))GW/668899 G G G G5F G G GGG"#4 {#BCCJooo44G Nr$ceZdZdS)_ImportFromStringErrorN)r9r:r;r=r$r"rarasDr$ra import_strc.t|ts|S|d\}}}|r|s%d}t|| t j|}nE#t$r8}|j|kr|dd}t||d}~wwxYw|} | dD]}t||}n4#t$r'd}t||| wxYw|S) zRGiven a string that is in format ":", import the attribute.:zEImport string "{import_str}" must be in format":".)rbNz'Could not import module "{module_str}".) module_str.z;Attribute "{attrs_str}" not found in module "{module_str}".) attrs_strre) isinstancestr partitionraformatr4r5r6r!splitrr) rbrerrgmessager8excinstanceattr_strs r"_import_from_stringrqsg j# & &)33C88J9 LYL V %W^^z^%J%JKKKL(44 LLL 8z ! !4 ;$W^^z^%J%JKKK L H !,, 3 3Hx22HH 3    O$ NNY:N F F    Os$A00 B2:3B--B28(C!!1DcJeZdZdeeeffdZdeeefddfdZdS)_DictPropagatorr>cHi}tj||S)z*Inject trace context into otel propagator.)rBrinject context_dicts r"inject_current_contextz&_DictPropagator.inject_current_contexts$')  '' 555r$rw_opentelemetry.Contextcpttjtj|S)z,Given a trace context, extract as a Context.)rrBrrextractrvs r"r{z_DictPropagator.extracts-  "N$<$D$D\$R$R   r$N)r9r:r;r rrxr{r=r$r"rsrssbDcN  d38n 1I      r$rsparent_contextry)NNNc#K||}nt}tj|} dVtj|dS#tj|wxYw)z(Uses the Ray trace context for the span.N)rBrrattachdetach)r| new_contexttokens r" _use_contextrs !$ $,,..  " ) )+ 6 6E- %%e,,,,,%%e,,,,s A&&!B function_namect}d|ttj||d}t jjj j t jjj kr| }|r||d<tt jjj dd}|r||d<|S)zXGet the Attributes of the function that will be reported as attributes in the trace.rN) ray.remote ray.functionray.pid ray.job_id ray.node_idz ray.task_id worker_idN ray.worker_id)rriosgetpid get_job_id get_node_idray_privateworker global_workermode WORKER_MODE get_task_idrhex)rruntime_context span_argstask_idrs r"_function_hydrate_span_argsrs*++O!%ry{{##%0022&2244 I |(-1D1PPP!--//  /'.Im $ +9;MMI5%.]]__ /" r$func.c |dS)z>Returns the function span name that has span kind of producer. ray.remoter=rs r"_function_span_producer_namers   r$c |dS)z>Returns the function span name that has span kind of consumer. ray.remote_workerr=rs r"_function_span_consumer_namers & & &&r$class_methodc4t|r|j}t|r|j}t}d|||d|tt j||d}tj j j j tj j j kr|}|r||d<ttj j j dd}|r||d<|S)zUGet the Attributes of the actor that will be reported as attributes in the trace.actorrf)rzray.actor_classzray.actor_methodrrrr ray.actor_idrNr)callabler9rrirrrrrrrrrr get_actor_idrr)rrrractor_idrs r"_actor_hydrate_span_argsrs !!)++O!"!,,F,,ry{{##%0022&2244I |(-1D1PPP"//11  1(0In % +9;MMI5%.]]__ /" r$ct|ts|j}t|ts|j}|d|dS)z;Returns the actor span name that has span kind of producer.rfrrhrir9rrs r"_actor_span_producer_namer sM fc " "! fc " "! * *v * * **r$ct|ts|j}t|ts|j}|d|dS)z;Returns the actor span name that has span kind of consumer.rfrrrs r"_actor_span_consumer_namersM fc " "! fc " "! 1 1v 1 1 11r$c t ddtdtttfdtdtdtf fd }|S) zfTrace the execution of a remote task. Inject the current span context into kwargs for propagation.Nargskwargs_args_kwargsr>ctr|jr|d|vsJ|||g|Ri|Sd|vsJtjt }|t|jtjj j t|j5t |d<|||g|Ri|cdddS#1swxYwYdS)N_ray_trace_ctxrG attributes)r7_is_cross_languagerBr get_tracerr9start_as_current_spanr_function_nameSpanKindPRODUCERrrsrx)r rrrrtracerrs r"_invocation_remote_spanz9_tracing_task_invocation.._invocation_remote_span)sv#$$ A(? A!'v55556$f@u@@@@@ @v----%00::  ) ) ()< = =%.7243FGG*   A A (7'M'M'O'OF# $6$f@u@@@@@ A A A A A A A A A A A A A A A A A As'*CC"%C"NN)rrr )rrs` r"_tracing_task_invocationr%s 6]]+/AAAsCx(A A  A AAAAA]A2 #"r$c HtsSttjdtjjd_t dddtdttttfdtdtffd}|S) zWrap the function argument passed to RemoteFunction's __init__ so that future execution of that function will include tracing. Use the provided trace context from kwargs. rNdefaultrrrr>c||i|Stjt}jdzjz}t t |5|t|tjj j t|5|i|cdddcdddS#1swxYwYddddS#1swxYwYdS)Nrfr) rBrrr9r:rrsr{rrrCONSUMERr)rrrrrrNs r"_function_with_tracingz=_inject_tracing_into_function.._function_with_tracingUs  !8T,V,, ,%00:: +c1H4EE   # #N 3 3   - -  ' ' ( 7 7%.72=AA (   - -8T,V,, - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -s7)AC)0C8 C)C C)C C))C-0C-) r7r_rUr KEYWORD_ONLY __signature__rrr r ri)rNrs` r"_inject_tracing_into_functionrFs  4 g/._invocation_actor_class_remote_spanrs >F#$$ A#611116$f@u@@@@@ @*5  w....%00::  ) )*:{CC%.7/ KHH*  '6'M'M'O'OF# $VD$BBBB'BBF   ~v/C/G/G/I/I J J J                  s(ADDD)rtuplerr )rrs` r"_tracing_actor_creationrns| 6]]GG+/sCx(   ]B /.r$c t ddttdtttfdtdtdtf fd }|S) z(Trace the invocation of an actor method.Nrrrrr>cdtr |jjr|d|vsJ |||g|Ri|S|jjj}|j}d|vsJt jt}| t||t jj j t||5}t|d<|d|jj |||g|Ri|cdddS#1swxYwYdS)Nrrr)r7_actor_ray_is_cross_language'_ray_actor_creation_function_descriptorr _method_namerBrrr9rrrrrrsrxrrr) r rrrrrrrrrs r" _start_spanz5_tracing_actor_method_invocation.._start_spans#$$ A (J A!'v55556$f@u@@@@@ @[HS ' w....%00::  ) )*:{CC%.7/ KHH*  A'6'M'M'O'OF# $   ~t{/H/L/L/N/N O O O6$f@u@@@@@ A A A A A A A A A A A A A A A A A Asc dddtdtdttttfdtdtf fd}|S)Nrr rrrr>c  tr| |g|Ri|Stjt}t t |5|t|j jtjj j t|j j5|g|Ri|cdddcdddS#1swxYwYddddS#1swxYwYdSzr Wrap the user's function with a function that will extract the trace context Nrr7rBrrr9rrsr{rr __class__rrrr rrrrrs r" _resume_spanzF_inject_tracing_into_class..span_wrapper.._resume_spans'(( 7N,Bvd6U666g6662@2F2Q2Q33F ''77 7 7++)$.*A6JJ#)2;3DN4KVTT, 7 7vd6U666g66 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7s7)AD C+ D+C/ /D2C/ 3DD Drr r rirrs` r" span_wrapperz0_inject_tracing_into_class..span_wrapperss8< 7 7 7 7 7%T#s(^4 7  7   7 7 7 7 7 7:r$c dddtdtdttttfdtdtf fd}|S)Nrr rrrr>c PKtr||g|Ri|d{VStjt}t t |5|t|j jjtjj j t|j jj5|g|Ri|d{VcdddcdddS#1swxYwYddddS#1swxYwYdSrrrs r"rzL_inject_tracing_into_class..async_span_wrapper.._resume_spans'(( =N,B#VD<5<<>F''77 = =++)$.*A6?SS#)2;3N+V_, = =$VD<5<<.async_span_wrapperss8< = = = = =%T#s(^4 =  =   = = = = = =:r$__del__rNr)rrrU getmembersrrrisgeneratorfunctionisasyncgenfunctionr_rrriscoroutinefunctionrsetattr)_clsrrmethodsr!rwrapped_methods r"_inject_tracing_into_classrsXc3h/C@8CH#5#@ '<==G&,&, f D$ ' ' ?6+B+B    &v . . '2LV2T2T   9    7    '"3"@$       &v . . A*U6]]+=+=f+E+EFFNN*U6]]<<+?+?@@NdN++++ Kr$):r4rUloggingr contextlibr functoolsrrtypesrtypingrrr r r r r rrrrray._private.workerray._private.inspect_utilrrrray.runtime_contextr getLoggerr9loggerrr@rBboolr7rCrMr_ Exceptionrarirqrsrrrrrrrrrrrrr=r$r"r sr %%%%%%                          433333  8 $ $33333333l#&T&&&& 4 ? h 9         Y   E*c/$:zD          -, -  - - - -s4 xS'9 c    'xS'9'c'''' ! #xS)) *! #xS)) *!!!!H + #xS)) * + #xS)) * +  + + + + 2 #xS)) * 2 #xS)) * 2  2 2 2 2###B%"%"%"P&/&/&/R"""Jmmmmmr$