Ë
    /UÁi¥  ã                   óò   — U d dl Z d dlmZmZ ddlmZ erddlmZ ddlmZ  e j                  dd¬	«      Z
e j                  d
   ed<    e j                  dd¬	«      Ze j                  d   ed<    G d„ d«      Zy)é    N)ÚTYPE_CHECKINGÚAnyé   )Úloggeré   )ÚSpan)ÚTraceÚcurrent_span)ÚdefaultúSpan[Any] | NoneÚ_current_spanÚcurrent_traceúTrace | NoneÚ_current_tracec                   ó|   — e Zd ZdZed	d„«       Zed
d„«       Zedd„«       Zedd„«       Zedd„«       Z	edd„«       Z
y)ÚScopez<
    Manages the current span and trace in the context.
    c                 ó*   — t         j                  «       S ©N)r   Úget©Úclss    úp/home/jay/workspace/tools/ai-image-gen/jaaz-app/server/venv/lib/python3.12/site-packages/agents/tracing/scope.pyÚget_current_spanzScope.get_current_span   s   € ä× Ñ Ó"Ð"ó    c                 ó,   — t         j                  |«      S r   )r   Úset)r   Úspans     r   Úset_current_spanzScope.set_current_span   s   € ä× Ñ  Ó&Ð&r   Nc                 ó.   — t         j                  |«       y r   )r   Úreset©r   Útokens     r   Úreset_current_spanzScope.reset_current_span!   s   € ä×Ñ˜EÕ"r   c                 ó*   — t         j                  «       S r   )r   r   r   s    r   Úget_current_tracezScope.get_current_trace%   s   € ä×!Ñ!Ó#Ð#r   c                 óx   — t        j                  d|r|j                  nd › «       t        j	                  |«      S )NzSetting current trace: )r   ÚdebugÚtrace_idr   r   )r   Útraces     r   Úset_current_tracezScope.set_current_trace)   s0   € ä‰Ð.Á¨u¯~ª~ÈDÐ.QÐRÔSÜ×!Ñ! %Ó(Ð(r   c                 óX   — t        j                  d«       t        j                  |«       y )NzResetting current trace)r   r'   r   r    r!   s     r   Úreset_current_tracezScope.reset_current_trace.   s   € ä‰Ð.Ô/Ü×Ñ˜UÕ#r   )Úreturnr   )r   r   r-   ú#contextvars.Token[Span[Any] | None])r"   r.   r-   N)r-   r   )r)   r   r-   úcontextvars.Token[Trace | None])r"   r/   r-   N)Ú__name__Ú
__module__Ú__qualname__Ú__doc__Úclassmethodr   r   r#   r%   r*   r,   © r   r   r   r      s   „ ñð ò#ó ð#ð ò'ó ð'ð ò#ó ð#ð ò$ó ð$ð ò)ó ð)ð ò$ó ñ$r   r   )ÚcontextvarsÚtypingr   r   r   Úspansr   Útracesr	   Ú
ContextVarr   Ú__annotations__r   r   r5   r   r   ú<module>r<      sw   ðä ß %å áÝÝà<R¸K×<RÑ<RØ˜Dô=€ˆ{×%Ñ%Ð&8Ñ9ó ð :P¸×9OÑ9OØ˜Tô:€×&Ñ& ~Ñ6ó ÷
$ò $r   