diff --git a/swarms/telemetry/__init__.py b/swarms/telemetry/__init__.py index a7f92a78..5a13025d 100644 --- a/swarms/telemetry/__init__.py +++ b/swarms/telemetry/__init__.py @@ -5,9 +5,35 @@ from swarms.telemetry.main import ( log_agent_data, ) -__all__ = [ - "generate_user_id", - "get_machine_id", - "get_comprehensive_system_info", - "log_agent_data", -] +try: + from swarms.telemetry.opentelemetry_integration import ( + get_tracer, + get_meter, + trace_span, + trace_function, + record_metric, + get_current_trace_context, + set_trace_context, + log_event, + ) + __all__ = [ + "generate_user_id", + "get_machine_id", + "get_comprehensive_system_info", + "log_agent_data", + "get_tracer", + "get_meter", + "trace_span", + "trace_function", + "record_metric", + "get_current_trace_context", + "set_trace_context", + "log_event", + ] +except ImportError: + __all__ = [ + "generate_user_id", + "get_machine_id", + "get_comprehensive_system_info", + "log_agent_data", + ]