.hxddlmZddlZddlZddlmZddlmZddlmZddl m Z dd l m Z dd l mZmZmZmZdd lmZmZmZGd d e ZGddZeZdS)) annotationsN)Any)logger)util)TracingProcessor)Scope)NoOpSpanSpanSpanImpl TSpanData) NoOpTraceTrace TraceImplcVeZdZdZdZddZddZdd ZddZddZ ddZ ddZ dZ dS) SynchronousMultiTracingProcessorzV Forwards all calls to a list of TracingProcessors, in order of registration. cDd|_tj|_dS)N) _processors threadingLock_lockselfs Q/var/www/FlaskApp/flask-venv/lib/python3.11/site-packages/agents/tracing/setup.py__init__z)SynchronousMultiTracingProcessor.__init__s9;^%% tracing_processorr ch|j5|xj|fz c_ddddS#1swxYwYdSzj Add a processor to the list of processors. Each processor will receive all traces/spans. N)rr)rrs radd_tracing_processorz6SynchronousMultiTracingProcessor.add_tracing_processorsZ 5 5   !2 4 4   5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5s '++ processorslist[TracingProcessor]cn|j5t||_ddddS#1swxYwYdSz_ Set the list of processors. This will replace the current list of processors. N)rtuplerrr#s rset_processorsz/SynchronousMultiTracingProcessor.set_processors sZ 1 1$Z00D  1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1s *..tracerreturnNonecD|jD]}||dS)z1 Called when a trace is started. N)ron_trace_startrr* processors rr.z/SynchronousMultiTracingProcessor.on_trace_start's7) , ,I  $ $U + + + + , ,rcD|jD]}||dS)z2 Called when a trace is finished. N)r on_trace_endr/s rr2z-SynchronousMultiTracingProcessor.on_trace_end.s7) * *I  " "5 ) ) ) ) * *rspan Span[Any]cD|jD]}||dS)z0 Called when a span is started. N)r on_span_startrr3r0s rr6z.SynchronousMultiTracingProcessor.on_span_start5s7) * *I  # #D ) ) ) ) * *rcD|jD]}||dS)z1 Called when a span is finished. N)r on_span_endr7s rr9z,SynchronousMultiTracingProcessor.on_span_end<s7) ( (I  ! !$ ' ' ' ' ( (rcp|jD]-}tjd||.dS)z4 Called when the application stops. zShutting down trace processor N)rrdebugshutdownrr0s rr<z)SynchronousMultiTracingProcessor.shutdownCsN) ! !I LE)EE F F F     ! !rcB|jD]}|dS)z> Force the processors to flush their buffers. N)r force_flushr=s rr?z,SynchronousMultiTracingProcessor.force_flushKs5) $ $I  ! ! # # # # $ $rN)rr r#r$)r*rr+r,)r3r4r+r,r+r,) __name__ __module__ __qualname____doc__rr"r)r.r2r6r9r<r?rrrrrs&&& 55551111,,,,********((((!!!!$$$$$rrcfeZdZdZd$dZd%dZd&d Zd'd Zd(dZ d)d*dZ d+d,d"Z d-d#Z dS). TraceProviderct|_tjdddv|_dS)NOPENAI_AGENTS_DISABLE_TRACINGfalse)true1)r_multi_processorosenvirongetlower _disabledrs rrzTraceProvider.__init__TsC @ B B(GQQWWYY^  rr0r c:|j|dSr!)rMr"r=s rregister_processorz TraceProvider.register_processor[s! 33I>>>>>rr#r$c:|j|dSr&)rMr)r(s rr)zTraceProvider.set_processorsas! ,,Z88888rr+ Trace | Nonec(tjS)z= Returns the currently active trace, if any. )r get_current_tracers rrXzTraceProvider.get_current_tracegs&(((rSpan[Any] | Nonec(tjS)z< Returns the currently active span, if any. )r get_current_spanrs rr[zTraceProvider.get_current_spanms%'''rdisabledboolr,c||_dS)z2 Set whether tracing is disabled. N)rR)rr\s r set_disabledzTraceProvider.set_disabledss"rNFnamestrtrace_id str | Nonegroup_idmetadatadict[str, Any] | Nonerc|js|r%tjd|tS|pt j}tjd|d|t |||||jS)z% Create a new trace. z(Tracing is disabled. Not creating trace zCreating trace with id )r`rbrdrer0)rRrr;rr gen_trace_idrrM)rr`rbrdrer\s r create_tracezTraceProvider.create_traceys > X  LJDJJ K K K;; 2t022 @t@@h@@AAA+     r span_datarspan_idparentTrace | Span[Any] | NoneSpan[TSpanData]c|js|r&tjd|t|S|st j}t j}|#tjdt|St|tst|tr*tjd|d|dt|S|r|j nd}|j }nt|trFt|tr'tjd|dt|S|j }d}n_t|trJt|tr'tjd|dt|S|j }|j }tjd|d|t||||j| S) z$ Create a new span. z'Tracing is disabled. Not creating span NzSNo active trace. Make sure to start a trace with `trace()` firstReturning NoOpSpan.zParent z or z is no-op, returning NoOpSpanzCreating span rh)rbrl parent_idr0rk)rRrr;r r r[rXerror isinstancerrlrbrr r rM) rrkrlrmr\ current_span current_tracerqrbs r create_spanzTraceProvider.create_spans  > 'X ' LN9NN O O OI&& & ' 133L!355M$ * ***M955 +LRZ9[9[ + \l\\ \\\ ***0<F ,,$I$-HH  & & '&),, + LvLLLMMM ***HII  % % '&(++ + LvLLLMMM ***IH CiCC'CCDDD+     rc|jrdS tjd|jdS#t $r"}tjd|Yd}~dSd}~wwxYw)NzShutting down trace providerz$Error shutting down trace provider: )rRrr;rMr< Exceptionrr)res rr<zTraceProvider.shutdowns >  F E L7 8 8 8  ! * * , , , , , E E E LCCC D D D D D D D D D Es-: A&A!!A&)r0r r@)r+rV)r+rY)r\r]r+r,)NNNF) r`rarbrcrdrcrerfr\r]r+r)NNF) rkrrlrcrmrnr\r]r+rorA) rBrCrDrrTr)rXr[r_rjrvr<rrrrGrGSs   ???? 9999 )))) (((( """" $#*.      <#+/ 5 5 5 5 5 nEEEEEErrG) __future__rrNrtypingrrrprocessor_interfacer scoper spansr r r rtracesrrrrrGGLOBAL_TRACE_PROVIDERrrrrsI"""""" 111111666666666666//////////A$A$A$A$A$'7A$A$A$H@E@E@E@E@E@E@E@EF& r