update span name

Author: keith-decker

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

@@ -227,9 +227,8 @@ def _span_for_invocation(self, invocation: LLMInvocation):
         The span is not ended automatically on exiting the context; callers
         must finalize via _finalize_invocation.
         """
-        request_model = invocation.attributes.get("request_model")
         span = self._start_span(
-            name=f"{GenAI.GenAiOperationNameValues.CHAT.value} {request_model}",
+            name=f"{GenAI.GenAiOperationNameValues.CHAT.value} {invocation.request_model}",
             kind=SpanKind.CLIENT,
             parent_run_id=invocation.parent_run_id,
         )

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

@@ -66,12 +66,14 @@ def __init__(self, emitter_type_full: bool = True, **kwargs: Any):
 
     def start_llm(
         self,
+        request_model: str,
         prompts: List[InputMessage],
         run_id: UUID,
         parent_run_id: Optional[UUID] = None,
         **attributes: Any,
     ) -> None:
         invocation = LLMInvocation(
+            request_model=request_model,
             messages=prompts,
             run_id=run_id,
             parent_run_id=parent_run_id,

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

@@ -80,6 +80,7 @@ class LLMInvocation:
     """
 
     run_id: UUID
+    request_model: str
     parent_run_id: Optional[UUID] = None
     start_time: float = field(default_factory=time.time)
     end_time: Optional[float] = None

util/opentelemetry-util-genai/tests/test_utils.py

@@ -132,7 +132,11 @@ def test_llm_start_and_stop_creates_span(self):  # pylint: disable=no-self-use
 
         # Start and stop LLM invocation
         self.telemetry_handler.start_llm(
-            [message], run_id=run_id, custom_attr="value", system="test-system"
+            request_model="test-model",
+            prompts=[message],
+            run_id=run_id,
+            custom_attr="value",
+            system="test-system",
         )
         invocation = self.telemetry_handler.stop_llm(
             run_id, chat_generations=[chat_generation], extra="info"
@@ -142,7 +146,7 @@ def test_llm_start_and_stop_creates_span(self):  # pylint: disable=no-self-use
         spans = self.span_exporter.get_finished_spans()
         assert len(spans) == 1
         span = spans[0]
-        assert span.name == "test-system.chat"
+        assert span.name == "chat test-model"
         assert span.kind == trace.SpanKind.CLIENT
 
         # Verify span attributes