From 9a9a344855d9110eab37de8c7532b34d7649f8b3 Mon Sep 17 00:00:00 2001
From: CI-DEV <154627941+IlumCI@users.noreply.github.com>
Date: Wed, 12 Nov 2025 19:04:21 +0200
Subject: [PATCH] Update __all__ export to include OpenTelemetry functions

Added OpenTelemetry integration functions to __all__ export.
---
 swarms/telemetry/__init__.py | 38 ++++++++++++++++++++++++++++++------
 1 file changed, 32 insertions(+), 6 deletions(-)

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",
+    ]