&`iR"bddlZddlZddlmZddlmZmZddlmZm Z m Z ddl Z GddZ dej vZe Zedd d d d d dddddddd ZedGddZedGddZddZde edefdZdS)N) defaultdict)asdict dataclass)DictListUnionceZdZdZdZdZdS) _NullLogSpanz,A log span context manager that does nothingcdSN)selfs j/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/_private/profiling.py __enter__z_NullLogSpan.__enter__  cdSr r )rtypevaluetbs r__exit__z_NullLogSpan.__exit__rrN)__name__ __module__ __qualname____doc__rrr rrr r s866        rr RAY_PROFILINGcdS)N generic_workr r rrrsNrcq_build_abandoned rail_response rail_loadrail_animation rail_idledetailed_memory_dumpgoodterriblevsync_highlight_colorbackground_memory_dump) worker_idletaskztask:deserialize_argumentsz task:executeztask:store_outputswait_for_functionzray.getzray.putzray.wait submit_taskfetch_and_run_functionregister_remote_functionT)initceZdZUeed<eed<eed<eed<eed<eed<eed<eeeeeffed<d Zeed <d S) ChromeTracingCompleteEventcatnamepidtidtsdurcnameargsXphN) rrrstr__annotations__intrrr<r rrr2r2,s~ HHH III HHH HHH GGG HHH JJJ sE#s(O# $$$$BMMMMMrr2cZeZdZUeed<eeefed<eed<dZeed<dZeed<dS)ChromeTracingMetadataEventr4r:r5Nr6Mr<) rrrr=r>rr?r6r<r rrrArAFsR III sCx. HHHCOOOBMMMMMrrActstStjjj}|jtjjjkrtS|j | d|S)aFProfile a span of time so that it appears in the timeline visualization. Note that this only works in the raylet code path. This function can be used as follows (both on the driver or within a task). .. testcode:: import ray._private.profiling as profiling with profiling.profile("custom event", extra_data={'key': 'val'}): # Do some computation here. x = 1 * 2 Optionally, a dictionary can be passed as the "extra_data" argument, and it can have keys "name" and "cname" if you want to override the default timeline display text and box color. Other values will appear at the bottom of the chrome tracing GUI when you click on the box corresponding to this profile span. Args: event_type: A string describing the type of the event. extra_data: This must be a dictionary mapping strings to strings. This data will be added to the json objects that are used to populate the timeline, so if you want to set a particular color, you can simply set the "cname" attribute to an appropriate color. Similarly, if you set the "name" attribute, then that will set the text displayed on the box in the timeline. Returns: An object that can profile a span of time via a "with" statement. ascii) PROFILING_ENABLED NULL_LOG_SPANray_privateworker global_workermode LOCAL_MODE core_worker profile_eventencode) event_type extra_datarIs rprofilerRTsb@  \ .F {cl)444   + +J,=,=g,F,F S SSrtasksreturnctg}i}d}i}d}|D]j}|dg}|rN|d}|d} |d} | dz|dz} | dvrG| D]} | d } |d | d <|d | d <|d | d <|d | d <|d| d<| d}||vr |||<|dz }||| f|vr||||| f<|dz }t| d}|}d| vrt| d d}d| vr| d}t|||||||| f| ddz| ddz| ddzz || }|t |!l|D]<\}}|t t d|dd|i=|D]G\}}|t t dd||dd|diHtj|S)zGenerate a chrome/perfetto tracing dump using task events. Args: tasks: List of tasks generated by a state API list_tasks(detail=True). Returns: Json serialized dump to create a chrome/perfetto tracing. rprofiling_datanode_ip_addresseventscomponent_type: component_id)rIdriverrQtask_idjob_idattempt_numberfunc_or_class_nameactor_id event_namer9r4 start_timeg@@end_time)r3r4r5r6r7r8r9r: process_namezNode )r4r5r: thread_namerB)r4r<r6r5r:) get_default_color_mappingr2appendritemsrAjsondumps)rS all_events node_to_indexnode_idxworker_to_index worker_idxr+rVrWcomponent_eventsrYr[eventrQrbr9r4 new_eventnodeirIs rchrome_tracing_dumprx|s/JMHOJ7575"2B77 5 5,->?O-h7 +,<=N)C/.2PPL%999), 5, 5"<0 (,Y 9%'+H~ 8$/34D/E +,378L3M /0)-j)9 :&"<0 #-775=M/2MH"/2 ))) #$&7F!OJ/u\/BC!j((253Fw3OPEZ''%f-D6"%o6')G(VW\*S0z*S0U<5H35NO#    !!&"3"34444 &&((   a *' .$..1      %**,,    *&q  &),     :j ! !!rr )rlos collectionsr dataclassesrrtypingrrrrGr environrErFrir2rArRdictr=rxr rrrs ######))))))))$$$$$$$$$$         $rz1  %+&1()3+/"8$:  & 2         %T%T%T%TPt" :t"t"t"t"t"t"t"r