diff --git a/tests/integration/inference/test_provider_data_routing.py b/tests/integration/inference/test_provider_data_routing.py index 9b9806345..a5817e0ce 100644 --- a/tests/integration/inference/test_provider_data_routing.py +++ b/tests/integration/inference/test_provider_data_routing.py @@ -24,7 +24,7 @@ from llama_stack.apis.inference.inference import ( OpenAIChatCompletionUsage, OpenAIChoice, ) -from llama_stack.core.telemetry.telemetry import MetricEvent +from llama_stack.apis.telemetry.telemetry import MetricEvent class OpenAIChatCompletionWithMetrics(OpenAIChatCompletion):