o 3Ih@sddlmZddlZddlZddlmZddlmZddlm Z ddl m Z m Z ddl mZddlmZeeZd'd d ZeZd(ddZd)d*ddZ d)d+ddZd,ddZd-ddZd.d"d#Zed$Zed%Zgd&ZdS)/) annotationsN)Token)environ)uuid4)Context_RuntimeContext)OTEL_PYTHON_CONTEXT) entry_pointsreturnrc Cshd}tt|}ztttd|dWSty3t d||tttd|dYSw)zWInitialize the RuntimeContext Returns: An instance of RuntimeContext. contextvars_contextopentelemetry_context)groupnamez*Failed to load context: %s, fallback to %s) rgetrnextiterr load Exceptionlogger exception)default_contextconfigured_contextr^/home/air/sanwanet/gpt-api/venv/lib/python3.10/site-packages/opentelemetry/context/__init__.py_load_runtime_contexts<  rkeynamestrcCs|dttS)aeTo allow cross-cutting concern to control access to their local state, the RuntimeContext API provides a function which takes a keyname as input, and returns a unique key. Args: keyname: The key name is for debugging purposes and is not required to be unique. Returns: A unique string representing the newly created key. -)rr)rrrr create_keyIs rkeycontexttyping.Optional[Context]'object'cCs|dur ||St|S)avTo access the local state of a concern, the RuntimeContext API provides a function which takes a context and a key as input, and returns a value. Args: key: The key of the value to retrieve. context: The context from which to retrieve the value, if None, the current context is used. Returns: The value associated with the key. N)r get_current)rr rrr get_valueUs r$valuercCs&|durt}|}|||<t|S)aTo record the local state of a cross-cutting concern, the RuntimeContext API provides a function which takes a context, a key, and a value as input, and returns an updated context which contains the new value. Args: key: The key of the entry to set. value: The value of the entry to set. context: The context to copy, if None, the current context is used. Returns: A new `Context` containing the value set. N)r#copyr)rr%r new_valuesrrr set_valueds r(cCstS)zTo access the context associated with program execution, the Context API provides a function which takes no arguments and returns a Context. Returns: The current `Context` object. )_RUNTIME_CONTEXTr#rrrrr#{sr#Token[Context]cCs t|S)aAssociates a Context with the caller's current execution unit. Returns a token that can be used to restore the previous Context. Args: context: The Context to set as current. Returns: A token that can be used with `detach` to reset the context. )r)attach)r rrrr+s r+tokenNonecCs0zt|WdStytdYdSw)zResets the Context associated with the caller's current execution unit to the value it had before attaching a specified Context. Args: token: The Token that was returned by a previous call to attach a Context. zFailed to detach contextN)r)detachrrr)r,rrrr.s  r.suppress_instrumentationsuppress_http_instrumentation)rr+rr.r#r$r()r r)rrr r)N)rrr r!r r")rrr%r"r r!r r)r r)r rr r*)r,r*r r-) __future__rloggingtyping contextvarsrosruuidropentelemetry.context.contextrr#opentelemetry.environment_variablesr&opentelemetry.util._importlib_metadatar getLogger__name__rrr)rr$r(r#r+r._SUPPRESS_INSTRUMENTATION_KEY"_SUPPRESS_HTTP_INSTRUMENTATION_KEY__all__rrrrs0        '