fix: satisfy misc ci checks

Author: sipercai

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

@@ -115,7 +115,7 @@ def on_chat_model_start(
         for sub_messages in messages:
             for message in sub_messages:
                 # Cast to Any to avoid type checking issues with LangChain's complex content type
-                raw_content: Any = message.content  # type: ignore[misc]
+                raw_content: Any = message.content
                 role = message.type
                 parts: list[Text] = []
 

instrumentation-loongsuite/loongsuite-instrumentation-langchain/src/opentelemetry/instrumentation/langchain/internal/_tracer.py

@@ -436,7 +436,9 @@ def _start_chain(self, run: Run) -> None:
 
         # Attach chain span context so non-LangChain children nest correctly.
         current_context = (
-            parent_ctx if parent_ctx is not None else otel_context.get_current()
+            parent_ctx
+            if parent_ctx is not None
+            else otel_context.get_current()
         )
         ctx = set_span_in_context(span, current_context)
         token = otel_context.attach(ctx)
@@ -725,7 +727,9 @@ def _enter_react_step(self, agent_run_id: UUID) -> None:
         self._handler.start_react_step(inv, context=agent_rd.original_context)
 
         step_ctx = (
-            otel_context.get_current() if inv.span else agent_rd.original_context
+            otel_context.get_current()
+            if inv.span
+            else agent_rd.original_context
         )
         agent_rd.active_step = _RunData(
             run_kind="react_step",

instrumentation-loongsuite/loongsuite-instrumentation-langchain/tests/test_agent_spans.py

@@ -116,7 +116,9 @@ def test_agent_context_colors_child_llm_and_tool_spans(
 
     spans = span_exporter.get_finished_spans()
     llm_span = next(span for span in spans if span.name == "chat gpt-4o-mini")
-    tool_span = next(span for span in spans if span.name == "execute_tool search")
+    tool_span = next(
+        span for span in spans if span.name == "execute_tool search"
+    )
 
     assert llm_span.attributes[GenAI.GEN_AI_AGENT_NAME] == "AgentExecutor"
     assert tool_span.attributes[GenAI.GEN_AI_AGENT_NAME] == "AgentExecutor"

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

@@ -64,7 +64,7 @@
 import logging
 import timeit
 from contextlib import contextmanager
-from typing import Iterator
+from typing import Any, Iterator, Protocol
 
 from opentelemetry import baggage
 from opentelemetry import context as otel_context
@@ -96,7 +96,6 @@
 )
 from opentelemetry.util.genai.types import (
     Error,
-    GenAIInvocation,
     LLMInvocation,
 )
 from opentelemetry.util.genai.version import __version__
@@ -106,7 +105,13 @@
 
 # Aliyun Python Agent Extension
 _AUTO_INJECT_BAGGAGE_PREFIX = "traffic.llm_sdk."
-_AGENT_NAME_BAGGAGE_KEY = f"{_AUTO_INJECT_BAGGAGE_PREFIX}{GenAI.GEN_AI_AGENT_NAME}"
+_AGENT_NAME_BAGGAGE_KEY = (
+    f"{_AUTO_INJECT_BAGGAGE_PREFIX}{GenAI.GEN_AI_AGENT_NAME}"
+)
+
+
+class _InvocationWithAttributes(Protocol):
+    attributes: dict[str, Any]
 
 
 # LoongSuite Extension
@@ -135,7 +140,7 @@ def _current_context(context: Context | None = None) -> Context:
 
 
 def _inject_agent_name_from_baggage(
-    invocation: GenAIInvocation, context: Context
+    invocation: _InvocationWithAttributes, context: Context
 ) -> None:
     if GenAI.GEN_AI_AGENT_NAME in invocation.attributes:
         return

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

@@ -711,8 +711,12 @@ def test_agent_context_colors_llm_and_tool_spans(self):
             self.telemetry_handler.stop_invoke_agent(agent_invocation)
 
         spans = self.span_exporter.get_finished_spans()
-        llm_span = next(span for span in spans if span.name == "chat gpt-4o-mini")
-        tool_span = next(span for span in spans if span.name == "execute_tool search")
+        llm_span = next(
+            span for span in spans if span.name == "chat gpt-4o-mini"
+        )
+        tool_span = next(
+            span for span in spans if span.name == "execute_tool search"
+        )
         self.assertEqual(
             llm_span.attributes.get(GenAI.GEN_AI_AGENT_NAME),
             "PlannerAgent",
@@ -744,7 +748,9 @@ def test_explicit_agent_parent_context_colors_llm_span(self):
             self.telemetry_handler.stop_invoke_agent(agent_invocation)
 
         spans = self.span_exporter.get_finished_spans()
-        llm_span = next(span for span in spans if span.name == "chat gpt-4o-mini")
+        llm_span = next(
+            span for span in spans if span.name == "chat gpt-4o-mini"
+        )
         self.assertEqual(
             llm_span.attributes.get(GenAI.GEN_AI_AGENT_NAME),
             "ExplicitParentAgent",