Change typing

Author: holtskinner

src/a2a/utils/telemetry.py

@@ -59,15 +59,19 @@ def internal_method(self):
 import logging
 
 from collections.abc import Callable
-from typing import Any
+from typing import TYPE_CHECKING, Any
 
 
+if TYPE_CHECKING:
+    from opentelemetry.trace import SpanKind as SpanKindType
+else:
+    SpanKindType = object
+
 logger = logging.getLogger(__name__)
 
 try:
     from opentelemetry import trace
-    from opentelemetry.trace import SpanKind as _SpanKind
-    from opentelemetry.trace import StatusCode
+    from opentelemetry.trace import SpanKind, StatusCode
 
 except ImportError:
     logger.debug(
@@ -91,12 +95,9 @@ def __getattr__(self, name: str) -> '_NoOp':
             return self
 
     trace = _NoOp()
-    _SpanKind = _NoOp()
+    SpanKind = _NoOp()
     StatusCode = _NoOp()
 
-SpanKind = _SpanKind
-__all__ = ['SpanKind']
-
 INSTRUMENTING_MODULE_NAME = 'a2a-python-sdk'
 INSTRUMENTING_MODULE_VERSION = '1.0.0'
 
@@ -105,7 +106,7 @@ def trace_function(  # noqa: PLR0915
     func: Callable | None = None,
     *,
     span_name: str | None = None,
-    kind: SpanKind = SpanKind.INTERNAL,
+    kind: SpanKindType = SpanKind.INTERNAL,
     attributes: dict[str, Any] | None = None,
     attribute_extractor: Callable | None = None,
 ) -> Callable:
@@ -252,7 +253,7 @@ def sync_wrapper(*args, **kwargs) -> Any:
 def trace_class(
     include_list: list[str] | None = None,
     exclude_list: list[str] | None = None,
-    kind: SpanKind = SpanKind.INTERNAL,
+    kind: SpanKindType = SpanKind.INTERNAL,
 ) -> Callable:
     """A class decorator to automatically trace specified methods of a class.