Update __all__ export to include OpenTelemetry functions

Added OpenTelemetry integration functions to __all__ export.
1 month ago · 9a9a344855
parent bd5b497e45
commit 9a9a344855
1 changed files with 32 additions and 6 deletions
--- 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",
+    ]