lint

Author: lmolkova

instrumentation-genai/opentelemetry-instrumentation-openai-v2/tests/test_completion_hook.py

@@ -122,7 +122,6 @@ def test_default_hook_loaded_from_env(
     assert kwargs["outputs"]
     assert kwargs["span"] is not None
 
-    # Content goes to the hook only — not to span attributes or log records
     spans = span_exporter.get_finished_spans()
     assert len(spans) == 1
     span_attrs = spans[0].attributes or {}

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

@@ -69,6 +69,7 @@
 from opentelemetry import context as otel_context
 from opentelemetry._logs import (
     LoggerProvider,
+    LogRecord,
     get_logger,
 )
 from opentelemetry.metrics import MeterProvider, get_meter
@@ -219,7 +220,7 @@ def _call_completion_hook(
         self,
         invocation: LLMInvocation,
         span: Span,
-        log_record: object,
+        log_record: LogRecord | None,
     ) -> None:
         if self._completion_hook is not None:
             self._completion_hook.on_completion(