update tracer

Author: mabdinur

ddtrace/__init__.py

@@ -29,7 +29,7 @@
 from .internal.utils.deprecations import DDTraceDeprecationWarning  # noqa: E402
 from ddtrace._trace.pin import Pin  # noqa: E402
 from ddtrace._trace.span import Span  # noqa: E402
-from ddtrace._trace.tracer import Tracer  # noqa: E402
+from ddtrace.trace import Tracer  # noqa: E402
 from ddtrace.vendor import debtcollector
 from .version import get_version  # noqa: E402
 

ddtrace/appsec/_capabilities.py

@@ -59,7 +59,7 @@ def _asm_feature_is_required() -> bool:
     return (_FEATURE_REQUIRED & flags) != 0
 
 
-def _rc_capabilities(test_tracer: Optional[ddtrace._trace.tracer.Tracer] = None) -> Flags:
+def _rc_capabilities(test_tracer: Optional[ddtrace.trace.Tracer] = None) -> Flags:
     tracer = ddtrace.tracer if test_tracer is None else test_tracer
     value = Flags(0)
     if ddtrace.config._remote_config_enabled:
@@ -74,7 +74,7 @@ def _rc_capabilities(test_tracer: Optional[ddtrace._trace.tracer.Tracer] = None)
     return value
 
 
-def _appsec_rc_capabilities(test_tracer: Optional[ddtrace._trace.tracer.Tracer] = None) -> str:
+def _appsec_rc_capabilities(test_tracer: Optional[ddtrace.trace.Tracer] = None) -> str:
     r"""return the bit representation of the composed capabilities in base64
     bit 0: Reserved
     bit 1: ASM 1-click Activation

ddtrace/contrib/internal/sqlalchemy/engine.py

@@ -37,7 +37,7 @@ def trace_engine(engine, tracer=None, service=None):
     Add tracing instrumentation to the given sqlalchemy engine or instance.
 
     :param sqlalchemy.Engine engine: a SQLAlchemy engine class or instance
-    :param ddtrace._trace.tracer.Tracer tracer: a tracer instance. will default to the global
+    :param ddtrace.trace.Tracer tracer: a tracer instance. will default to the global
     :param str service: the name of the service to trace.
     """
     tracer = tracer or ddtrace.tracer  # by default use global

ddtrace/debugging/_debugger.py

@@ -23,7 +23,6 @@
 
 import ddtrace
 from ddtrace import config as ddconfig
-from ddtrace._trace.tracer import Tracer
 from ddtrace.debugging._config import di_config
 from ddtrace.debugging._function.discovery import FunctionDiscovery
 from ddtrace.debugging._function.store import FullyNamedContextWrappedFunction
@@ -55,6 +54,7 @@
 from ddtrace.internal.remoteconfig.worker import remoteconfig_poller
 from ddtrace.internal.service import Service
 from ddtrace.internal.wrapping.context import WrappingContext
+from ddtrace.trace import Tracer
 
 
 log = get_logger(__name__)

ddtrace/llmobs/_llmobs.py

@@ -295,7 +295,7 @@ def enable(
         api_key: Optional[str] = None,
         env: Optional[str] = None,
         service: Optional[str] = None,
-        _tracer: Optional[ddtrace._trace.tracer.Tracer] = None,
+        _tracer: Optional[ddtrace.trace.Tracer] = None,
     ) -> None:
         """
         Enable LLM Observability tracing.

ddtrace/opentracer/span.py

@@ -22,7 +22,7 @@
 
 
 if TYPE_CHECKING:  # pragma: no cover
-    from ddtrace._trace.tracer import Tracer  # noqa:F401
+    from ddtrace.trace import Tracer  # noqa:F401
 
 
 _TagNameType = Union[Text, bytes]

ddtrace/profiling/collector/_lock.py

@@ -10,7 +10,6 @@
 
 import wrapt
 
-from ddtrace._trace.tracer import Tracer
 from ddtrace.internal.datadog.profiling import ddup
 from ddtrace.internal.logger import get_logger
 from ddtrace.profiling import _threading
@@ -20,6 +19,7 @@
 from ddtrace.profiling.collector import _traceback
 from ddtrace.profiling.recorder import Recorder
 from ddtrace.settings.profiling import config
+from ddtrace.trace import Tracer
 
 
 LOG = get_logger(__name__)

ddtrace/profiling/collector/pytorch.py

@@ -7,12 +7,12 @@
 
 import wrapt
 
-from ddtrace._trace.tracer import Tracer
 from ddtrace.internal.datadog.profiling import ddup
 from ddtrace.profiling import _threading
 from ddtrace.profiling import collector
 from ddtrace.profiling.recorder import Recorder
 from ddtrace.settings.profiling import config
+from ddtrace.trace import Tracer
 
 
 LOG = logging.getLogger(__name__)

ddtrace/profiling/collector/stack.pyi

@@ -4,4 +4,4 @@ import ddtrace
 from ddtrace.profiling import collector
 
 class StackCollector(collector.PeriodicCollector):
-    tracer: typing.Optional[ddtrace._trace.tracer.Tracer]
+    tracer: typing.Optional[ddtrace.trace.Tracer]

ddtrace/profiling/collector/stack.pyx

@@ -10,7 +10,7 @@ import typing
 from ddtrace.internal._unpatched import _threading as ddtrace_threading
 from ddtrace._trace import context
 from ddtrace._trace import span as ddspan
-from ddtrace._trace.tracer import Tracer
+from ddtrace.trace import Tracer
 from ddtrace.internal._threads import periodic_threads
 from ddtrace.internal.datadog.profiling import ddup
 from ddtrace.internal.datadog.profiling import stack_v2