fix typing

Agenta-AI · Sep 10, 2024 · 6316562 · 6316562
1 parent 79afaa4
commit 6316562
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/agenta-cli/agenta/sdk/tracing/spans.py b/agenta-cli/agenta/sdk/tracing/spans.py
@@ -1,5 +1,5 @@
 from json import loads, dumps
-from typing import Optional, Any, Dict
+from typing import Optional, Union, Any, Dict
 
 from opentelemetry.sdk.trace import Span, Status, StatusCode
 from opentelemetry.sdk.trace.export import ReadableSpan
@@ -42,7 +42,7 @@ def set_attributes(span: Span, namespace: str, attributes: Dict[str, Any]) -> No
         )
 
 
-def get_attributes(span: ReadableSpan | Span, namespace: str):
+def get_attributes(span: Union[ReadableSpan, Span], namespace: str):
     return {
         _decode_key(namespace, key): _decode_value(value)
         for key, value in span.attributes.items()