LQConstPlanWriterAgent/foundation/infrastructure/tracing/__init__.py

"""
链路追踪模块

提供分布式链路追踪功能
"""

from .trace_context import TraceContext, auto_trace

try:
    from .celery_trace import CeleryTraceManager, init, add_trace_to_celery_task
except ModuleNotFoundError:
    CeleryTraceManager = None

    def init():
        return None

    def add_trace_to_celery_task(celery_task_func):
        return celery_task_func

__all__ = [
    "TraceContext",
    "auto_trace",
    "CeleryTraceManager",
    "init",
    "add_trace_to_celery_task"
]