
    Cj                         % S SK r S SKJrJr  SSKJr  \(       a  SSKJr  SSKJr  \ R                  " SSS	9r
\ R                  S
   \S'   \ R                  " SSS	9r\ R                  S   \S'    " S S5      rg)    N)TYPE_CHECKINGAny   )logger   )Span)Tracecurrent_span)defaultSpan[Any] | None_current_spancurrent_traceTrace | None_current_tracec                       \ rS rSrSr\SS j5       r\SS j5       r\SS j5       r\SS j5       r	\SS j5       r
\SS	 j5       rS
rg)Scope   z4
Manages the current span and trace in the context.
c                 *    [         R                  5       $ N)r   getclss    f/home/rurouni/.local/share/pipx/venvs/strix-agent/lib/python3.13/site-packages/agents/tracing/scope.pyget_current_spanScope.get_current_span   s      ""    c                 ,    [         R                  U5      $ r   )r   set)r   spans     r   set_current_spanScope.set_current_span   s      &&r   Nc                 .    [         R                  U5        g r   )r   resetr   tokens     r   reset_current_spanScope.reset_current_span!   s    E"r   c                 *    [         R                  5       $ r   )r   r   r   s    r   get_current_traceScope.get_current_trace%   s    !!##r   c                     [         R                  " SU(       a  UR                  OS  35        [        R	                  U5      $ )NzSetting current trace: )r   debugtrace_idr   r   )r   traces     r   set_current_traceScope.set_current_trace)   s0    .u~~D.QRS!!%((r   c                 Z    [         R                  " S5        [        R                  U5        g )NzResetting current trace)r   r,   r   r#   r$   s     r   reset_current_traceScope.reset_current_trace.   s    ./U#r    )returnr   )r   r   r5   #contextvars.Token[Span[Any] | None])r%   r6   r5   N)r5   r   )r.   r   r5   contextvars.Token[Trace | None])r%   r7   r5   N)__name__
__module____qualname____firstlineno____doc__classmethodr   r    r&   r)   r/   r2   __static_attributes__r4   r   r   r   r      s     # # ' ' # # $ $ ) ) $ $r   r   )contextvarstypingr   r   r   spansr   tracesr	   
ContextVarr   __annotations__r   r   r4   r   r   <module>rE      ss     % <G<R<RD={%%&89  :E9O9OT:&&~6 
$ $r   