From d858c23bab7c561e8a32ff7c440244a16e5af508 Mon Sep 17 00:00:00 2001 From: Tao Chen Date: Thu, 4 Dec 2025 11:11:05 -0800 Subject: [PATCH] Fix observability module mypy error (#2641) --- .../packages/core/agent_framework/observability.py | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/python/packages/core/agent_framework/observability.py b/python/packages/core/agent_framework/observability.py index 8d2f9826ab..7ec778b8d2 100644 --- a/python/packages/core/agent_framework/observability.py +++ b/python/packages/core/agent_framework/observability.py @@ -20,7 +20,7 @@ from .exceptions import AgentInitializationError, ChatClientInitializationError if TYPE_CHECKING: # pragma: no cover from azure.core.credentials import TokenCredential - from opentelemetry.sdk._logs._internal.export import LogRecordExporter + from opentelemetry.sdk._logs.export import LogRecordExporter from opentelemetry.sdk.metrics.export import MetricExporter from opentelemetry.sdk.resources import Resource from opentelemetry.sdk.trace.export import SpanExporter @@ -268,9 +268,9 @@ def _get_otlp_exporters(endpoints: list[str]) -> list["LogRecordExporter | SpanE exporters: list["LogRecordExporter | SpanExporter | MetricExporter"] = [] for endpoint in endpoints: - exporters.append(OTLPLogExporter(endpoint=endpoint)) # type: ignore[arg-type] - exporters.append(OTLPSpanExporter(endpoint=endpoint)) # type: ignore[arg-type] - exporters.append(OTLPMetricExporter(endpoint=endpoint)) # type: ignore[arg-type] + exporters.append(OTLPLogExporter(endpoint=endpoint)) + exporters.append(OTLPSpanExporter(endpoint=endpoint)) + exporters.append(OTLPMetricExporter(endpoint=endpoint)) return exporters @@ -493,8 +493,7 @@ class ObservabilitySettings(AFBaseSettings): """Configure tracing, logging, events and metrics with the provided exporters.""" from opentelemetry._logs import set_logger_provider from opentelemetry.sdk._logs import LoggerProvider, LoggingHandler - from opentelemetry.sdk._logs._internal.export import LogRecordExporter - from opentelemetry.sdk._logs.export import BatchLogRecordProcessor + from opentelemetry.sdk._logs.export import BatchLogRecordProcessor, LogRecordExporter from opentelemetry.sdk.metrics import MeterProvider from opentelemetry.sdk.metrics.export import MetricExporter, PeriodicExportingMetricReader from opentelemetry.sdk.metrics.view import DropAggregation, View @@ -522,7 +521,7 @@ class ObservabilitySettings(AFBaseSettings): logger_provider.add_log_record_processor(BatchLogRecordProcessor(exporter)) should_add_console_exporter = False if should_add_console_exporter: - from opentelemetry.sdk._logs._internal.export import ConsoleLogExporter + from opentelemetry.sdk._logs.export import ConsoleLogExporter logger_provider.add_log_record_processor(BatchLogRecordProcessor(ConsoleLogExporter()))