update token types

Author: keith-decker

util/opentelemetry-util-genai/src/opentelemetry/util/genai/span_utils.py

@@ -60,9 +60,7 @@ def _apply_common_span_attributes(
         # TODO: clean provider name to match GenAiProviderNameValues?
         span.set_attribute(GenAI.GEN_AI_PROVIDER_NAME, provider)
 
-    finish_reasons: List[str] = []
-    for gen in invocation.output_messages:
-        finish_reasons.append(gen.finish_reason)
+    finish_reasons = [gen.finish_reason for gen in invocation.output_messages]
     if finish_reasons:
         span.set_attribute(
             GenAI.GEN_AI_RESPONSE_FINISH_REASONS, finish_reasons

util/opentelemetry-util-genai/src/opentelemetry/util/genai/types.py

@@ -23,7 +23,6 @@
 
 from opentelemetry.context import Context
 from opentelemetry.trace import Span
-from opentelemetry.util.types import AttributeValue
 
 ContextToken: TypeAlias = Token[Context]
 
@@ -115,8 +114,8 @@ class LLMInvocation:
     provider: Optional[str] = None
     response_model_name: Optional[str] = None
     response_id: Optional[str] = None
-    input_tokens: Optional[AttributeValue] = None
-    output_tokens: Optional[AttributeValue] = None
+    input_tokens: Optional[int] = None
+    output_tokens: Optional[int] = None
     attributes: Dict[str, Any] = field(default_factory=_new_str_any_dict)