some fixes

Author: lmolkova

dev-requirements.txt

@@ -1,6 +1,6 @@
 pylint==3.0.2
 httpretty==1.1.4
-pyright==v1.1.404
+pyright==v1.1.408
 sphinx==7.1.2
 sphinx-rtd-theme==2.0.0rc4
 sphinx-autodoc-typehints==1.25.2

instrumentation-genai/opentelemetry-instrumentation-anthropic/src/opentelemetry/instrumentation/anthropic/patch.py

@@ -24,8 +24,10 @@
     gen_ai_attributes as GenAIAttributes,
 )
 from opentelemetry.util.genai.handler import TelemetryHandler
+from opentelemetry.util.genai.inference_invocation import (
+    LLMInvocation,  # pyright: ignore[reportDeprecated]  # TODO: migrate to InferenceInvocation
+)
 from opentelemetry.util.genai.types import Error
-from opentelemetry.util.genai.inference_invocation import LLMInvocation  # pyright: ignore[reportDeprecated]  # TODO: migrate to InferenceInvocation
 from opentelemetry.util.genai.utils import (
     should_capture_content_on_spans_in_experimental_mode,
 )

instrumentation-genai/opentelemetry-instrumentation-anthropic/src/opentelemetry/instrumentation/anthropic/wrappers.py

@@ -19,12 +19,14 @@
 from typing import TYPE_CHECKING, Callable, Iterator, Optional
 
 from opentelemetry.util.genai.handler import TelemetryHandler
+from opentelemetry.util.genai.inference_invocation import (
+    LLMInvocation,  # pyright: ignore[reportDeprecated]  # TODO: migrate to InferenceInvocation
+)
 from opentelemetry.util.genai.types import (
     Error,
     MessagePart,
     OutputMessage,
 )
-from opentelemetry.util.genai.inference_invocation import LLMInvocation  # pyright: ignore[reportDeprecated]  # TODO: migrate to InferenceInvocation
 
 from .messages_extractors import (
     GEN_AI_USAGE_CACHE_CREATION_INPUT_TOKENS,

instrumentation-genai/opentelemetry-instrumentation-langchain/src/opentelemetry/instrumentation/langchain/callback_handler.py

@@ -25,7 +25,9 @@
     _InvocationManager,
 )
 from opentelemetry.util.genai.handler import TelemetryHandler
-from opentelemetry.util.genai.inference_invocation import LLMInvocation  # pyright: ignore[reportDeprecated]  # TODO: migrate to InferenceInvocation
+from opentelemetry.util.genai.inference_invocation import (
+    LLMInvocation,  # pyright: ignore[reportDeprecated]  # TODO: migrate to InferenceInvocation
+)
 from opentelemetry.util.genai.types import (
     Error,
     InputMessage,
@@ -171,7 +173,8 @@ def on_llm_end(
     ) -> None:
         llm_invocation = self._invocation_manager.get_invocation(run_id=run_id)
         if llm_invocation is None or not isinstance(
-            llm_invocation, LLMInvocation  # pyright: ignore[reportDeprecated]
+            llm_invocation,
+            LLMInvocation,  # pyright: ignore[reportDeprecated]
         ):
             # If the invocation does not exist, we cannot set attributes or end it
             return
@@ -262,7 +265,8 @@ def on_llm_error(
     ) -> None:
         llm_invocation = self._invocation_manager.get_invocation(run_id=run_id)
         if llm_invocation is None or not isinstance(
-            llm_invocation, LLMInvocation  # pyright: ignore[reportDeprecated]
+            llm_invocation,
+            LLMInvocation,  # pyright: ignore[reportDeprecated]
         ):
             # If the invocation does not exist, we cannot set attributes or end it
             return