Skip to content

Commit d858c23

Browse files
authored
Fix observability module mypy error (#2641)
1 parent 365e438 commit d858c23

File tree

1 file changed

+6
-7
lines changed

1 file changed

+6
-7
lines changed

python/packages/core/agent_framework/observability.py

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020

2121
if TYPE_CHECKING: # pragma: no cover
2222
from azure.core.credentials import TokenCredential
23-
from opentelemetry.sdk._logs._internal.export import LogRecordExporter
23+
from opentelemetry.sdk._logs.export import LogRecordExporter
2424
from opentelemetry.sdk.metrics.export import MetricExporter
2525
from opentelemetry.sdk.resources import Resource
2626
from opentelemetry.sdk.trace.export import SpanExporter
@@ -268,9 +268,9 @@ def _get_otlp_exporters(endpoints: list[str]) -> list["LogRecordExporter | SpanE
268268
exporters: list["LogRecordExporter | SpanExporter | MetricExporter"] = []
269269

270270
for endpoint in endpoints:
271-
exporters.append(OTLPLogExporter(endpoint=endpoint)) # type: ignore[arg-type]
272-
exporters.append(OTLPSpanExporter(endpoint=endpoint)) # type: ignore[arg-type]
273-
exporters.append(OTLPMetricExporter(endpoint=endpoint)) # type: ignore[arg-type]
271+
exporters.append(OTLPLogExporter(endpoint=endpoint))
272+
exporters.append(OTLPSpanExporter(endpoint=endpoint))
273+
exporters.append(OTLPMetricExporter(endpoint=endpoint))
274274
return exporters
275275

276276

@@ -493,8 +493,7 @@ def _configure_providers(self, exporters: list["LogRecordExporter | MetricExport
493493
"""Configure tracing, logging, events and metrics with the provided exporters."""
494494
from opentelemetry._logs import set_logger_provider
495495
from opentelemetry.sdk._logs import LoggerProvider, LoggingHandler
496-
from opentelemetry.sdk._logs._internal.export import LogRecordExporter
497-
from opentelemetry.sdk._logs.export import BatchLogRecordProcessor
496+
from opentelemetry.sdk._logs.export import BatchLogRecordProcessor, LogRecordExporter
498497
from opentelemetry.sdk.metrics import MeterProvider
499498
from opentelemetry.sdk.metrics.export import MetricExporter, PeriodicExportingMetricReader
500499
from opentelemetry.sdk.metrics.view import DropAggregation, View
@@ -522,7 +521,7 @@ def _configure_providers(self, exporters: list["LogRecordExporter | MetricExport
522521
logger_provider.add_log_record_processor(BatchLogRecordProcessor(exporter))
523522
should_add_console_exporter = False
524523
if should_add_console_exporter:
525-
from opentelemetry.sdk._logs._internal.export import ConsoleLogExporter
524+
from opentelemetry.sdk._logs.export import ConsoleLogExporter
526525

527526
logger_provider.add_log_record_processor(BatchLogRecordProcessor(ConsoleLogExporter()))
528527

0 commit comments

Comments
 (0)