fix: linting problems

Author: andrei-rusu

src/uipath_llamaindex/_cli/_tracing/_attribute_normalizer.py

@@ -7,12 +7,10 @@
 
 import json
 import logging
-from collections.abc import Mapping
 from typing import Any, Optional
 
 from opentelemetry.context import Context
-from opentelemetry.sdk.trace import Span, SpanProcessor
-from opentelemetry.util.types import AttributeValue
+from opentelemetry.sdk.trace import ReadableSpan, Span, SpanProcessor
 
 logger = logging.getLogger(__name__)
 
@@ -27,17 +25,16 @@ def on_start(self, span: Span, parent_context: Optional[Context] = None) -> None
         """Called when span starts - no action needed."""
         pass
 
-    def on_end(self, span: Span) -> None:
+    def on_end(self, span: ReadableSpan) -> None:
         """Normalize tool call attributes before span is consumed by exporters/terminal."""
-        if not span._attributes:
+        # Access internal mutable attributes (BoundedAttributes at runtime)
+        # ReadableSpan._attributes is typed as Mapping but mutable in practice
+        attrs = getattr(span, "_attributes", None)
+        if not attrs:
             return
 
         try:
-            # Get the mutable internal attributes dict
-            attrs: Mapping[str, AttributeValue] = span._attributes
-
             if attrs.get("openinference.span.kind", None) == "TOOL":
-                # Normalize tool call attributes
                 for key in ("input.value", "output.value"):
                     if key in attrs:
                         original = attrs[key]
@@ -52,7 +49,6 @@ def on_end(self, span: Span) -> None:
                                 )
 
         except Exception as e:
-            # Don't crash span processing if normalization fails
             logger.debug(
                 f"Failed to normalize span '{getattr(span, 'name', 'unknown')}': {e}"
             )