o h@sUddlZddlmZmZddlmZer!ddlmZddlmZej ddd Z ej d e d <ej d dd Z ej d e d<GdddZ dS)N) TYPE_CHECKINGAny)logger)Span)Trace current_span)defaultSpan[Any] | None _current_span current_trace Trace | None_current_tracec@sdeZdZdZedddZeddd Zedd d ZedddZedddZ edddZ d S)Scopez< Manages the current span and trace in the context. returnr cCtSN)r getclsrU/var/www/html/openai_agents/venv/lib/python3.10/site-packages/agents/tracing/scope.pyget_current_spanzScope.get_current_spanspan#contextvars.Token[Span[Any] | None]cCs t|Sr)r set)rrrrrset_current_spans zScope.set_current_spantokenNcCst|dSr)r resetrrrrrreset_current_span!szScope.reset_current_spanrcCrr)rrrrrrget_current_trace%rzScope.get_current_tracetracecontextvars.Token[Trace | None]cCs$td|r|jndt|S)NzSetting current trace: )rdebugtrace_idrr)rr$rrrset_current_trace)s zScope.set_current_tracecCstdt|dS)NzResetting current trace)rr&rr r!rrrreset_current_trace.s zScope.reset_current_trace)rr )rr rr)rrrN)rr)r$rrr%)rr%rN) __name__ __module__ __qualname____doc__ classmethodrrr"r#r(r)rrrrrs     r) contextvarstypingrrrspansrtracesr ContextVarr __annotations__rrrrrrs