paid-python 0.4.0__tar.gz → 0.4.1a0__tar.gz

{paid_python-0.4.0 → paid_python-0.4.1a0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: paid-python
-Version: 0.4.0
+Version: 0.4.1a0
 Summary: 
 Requires-Python: >=3.9,<3.14
 Classifier: Intended Audience :: Developers

{paid_python-0.4.0 → paid_python-0.4.1a0}/pyproject.toml

@@ -3,7 +3,7 @@ name = "paid-python"
 
 [tool.poetry]
 name = "paid-python"
-version = "0.4.0"
+version = "0.4.1-alpha0"
 description = ""
 readme = "README.md"
 authors = []
@@ -26,7 +26,8 @@ classifiers = [
     "Typing :: Typed"
 ]
 packages = [
-    { include = "paid", from = "src"}
+    { include = "paid", from = "src"},
+    { include = "opentelemetry", from = "vendor/patched-instrumentation-openllmetry/packages/opentelemetry-instrumentation-openai" },
 ]
 
 [project.urls]
@@ -44,7 +45,7 @@ opentelemetry-exporter-otlp-proto-http = ">=1.23.0"
 mutagen = ">=1.47.0"
 python-dotenv = ">=0.15.0"
 opentelemetry-instrumentation-anthropic = ">=0.47.0"
-# opentelemetry-instrumentation-openai is vendored from https://github.com/paid-ai/openllmetry-instrumentation-patch (patched version)
+# opentelemetry-instrumentation-openai is vendored from git a submodule under vendor/patched-instrumentation-openllmetry
 openinference-instrumentation-langchain = ">=0.1.55"
 openinference-instrumentation-openai-agents = ">=1.0.0"
 openinference-instrumentation-bedrock = ">=0.1.0"
@@ -71,9 +72,15 @@ openai = ">=1.12.0"
 [tool.pytest.ini_options]
 testpaths = [ "tests" ]
 asyncio_mode = "auto"
+# TODO(Ata): remove this once the openai instrumentation isn't vendored anymore
+norecursedirs = ["vendor/patched-instrumentation-openllmetry"]
 
 [tool.mypy]
 plugins = ["pydantic.mypy"]
+exclude = [
+    # TODO(Ata): remove this once the openai instrumentation isn't vendored anymore
+    "^vendor/patched-instrumentation-openllmetry/",
+]
 
 [[tool.mypy.overrides]]
 module = [
@@ -87,8 +94,8 @@ module = [
     "agents",
     "agents.*",
     "langchain_core.*",
-    "paid._vendor", # remove once openai instrumentor is upstream
-    "paid._vendor.*", # remove once openai instrumentor is upstream
+    "opentelemetry.instrumentation.openai",  # TODO(Ata): remove this once the openai instrumentation isn't vendored anymore
+    "opentelemetry.instrumentation.openai.*",
 ]
 ignore_missing_imports = true
 ignore_errors = true # remove once openai instrumentor is upstream

{paid_python-0.4.0 → paid_python-0.4.1a0}/src/paid/client.py

@@ -1,8 +1,13 @@
 # This file was auto-generated by Fern from our API Definition.
+import os
 import typing
 import warnings
 
 import httpx
+from dotenv import load_dotenv
+
+# Load environment variables from .env file
+load_dotenv()
 from .agents.client import AgentsClient, AsyncAgentsClient
 from .contacts.client import AsyncContactsClient, ContactsClient
 from .core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
@@ -60,11 +65,19 @@ class Paid:
         *,
         base_url: typing.Optional[str] = None,
         environment: PaidEnvironment = PaidEnvironment.PRODUCTION,
-        token: typing.Union[str, typing.Callable[[], str]],
+        token: typing.Optional[typing.Union[str, typing.Callable[[], str]]] = None,
         timeout: typing.Optional[float] = None,
         follow_redirects: typing.Optional[bool] = True,
         httpx_client: typing.Optional[httpx.Client] = None,
     ):
+        # If token is not provided, try to get it from environment variable
+        if token is None:
+            token = os.environ.get("PAID_API_KEY")
+            if token is None:
+                raise ValueError(
+                    "API token must be provided either via the 'token' parameter or the 'PAID_API_KEY' environment variable"
+                )
+
         _defaulted_timeout = (
             timeout if timeout is not None else 60 if httpx_client is None else httpx_client.timeout.read
         )
@@ -338,11 +351,19 @@ class AsyncPaid:
         *,
         base_url: typing.Optional[str] = None,
         environment: PaidEnvironment = PaidEnvironment.PRODUCTION,
-        token: typing.Union[str, typing.Callable[[], str]],
+        token: typing.Optional[typing.Union[str, typing.Callable[[], str]]] = None,
         timeout: typing.Optional[float] = None,
         follow_redirects: typing.Optional[bool] = True,
         httpx_client: typing.Optional[httpx.AsyncClient] = None,
     ):
+        # If token is not provided, try to get it from environment variable
+        if token is None:
+            token = os.environ.get("PAID_API_KEY")
+            if token is None:
+                raise ValueError(
+                    "API token must be provided either via the 'token' parameter or the 'PAID_API_KEY' environment variable"
+                )
+
         _defaulted_timeout = (
             timeout if timeout is not None else 60 if httpx_client is None else httpx_client.timeout.read
         )

{paid_python-0.4.0 → paid_python-0.4.1a0}/src/paid/tracing/__init__.py

@@ -7,7 +7,7 @@ from .distributed_tracing import (
     unset_tracing_token,
 )
 from .signal import signal
-from .tracing import initialize_tracing, get_paid_tracer_provider
+from .tracing import get_paid_tracer_provider, initialize_tracing
 
 __all__ = [
     "generate_tracing_token",

{paid_python-0.4.0 → paid_python-0.4.1a0}/src/paid/tracing/autoinstrumentation.py

@@ -22,8 +22,7 @@ except ImportError:
     ANTHROPIC_AVAILABLE = False
 
 try:
-    # from opentelemetry.instrumentation.openai import OpenAIInstrumentor
-    from paid._vendor.opentelemetry.instrumentation.openai import OpenAIInstrumentor  # remove once openai instrumentor is upstream
+    from opentelemetry.instrumentation.openai import OpenAIInstrumentor
 
     OPENAI_AVAILABLE = True
 except ImportError:

{paid_python-0.4.0/src/paid/_vendor → paid_python-0.4.1a0/vendor/patched-instrumentation-openllmetry/packages/opentelemetry-instrumentation-openai}/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_python-0.4.0/src/paid/_vendor → paid_python-0.4.1a0/vendor/patched-instrumentation-openllmetry/packages/opentelemetry-instrumentation-openai}/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,
 )

paid_python-0.4.1a0/vendor/patched-instrumentation-openllmetry/packages/opentelemetry-instrumentation-openai/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_python-0.4.0/src/paid/_vendor → paid_python-0.4.1a0/vendor/patched-instrumentation-openllmetry/packages/opentelemetry-instrumentation-openai}/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_python-0.4.0/src/paid/_vendor → paid_python-0.4.1a0/vendor/patched-instrumentation-openllmetry/packages/opentelemetry-instrumentation-openai}/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_python-0.4.0/src/paid/_vendor → paid_python-0.4.1a0/vendor/patched-instrumentation-openllmetry/packages/opentelemetry-instrumentation-openai}/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_python-0.4.0/src/paid/_vendor → paid_python-0.4.1a0/vendor/patched-instrumentation-openllmetry/packages/opentelemetry-instrumentation-openai}/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_python-0.4.0/src/paid/_vendor → paid_python-0.4.1a0/vendor/patched-instrumentation-openllmetry/packages/opentelemetry-instrumentation-openai}/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_python-0.4.0/src/paid/_vendor → paid_python-0.4.1a0/vendor/patched-instrumentation-openllmetry/packages/opentelemetry-instrumentation-openai}/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_python-0.4.0/src/paid/_vendor → paid_python-0.4.1a0/vendor/patched-instrumentation-openllmetry/packages/opentelemetry-instrumentation-openai}/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