paid-python 0.3.6__py3-none-any.whl → 0.4.1__py3-none-any.whl

{paid/_vendor/opentelemetry → opentelemetry}/instrumentation/openai/__init__.py

@@ -1,8 +1,8 @@
 from typing import Callable, Collection, Optional
 
 from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
-from paid._vendor.opentelemetry.instrumentation.openai.shared.config import Config
-from paid._vendor.opentelemetry.instrumentation.openai.utils import is_openai_v1
+from opentelemetry.instrumentation.openai.shared.config import Config
+from opentelemetry.instrumentation.openai.utils import is_openai_v1
 from typing_extensions import Coroutine
 
 _instruments = ("openai >= 0.27.0",)
@@ -35,20 +35,20 @@ class OpenAIInstrumentor(BaseInstrumentor):
 
     def _instrument(self, **kwargs):
         if is_openai_v1():
-            from paid._vendor.opentelemetry.instrumentation.openai.v1 import OpenAIV1Instrumentor
+            from opentelemetry.instrumentation.openai.v1 import OpenAIV1Instrumentor
 
             OpenAIV1Instrumentor().instrument(**kwargs)
         else:
-            from paid._vendor.opentelemetry.instrumentation.openai.v0 import OpenAIV0Instrumentor
+            from opentelemetry.instrumentation.openai.v0 import OpenAIV0Instrumentor
 
             OpenAIV0Instrumentor().instrument(**kwargs)
 
     def _uninstrument(self, **kwargs):
         if is_openai_v1():
-            from paid._vendor.opentelemetry.instrumentation.openai.v1 import OpenAIV1Instrumentor
+            from opentelemetry.instrumentation.openai.v1 import OpenAIV1Instrumentor
 
             OpenAIV1Instrumentor().uninstrument(**kwargs)
         else:
-            from paid._vendor.opentelemetry.instrumentation.openai.v0 import OpenAIV0Instrumentor
+            from opentelemetry.instrumentation.openai.v0 import OpenAIV0Instrumentor
 
             OpenAIV0Instrumentor().uninstrument(**kwargs)

{paid/_vendor/opentelemetry → opentelemetry}/instrumentation/openai/shared/__init__.py

@@ -5,8 +5,8 @@ import openai
 import pydantic
 from importlib.metadata import version
 
-from paid._vendor.opentelemetry.instrumentation.openai.shared.config import Config
-from paid._vendor.opentelemetry.instrumentation.openai.utils import (
+from opentelemetry.instrumentation.openai.shared.config import Config
+from opentelemetry.instrumentation.openai.utils import (
     dont_throw,
     is_openai_v1,
 )

opentelemetry/instrumentation/openai/shared/audio_wrappers.py

@@ -0,0 +1,247 @@
+import logging
+import time
+
+from opentelemetry import context as context_api
+from opentelemetry.instrumentation.openai.shared import (
+    _set_client_attributes,
+    _set_request_attributes,
+    _set_response_attributes,
+    _set_span_attribute,
+    metric_shared_attributes,
+    model_as_dict,
+)
+from opentelemetry.instrumentation.openai.utils import (
+    _with_audio_telemetry_wrapper,
+    dont_throw,
+    is_openai_v1,
+    start_as_current_span_async,
+)
+from opentelemetry.instrumentation.utils import _SUPPRESS_INSTRUMENTATION_KEY
+from opentelemetry.metrics import Counter, Histogram
+from opentelemetry.semconv.attributes.error_attributes import ERROR_TYPE
+from opentelemetry.semconv_ai import (
+    SUPPRESS_LANGUAGE_MODEL_INSTRUMENTATION_KEY,
+    SpanAttributes,
+)
+from opentelemetry.trace import SpanKind, Status, StatusCode
+
+SPAN_NAME = "openai.audio.transcriptions"
+
+logger = logging.getLogger(__name__)
+
+
+def _get_audio_duration(file):
+    """
+    Extract audio duration from file object.
+    Returns duration in seconds, or None if unable to determine.
+    """
+    try:
+        # Try to get duration from common audio libraries
+        # First check if it's a file-like object with a name attribute
+        if hasattr(file, "name"):
+            file_path = file.name
+        elif isinstance(file, (str, bytes)):
+            # If it's a path string or bytes
+            return None
+        else:
+            # If it's a file-like object without name, we can't easily determine duration
+            return None
+
+        # Try mutagen (supports many formats)
+        try:
+            from mutagen import File as MutagenFile
+
+            audio = MutagenFile(file_path)
+            if audio and hasattr(audio.info, "length"):
+                return audio.info.length
+        except (ImportError, Exception):
+            pass
+
+    except Exception as e:
+        logger.debug(f"Unable to extract audio duration: {e}")
+
+    return None
+
+
+@_with_audio_telemetry_wrapper
+def transcription_wrapper(
+    tracer,
+    duration_histogram: Histogram,
+    exception_counter: Counter,
+    wrapped,
+    instance,
+    args,
+    kwargs,
+):
+    if context_api.get_value(_SUPPRESS_INSTRUMENTATION_KEY) or context_api.get_value(
+        SUPPRESS_LANGUAGE_MODEL_INSTRUMENTATION_KEY
+    ):
+        return wrapped(*args, **kwargs)
+
+    with tracer.start_as_current_span(
+        name=SPAN_NAME,
+        kind=SpanKind.CLIENT,
+    ) as span:
+        _handle_request(span, kwargs, instance)
+
+        try:
+            # record time for duration
+            start_time = time.time()
+            response = wrapped(*args, **kwargs)
+            end_time = time.time()
+        except Exception as e:  # pylint: disable=broad-except
+            end_time = time.time()
+            duration = end_time - start_time if "start_time" in locals() else 0
+            attributes = {
+                "error.type": e.__class__.__name__,
+            }
+
+            # if there are legal duration, record it
+            if duration > 0 and duration_histogram:
+                duration_histogram.record(duration, attributes=attributes)
+            if exception_counter:
+                exception_counter.add(1, attributes=attributes)
+
+            span.set_attribute(ERROR_TYPE, e.__class__.__name__)
+            span.record_exception(e)
+            span.set_status(Status(StatusCode.ERROR, str(e)))
+            span.end()
+
+            raise
+
+        duration = end_time - start_time
+
+        _handle_response(
+            response,
+            span,
+            instance,
+            duration_histogram,
+            duration,
+        )
+
+        return response
+
+
+@_with_audio_telemetry_wrapper
+async def atranscription_wrapper(
+    tracer,
+    duration_histogram: Histogram,
+    exception_counter: Counter,
+    wrapped,
+    instance,
+    args,
+    kwargs,
+):
+    if context_api.get_value(_SUPPRESS_INSTRUMENTATION_KEY) or context_api.get_value(
+        SUPPRESS_LANGUAGE_MODEL_INSTRUMENTATION_KEY
+    ):
+        return await wrapped(*args, **kwargs)
+
+    async with start_as_current_span_async(
+        tracer=tracer,
+        name=SPAN_NAME,
+        kind=SpanKind.CLIENT,
+    ) as span:
+        _handle_request(span, kwargs, instance)
+
+        try:
+            # record time for duration
+            start_time = time.time()
+            response = await wrapped(*args, **kwargs)
+            end_time = time.time()
+        except Exception as e:  # pylint: disable=broad-except
+            end_time = time.time()
+            duration = end_time - start_time if "start_time" in locals() else 0
+            attributes = {
+                "error.type": e.__class__.__name__,
+            }
+
+            # if there are legal duration, record it
+            if duration > 0 and duration_histogram:
+                duration_histogram.record(duration, attributes=attributes)
+            if exception_counter:
+                exception_counter.add(1, attributes=attributes)
+
+            span.set_attribute(ERROR_TYPE, e.__class__.__name__)
+            span.record_exception(e)
+            span.set_status(Status(StatusCode.ERROR, str(e)))
+            span.end()
+
+            raise
+
+        duration = end_time - start_time
+
+        _handle_response(
+            response,
+            span,
+            instance,
+            duration_histogram,
+            duration,
+        )
+
+        return response
+
+
+@dont_throw
+def _handle_request(span, kwargs, instance):
+    _set_request_attributes(span, kwargs, instance)
+    _set_client_attributes(span, instance)
+
+    # Extract and set audio duration
+    file_param = kwargs.get("file")
+    if file_param:
+        audio_duration = _get_audio_duration(file_param)
+        if audio_duration is not None:
+            # _set_span_attribute(
+            #     span, SpanAttributes.LLM_OPENAI_AUDIO_INPUT_DURATION_SECONDS, audio_duration
+            # )
+            # TODO(Ata): come back here later when semconv is published
+            _set_span_attribute(
+                span, 'gen_ai.openai.audio.input.duration_seconds', audio_duration
+            )
+        else:
+            print("REMOVE ME : ATA-DBG : COULD NOT READ AUDIO FILE WITH MUTAGEN")
+
+
+@dont_throw
+def _handle_response(
+    response,
+    span,
+    instance=None,
+    duration_histogram=None,
+    duration=None,
+):
+    if is_openai_v1():
+        response_dict = model_as_dict(response)
+    else:
+        response_dict = response
+
+    # metrics record
+    _set_transcription_metrics(
+        instance,
+        duration_histogram,
+        response_dict,
+        duration,
+    )
+
+    # span attributes
+    _set_response_attributes(span, response_dict)
+
+
+def _set_transcription_metrics(
+    instance,
+    duration_histogram,
+    response_dict,
+    duration,
+):
+    from opentelemetry.instrumentation.openai.shared import _get_openai_base_url
+
+    shared_attributes = metric_shared_attributes(
+        response_model=response_dict.get("model") or None,
+        operation="audio.transcriptions",
+        server_address=_get_openai_base_url(instance),
+    )
+
+    # duration metrics
+    if duration and isinstance(duration, (float, int)) and duration_histogram:
+        duration_histogram.record(duration, attributes=shared_attributes)

{paid/_vendor/opentelemetry → opentelemetry}/instrumentation/openai/shared/chat_wrappers.py

@@ -8,7 +8,7 @@ from typing import List, Optional, Union
 
 from opentelemetry import context as context_api
 import pydantic
-from paid._vendor.opentelemetry.instrumentation.openai.shared import (
+from opentelemetry.instrumentation.openai.shared import (
     OPENAI_LLM_USAGE_TOKEN_TYPES,
     _get_openai_base_url,
     _set_client_attributes,
@@ -24,14 +24,14 @@ from paid._vendor.opentelemetry.instrumentation.openai.shared import (
     propagate_trace_context,
     set_tools_attributes,
 )
-from paid._vendor.opentelemetry.instrumentation.openai.shared.config import Config
-from paid._vendor.opentelemetry.instrumentation.openai.shared.event_emitter import emit_event
-from paid._vendor.opentelemetry.instrumentation.openai.shared.event_models import (
+from opentelemetry.instrumentation.openai.shared.config import Config
+from opentelemetry.instrumentation.openai.shared.event_emitter import emit_event
+from opentelemetry.instrumentation.openai.shared.event_models import (
     ChoiceEvent,
     MessageEvent,
     ToolCall,
 )
-from paid._vendor.opentelemetry.instrumentation.openai.utils import (
+from opentelemetry.instrumentation.openai.utils import (
     _with_chat_telemetry_wrapper,
     dont_throw,
     is_openai_v1,

{paid/_vendor/opentelemetry → opentelemetry}/instrumentation/openai/shared/completion_wrappers.py

@@ -2,7 +2,7 @@ import logging
 
 from opentelemetry import context as context_api
 from opentelemetry import trace
-from paid._vendor.opentelemetry.instrumentation.openai.shared import (
+from opentelemetry.instrumentation.openai.shared import (
     _set_client_attributes,
     _set_functions_attributes,
     _set_request_attributes,
@@ -13,14 +13,14 @@ from paid._vendor.opentelemetry.instrumentation.openai.shared import (
     model_as_dict,
     propagate_trace_context,
 )
-from paid._vendor.opentelemetry.instrumentation.openai.shared.config import Config
+from opentelemetry.instrumentation.openai.shared.config import Config
 from opentelemetry.semconv.attributes.error_attributes import ERROR_TYPE
-from paid._vendor.opentelemetry.instrumentation.openai.shared.event_emitter import emit_event
-from paid._vendor.opentelemetry.instrumentation.openai.shared.event_models import (
+from opentelemetry.instrumentation.openai.shared.event_emitter import emit_event
+from opentelemetry.instrumentation.openai.shared.event_models import (
     ChoiceEvent,
     MessageEvent,
 )
-from paid._vendor.opentelemetry.instrumentation.openai.utils import (
+from opentelemetry.instrumentation.openai.utils import (
     _with_tracer_wrapper,
     dont_throw,
     is_openai_v1,

{paid/_vendor/opentelemetry → opentelemetry}/instrumentation/openai/shared/embeddings_wrappers.py

@@ -3,7 +3,7 @@ import time
 from collections.abc import Iterable
 
 from opentelemetry import context as context_api
-from paid._vendor.opentelemetry.instrumentation.openai.shared import (
+from opentelemetry.instrumentation.openai.shared import (
     OPENAI_LLM_USAGE_TOKEN_TYPES,
     _get_openai_base_url,
     _set_client_attributes,
@@ -15,13 +15,13 @@ from paid._vendor.opentelemetry.instrumentation.openai.shared import (
     model_as_dict,
     propagate_trace_context,
 )
-from paid._vendor.opentelemetry.instrumentation.openai.shared.config import Config
-from paid._vendor.opentelemetry.instrumentation.openai.shared.event_emitter import emit_event
-from paid._vendor.opentelemetry.instrumentation.openai.shared.event_models import (
+from opentelemetry.instrumentation.openai.shared.config import Config
+from opentelemetry.instrumentation.openai.shared.event_emitter import emit_event
+from opentelemetry.instrumentation.openai.shared.event_models import (
     ChoiceEvent,
     MessageEvent,
 )
-from paid._vendor.opentelemetry.instrumentation.openai.utils import (
+from opentelemetry.instrumentation.openai.utils import (
     _with_embeddings_telemetry_wrapper,
     dont_throw,
     is_openai_v1,

{paid/_vendor/opentelemetry → opentelemetry}/instrumentation/openai/shared/event_emitter.py

@@ -3,11 +3,11 @@ from enum import Enum
 from typing import Union
 
 from opentelemetry._logs import LogRecord
-from paid._vendor.opentelemetry.instrumentation.openai.shared.event_models import (
+from opentelemetry.instrumentation.openai.shared.event_models import (
     ChoiceEvent,
     MessageEvent,
 )
-from paid._vendor.opentelemetry.instrumentation.openai.utils import (
+from opentelemetry.instrumentation.openai.utils import (
     should_emit_events,
     should_send_prompts,
 )

{paid/_vendor/opentelemetry → opentelemetry}/instrumentation/openai/shared/image_gen_wrappers.py

@@ -1,13 +1,13 @@
 import time
 
 from opentelemetry import context as context_api
-from paid._vendor.opentelemetry.instrumentation.openai import is_openai_v1
-from paid._vendor.opentelemetry.instrumentation.openai.shared import (
+from opentelemetry.instrumentation.openai import is_openai_v1
+from opentelemetry.instrumentation.openai.shared import (
     _get_openai_base_url,
     metric_shared_attributes,
     model_as_dict,
 )
-from paid._vendor.opentelemetry.instrumentation.openai.utils import (
+from opentelemetry.instrumentation.openai.utils import (
     _with_image_gen_metric_wrapper,
 )
 from opentelemetry.instrumentation.utils import _SUPPRESS_INSTRUMENTATION_KEY

{paid/_vendor/opentelemetry → opentelemetry}/instrumentation/openai/utils.py

@@ -9,7 +9,7 @@ from packaging import version as pkg_version
 
 from opentelemetry import context as context_api
 from opentelemetry._logs import Logger
-from paid._vendor.opentelemetry.instrumentation.openai.shared.config import Config
+from opentelemetry.instrumentation.openai.shared.config import Config
 
 import openai
 
@@ -83,6 +83,29 @@ def _with_embeddings_telemetry_wrapper(func):
     return _with_embeddings_telemetry
 
 
+def _with_audio_telemetry_wrapper(func):
+    """Wrapper to convert the audio wrapper function into the expected format for wrapt."""
+    def _with_audio_telemetry(
+        tracer,
+        duration_histogram,
+        exception_counter,
+    ):
+        def wrapper(wrapped, instance, args, kwargs):
+            return func(
+                tracer,
+                duration_histogram,
+                exception_counter,
+                wrapped,
+                instance,
+                args,
+                kwargs,
+            )
+
+        return wrapper
+
+    return _with_audio_telemetry
+
+
 def _with_chat_telemetry_wrapper(func):
     def _with_chat_telemetry(
         tracer,

{paid/_vendor/opentelemetry → opentelemetry}/instrumentation/openai/v0/__init__.py

@@ -2,21 +2,21 @@ from typing import Collection
 
 from opentelemetry._logs import get_logger
 from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
-from paid._vendor.opentelemetry.instrumentation.openai.shared.chat_wrappers import (
+from opentelemetry.instrumentation.openai.shared.chat_wrappers import (
     achat_wrapper,
     chat_wrapper,
 )
-from paid._vendor.opentelemetry.instrumentation.openai.shared.completion_wrappers import (
+from opentelemetry.instrumentation.openai.shared.completion_wrappers import (
     acompletion_wrapper,
     completion_wrapper,
 )
-from paid._vendor.opentelemetry.instrumentation.openai.shared.config import Config
-from paid._vendor.opentelemetry.instrumentation.openai.shared.embeddings_wrappers import (
+from opentelemetry.instrumentation.openai.shared.config import Config
+from opentelemetry.instrumentation.openai.shared.embeddings_wrappers import (
     aembeddings_wrapper,
     embeddings_wrapper,
 )
-from paid._vendor.opentelemetry.instrumentation.openai.utils import is_metrics_enabled
-from paid._vendor.opentelemetry.instrumentation.openai.version import __version__
+from opentelemetry.instrumentation.openai.utils import is_metrics_enabled
+from opentelemetry.instrumentation.openai.version import __version__
 from opentelemetry.instrumentation.utils import unwrap
 from opentelemetry.metrics import get_meter
 from opentelemetry.semconv._incubating.metrics import gen_ai_metrics as GenAIMetrics

{paid/_vendor/opentelemetry → opentelemetry}/instrumentation/openai/v1/__init__.py

@@ -2,24 +2,28 @@ from typing import Collection
 
 from opentelemetry._logs import get_logger
 from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
-from paid._vendor.opentelemetry.instrumentation.openai.shared.chat_wrappers import (
+from opentelemetry.instrumentation.openai.shared.chat_wrappers import (
     achat_wrapper,
     chat_wrapper,
 )
-from paid._vendor.opentelemetry.instrumentation.openai.shared.completion_wrappers import (
+from opentelemetry.instrumentation.openai.shared.completion_wrappers import (
     acompletion_wrapper,
     completion_wrapper,
 )
-from paid._vendor.opentelemetry.instrumentation.openai.shared.config import Config
-from paid._vendor.opentelemetry.instrumentation.openai.shared.embeddings_wrappers import (
+from opentelemetry.instrumentation.openai.shared.config import Config
+from opentelemetry.instrumentation.openai.shared.embeddings_wrappers import (
     aembeddings_wrapper,
     embeddings_wrapper,
 )
-from paid._vendor.opentelemetry.instrumentation.openai.shared.image_gen_wrappers import (
+from opentelemetry.instrumentation.openai.shared.image_gen_wrappers import (
     image_gen_metrics_wrapper,
 )
-from paid._vendor.opentelemetry.instrumentation.openai.utils import is_metrics_enabled
-from paid._vendor.opentelemetry.instrumentation.openai.v1.assistant_wrappers import (
+from opentelemetry.instrumentation.openai.shared.audio_wrappers import (
+    atranscription_wrapper,
+    transcription_wrapper,
+)
+from opentelemetry.instrumentation.openai.utils import is_metrics_enabled
+from opentelemetry.instrumentation.openai.v1.assistant_wrappers import (
     assistants_create_wrapper,
     messages_list_wrapper,
     runs_create_and_stream_wrapper,
@@ -27,14 +31,14 @@ from paid._vendor.opentelemetry.instrumentation.openai.v1.assistant_wrappers imp
     runs_retrieve_wrapper,
 )
 
-from paid._vendor.opentelemetry.instrumentation.openai.v1.responses_wrappers import (
+from opentelemetry.instrumentation.openai.v1.responses_wrappers import (
     async_responses_cancel_wrapper,
     async_responses_get_or_create_wrapper,
     responses_cancel_wrapper,
     responses_get_or_create_wrapper,
 )
 
-from paid._vendor.opentelemetry.instrumentation.openai.version import __version__
+from opentelemetry.instrumentation.openai.version import __version__
 from opentelemetry.instrumentation.utils import unwrap
 from opentelemetry.metrics import get_meter
 from opentelemetry.semconv._incubating.metrics import gen_ai_metrics as GenAIMetrics
@@ -247,6 +251,36 @@ class OpenAIV1Instrumentor(BaseInstrumentor):
             image_gen_metrics_wrapper(duration_histogram, image_gen_exception_counter),
         )
 
+        if is_metrics_enabled():
+            audio_transcription_exception_counter = meter.create_counter(
+                # name=Meters.LLM_AUDIO_TRANSCRIPTIONS_EXCEPTIONS, # TODO(Ata): come back here later when semconv is published
+                name='llm.openai.audio.transcriptions.exceptions',
+                unit="time",
+                description="Number of exceptions occurred during audio transcriptions operation",
+            )
+        else:
+            audio_transcription_exception_counter = None
+
+        wrap_function_wrapper(
+            "openai.resources.audio.transcriptions",
+            "Transcriptions.create",
+            transcription_wrapper(
+                tracer,
+                duration_histogram,
+                audio_transcription_exception_counter,
+            ),
+        )
+
+        wrap_function_wrapper(
+            "openai.resources.audio.transcriptions",
+            "AsyncTranscriptions.create",
+            atranscription_wrapper(
+                tracer,
+                duration_histogram,
+                audio_transcription_exception_counter,
+            ),
+        )
+
         # Beta APIs may not be available consistently in all versions
         self._try_wrap(
             "openai.resources.beta.assistants",
@@ -338,6 +372,8 @@ class OpenAIV1Instrumentor(BaseInstrumentor):
         unwrap("openai.resources.completions", "AsyncCompletions.create")
         unwrap("openai.resources.embeddings", "AsyncEmbeddings.create")
         unwrap("openai.resources.images", "Images.generate")
+        unwrap("openai.resources.audio.transcriptions", "Transcriptions.create")
+        unwrap("openai.resources.audio.transcriptions", "AsyncTranscriptions.create")
 
         # Beta APIs may not be available consistently in all versions
         try:

{paid/_vendor/opentelemetry → opentelemetry}/instrumentation/openai/v1/assistant_wrappers.py

@@ -3,17 +3,17 @@ import time
 
 from opentelemetry import context as context_api
 from opentelemetry import trace
-from paid._vendor.opentelemetry.instrumentation.openai.shared import (
+from opentelemetry.instrumentation.openai.shared import (
     _set_span_attribute,
     model_as_dict,
 )
-from paid._vendor.opentelemetry.instrumentation.openai.shared.config import Config
-from paid._vendor.opentelemetry.instrumentation.openai.shared.event_emitter import emit_event
-from paid._vendor.opentelemetry.instrumentation.openai.shared.event_models import (
+from opentelemetry.instrumentation.openai.shared.config import Config
+from opentelemetry.instrumentation.openai.shared.event_emitter import emit_event
+from opentelemetry.instrumentation.openai.shared.event_models import (
     ChoiceEvent,
     MessageEvent,
 )
-from paid._vendor.opentelemetry.instrumentation.openai.utils import (
+from opentelemetry.instrumentation.openai.utils import (
     _with_tracer_wrapper,
     dont_throw,
     should_emit_events,
@@ -309,7 +309,7 @@ def runs_create_and_stream_wrapper(tracer, wrapped, instance, args, kwargs):
                 span, f"{GenAIAttributes.GEN_AI_PROMPT}.{i}.content", instructions
             )
 
-        from paid._vendor.opentelemetry.instrumentation.openai.v1.event_handler_wrapper import (
+        from opentelemetry.instrumentation.openai.v1.event_handler_wrapper import (
             EventHandleWrapper,
         )
 

{paid/_vendor/opentelemetry → opentelemetry}/instrumentation/openai/v1/event_handler_wrapper.py

@@ -1,7 +1,7 @@
-from paid._vendor.opentelemetry.instrumentation.openai.shared import _set_span_attribute
-from paid._vendor.opentelemetry.instrumentation.openai.shared.event_emitter import emit_event
-from paid._vendor.opentelemetry.instrumentation.openai.shared.event_models import ChoiceEvent
-from paid._vendor.opentelemetry.instrumentation.openai.utils import should_emit_events
+from opentelemetry.instrumentation.openai.shared import _set_span_attribute
+from opentelemetry.instrumentation.openai.shared.event_emitter import emit_event
+from opentelemetry.instrumentation.openai.shared.event_models import ChoiceEvent
+from opentelemetry.instrumentation.openai.utils import should_emit_events
 from opentelemetry.semconv.attributes.error_attributes import ERROR_TYPE
 from opentelemetry.semconv._incubating.attributes import (
     gen_ai_attributes as GenAIAttributes,