Skip to content

Commit 88d6883

Browse files
test: Initialize OpenAIInstrumentation in test setup
1 parent e705b4d commit 88d6883

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

src/tests/openai/test_embeddings.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,16 @@
11
import pytest
2+
import httpx
3+
from openai import OpenAI
4+
from opentelemetry.trace import SpanKind, StatusCode
5+
from langtrace_python_sdk.constants.instrumentation.common import SpanAttributes
26
from langtrace_python_sdk.constants.instrumentation.openai import APIS
7+
from langtrace_python_sdk.instrumentation.openai import OpenAIInstrumentation
38
from tests.utils import assert_token_count
49
from importlib_metadata import version as v
5-
from langtrace.trace_attributes import SpanAttributes
6-
from openai import OpenAI
7-
import httpx
810

11+
# Initialize OpenAI instrumentation
12+
instrumentor = OpenAIInstrumentation()
13+
instrumentor.instrument()
914

1015
@pytest.mark.vcr()
1116
def test_embeddings_base_url(exporter, openai_client):

0 commit comments

Comments
 (0)