diff --git a/api/core/ops/ops_trace_manager.py b/api/core/ops/ops_trace_manager.py index 4b612596f4..5b54038193 100644 --- a/api/core/ops/ops_trace_manager.py +++ b/api/core/ops/ops_trace_manager.py @@ -42,8 +42,8 @@ class OpsTraceProviderConfigMap(dict[str, dict[str, Any]]): def __getitem__(self, provider: str) -> dict[str, Any]: match provider: case TracingProviderEnum.ARIZE: - from core.ops.entities.config_entity import ArizeConfig from core.ops.arize_phoenix_trace.arize_phoenix_trace import ArizePhoenixDataTrace + from core.ops.entities.config_entity import ArizeConfig return { "config_class": ArizeConfig, @@ -53,8 +53,8 @@ class OpsTraceProviderConfigMap(dict[str, dict[str, Any]]): } case TracingProviderEnum.PHOENIX: - from core.ops.entities.config_entity import PhoenixConfig from core.ops.arize_phoenix_trace.arize_phoenix_trace import ArizePhoenixDataTrace + from core.ops.entities.config_entity import PhoenixConfig return { "config_class": PhoenixConfig,