o g@sddlmZddlmZmZmZddlmZddlm Z m Z dZ edZ dde deed efd d Z ddeed e fd d ZdS))Optional) create_key get_value set_value)Context) INVALID_SPANSpanz current-spanNspancontextreturncCstt||d}|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 ctxr_/var/www/visachat/venv/lib/python3.10/site-packages/opentelemetry/trace/propagation/__init__.pyset_span_in_contexts rcCs&tt|d}|dust|tstS|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 N)rr isinstancerr)r r rrrget_current_span&s r)N)typingropentelemetry.contextrrropentelemetry.context.contextropentelemetry.trace.spanrrSPAN_KEYr rrrrrrs