agent/agent-framework
1
0
Fork 0
mirror of https://github.com/microsoft/agent-framework.git synced 2026-06-16 21:04:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix observability module mypy error (#2641)

Browse Source
This commit is contained in:
Tao Chen
2025-12-04 11:11:05 -08:00
committed by GitHub
Unverified
parent 365e438981
commit d858c23bab
1 changed files with 6 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/packages/core/agent_framework/observability.py
+6 -7
View File
@@ -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()))