
    g                         d dl mZ d dlmZmZmZ d dlmZ d dlm	Z	m
Z
 dZ ed      Z	 dde
dee   d	efd
Zddee   d	e
fdZy)    )Optional)
create_key	get_value	set_value)Context)INVALID_SPANSpanzcurrent-spanNspancontextreturnc                 *    t        t        | |      }|S )zSet the span in the given context.

    Args:
        span: The Span to set.
        context: a Context object. if one is not passed, the
            default current context is used instead.
    r   )r   	_SPAN_KEY)r
   r   ctxs      ]/var/www/openai/venv/lib/python3.12/site-packages/opentelemetry/trace/propagation/__init__.pyset_span_in_contextr      s     ItW
5CJ    c                 X    t        t        |       }|t        |t              st        S |S )zRetrieve the current span.

    Args:
        context: A Context object. If one is not passed, the
            default current context is used instead.

    Returns:
        The Span set in the context if it exists. INVALID_SPAN otherwise.
    r   )r   r   
isinstancer	   r   )r   r
   s     r   get_current_spanr   &   s)     Y0D|:dD1Kr   )N)typingr   opentelemetry.contextr   r   r   opentelemetry.context.contextr   opentelemetry.trace.spanr   r	   SPAN_KEYr   r   r    r   r   <module>r      sd     B B 1 7~&	 .2
!'*hw/ 4 r   