From e22f78171bb49677a18060950e7d8b78b26597a0 Mon Sep 17 00:00:00 2001 From: zhongxuanwang Date: Sun, 1 Jun 2025 13:15:37 -0700 Subject: [PATCH 1/2] Fix AttributeError bug Signed-off-by: zhongxuanwang --- src/aiq/observability/register.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/aiq/observability/register.py b/src/aiq/observability/register.py index d48249a5a..9f82d58c3 100644 --- a/src/aiq/observability/register.py +++ b/src/aiq/observability/register.py @@ -117,9 +117,9 @@ class OtelCollectorTelemetryExporter(TelemetryExporterBaseConfig, name="otelcoll @register_telemetry_exporter(config_type=OtelCollectorTelemetryExporter) async def otel_telemetry_exporter(config: OtelCollectorTelemetryExporter, builder: Builder): """Create an OpenTelemetry telemetry exporter.""" - # If the dependencies are not installed, a TelemetryOptionalImportError will be raised - opentelemetry = try_import_opentelemetry() - yield opentelemetry.sdk.trace.export.OTLPSpanExporter(config.endpoint) + + trace_exporter = telemetry_optional_import("opentelemetry.exporter.otlp.proto.http.trace_exporter") + yield trace_exporter.OTLPSpanExporter(endpoint=config.endpoint) class ConsoleLoggingMethod(LoggingBaseConfig, name="console"): From 68434ca471555107825701d0d3edaf4e71a3af1b Mon Sep 17 00:00:00 2001 From: zhongxuanwang Date: Sun, 1 Jun 2025 13:49:25 -0700 Subject: [PATCH 2/2] Remove an unused import Signed-off-by: zhongxuanwang --- src/aiq/observability/register.py | 1 - 1 file changed, 1 deletion(-) diff --git a/src/aiq/observability/register.py b/src/aiq/observability/register.py index 9f82d58c3..df8c6fff9 100644 --- a/src/aiq/observability/register.py +++ b/src/aiq/observability/register.py @@ -24,7 +24,6 @@ from aiq.data_models.logging import LoggingBaseConfig from aiq.data_models.telemetry_exporter import TelemetryExporterBaseConfig from aiq.utils.optional_imports import telemetry_optional_import -from aiq.utils.optional_imports import try_import_opentelemetry from aiq.utils.optional_imports import try_import_phoenix logger = logging.getLogger(__name__)