Import openai injector from promptflow.tracing

microsoft · Mar 7, 2024 · 58fb1d3 · 58fb1d3
1 parent e0bbeb5
commit 58fb1d3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/promptflow/tests/tracing_test/e2etests/test_trace.py b/src/promptflow/tests/tracing_test/e2etests/test_trace.py
@@ -4,7 +4,7 @@
 import pytest
 from opentelemetry.trace.status import StatusCode
 
-from promptflow._core.openai_injector import inject_openai_api
+from promptflow.tracing._openai_injector import inject_openai_api
 from promptflow.tracing.contracts.trace import TraceType
 
 from ..utils import execute_function_in_subprocess, prepare_memory_exporter