Restore GenAI semantic processor instrumentation

Author: nagkumar91

instrumentation-genai/opentelemetry-instrumentation-openai-agents-v2/src/opentelemetry/instrumentation/openai_agents/__init__.py

@@ -1,27 +1,13 @@
-# Copyright The OpenTelemetry Authors
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
 """OpenAI Agents instrumentation for OpenTelemetry."""
 
 from __future__ import annotations
 
+import importlib
+import logging
 import os
-from typing import Collection, Protocol
-
-from agents import tracing
-from agents.tracing.processor_interface import TracingProcessor
+from typing import Any, Collection
 
+from opentelemetry._events import get_event_logger
 from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
 from opentelemetry.semconv._incubating.attributes import (
     gen_ai_attributes as GenAI,
@@ -142,46 +128,6 @@ class OpenAIAgentsInstrumentor(BaseInstrumentor):
     def __init__(self) -> None:
         super().__init__()
         self._processor: GenAISemanticProcessor | None = None
-from .package import _instruments
-from .span_processor import _OpenAIAgentsSpanProcessor
-from .version import __version__  # noqa: F401
-
-
-class _ProcessorHolder(Protocol):
-    _processors: Collection[TracingProcessor]
-
-
-class _TraceProviderLike(Protocol):
-    _multi_processor: _ProcessorHolder
-
-
-__all__ = ["OpenAIAgentsInstrumentor"]
-
-
-def _resolve_system(_: str | None) -> str:
-    # OpenAI spans must report provider name "openai" per semantic conventions.
-    return GenAI.GenAiSystemValues.OPENAI.value
-
-
-def _get_registered_processors(
-    provider: _TraceProviderLike,
-) -> list[TracingProcessor]:
-    """Return tracing processors registered on the OpenAI Agents trace provider.
-
-    The provider exposes a private `_multi_processor` attribute with a `_processors`
-    collection that stores the currently registered processors in execution order.
-    """
-    multi = getattr(provider, "_multi_processor", None)
-    processors = getattr(multi, "_processors", ())
-    return list(processors)
-
-
-class OpenAIAgentsInstrumentor(BaseInstrumentor):
-    """Instrumentation that bridges OpenAI Agents tracing to OpenTelemetry spans."""
-
-    def __init__(self) -> None:
-        super().__init__()
-        self._processor: _OpenAIAgentsSpanProcessor | None = None
 
     def _instrument(self, **kwargs) -> None:
         if self._processor is not None:
@@ -237,17 +183,6 @@ def _instrument(self, **kwargs) -> None:
         )
 
         tracing = _load_tracing_module()
-        system = _resolve_system(kwargs.get("system"))
-        agent_name_override = kwargs.get("agent_name") or os.getenv(
-            "OTEL_GENAI_AGENT_NAME"
-        )
-
-        processor = _OpenAIAgentsSpanProcessor(
-            tracer=tracer,
-            system=system,
-            agent_name_override=agent_name_override,
-        )
-
         provider = tracing.get_trace_provider()
         existing = _get_registered_processors(provider)
         provider.set_processors([*existing, processor])
@@ -267,8 +202,6 @@ def _uninstrument(self, **kwargs) -> None:
             self._processor.shutdown()
         finally:
             self._processor = None
-        self._processor.shutdown()
-        self._processor = None
 
     def instrumentation_dependencies(self) -> Collection[str]:
         return _instruments

instrumentation-genai/opentelemetry-instrumentation-openai-agents-v2/tests/test_tracer.py

@@ -48,6 +48,12 @@
 from opentelemetry.trace import SpanKind  # noqa: E402
 
 GEN_AI_PROVIDER_NAME = GenAI.GEN_AI_PROVIDER_NAME
+GEN_AI_INPUT_MESSAGES = getattr(
+    GenAI, "GEN_AI_INPUT_MESSAGES", "gen_ai.input.messages"
+)
+GEN_AI_OUTPUT_MESSAGES = getattr(
+    GenAI, "GEN_AI_OUTPUT_MESSAGES", "gen_ai.output.messages"
+)
 
 
 def _instrument_with_provider(**instrument_kwargs):