Skip to content

Commit ea09524

Browse files
committed
Change import style
1 parent b138416 commit ea09524

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

src/a2a/utils/telemetry.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,8 @@ def internal_method(self):
7171

7272
try:
7373
from opentelemetry import trace
74-
from opentelemetry.trace import SpanKind, StatusCode
74+
from opentelemetry.trace import SpanKind as _SpanKind
75+
from opentelemetry.trace import StatusCode
7576

7677
except ImportError:
7778
logger.debug(
@@ -95,9 +96,12 @@ def __getattr__(self, name: str) -> '_NoOp':
9596
return self
9697

9798
trace = _NoOp()
98-
SpanKind = _NoOp()
99+
_SpanKind = _NoOp()
99100
StatusCode = _NoOp()
100101

102+
SpanKind = _SpanKind
103+
__all__ = ['SpanKind']
104+
101105
INSTRUMENTING_MODULE_NAME = 'a2a-python-sdk'
102106
INSTRUMENTING_MODULE_VERSION = '1.0.0'
103107

0 commit comments

Comments
 (0)