|
1 | 1 | # Copyright (c) Microsoft Corporation. |
2 | 2 | # Licensed under the MIT License. |
3 | 3 |
|
| 4 | +import os |
4 | 5 | import logging |
5 | 6 | import threading |
6 | 7 | from collections.abc import Callable |
|
10 | 11 | from opentelemetry.sdk.resources import SERVICE_NAME, SERVICE_NAMESPACE, Resource |
11 | 12 | from opentelemetry.sdk.trace import TracerProvider |
12 | 13 | from opentelemetry.sdk.trace.export import ConsoleSpanExporter |
| 14 | +from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter |
13 | 15 |
|
14 | 16 | from .exporters.agent365_exporter import _Agent365Exporter |
15 | 17 | from .exporters.agent365_exporter_options import Agent365ExporterOptions |
@@ -154,18 +156,27 @@ def _configure_internal( |
154 | 156 | "max_export_batch_size": exporter_options.max_export_batch_size, |
155 | 157 | } |
156 | 158 |
|
| 159 | + exporter = None |
157 | 160 | if is_agent365_exporter_enabled() and exporter_options.token_resolver is not None: |
158 | 161 | exporter = _Agent365Exporter( |
159 | 162 | token_resolver=exporter_options.token_resolver, |
160 | 163 | cluster_category=exporter_options.cluster_category, |
161 | 164 | use_s2s_endpoint=exporter_options.use_s2s_endpoint, |
162 | 165 | suppress_invoke_agent_input=suppress_invoke_agent_input, |
163 | 166 | ) |
| 167 | + |
164 | 168 | else: |
165 | | - exporter = ConsoleSpanExporter() |
166 | | - self._logger.warning( |
167 | | - "is_agent365_exporter_enabled() not enabled or token_resolver not set.Falling back to console exporter." |
168 | | - ) |
| 169 | + |
| 170 | + if os.environ.get("OTEL_EXPORTER_OTLP_ENDPOINT"): |
| 171 | + exporter = OTLPSpanExporter() |
| 172 | + self._logger.warning( |
| 173 | + "is_agent365_exporter_enabled() not enabled or token_resolver not set. Falling back to OTLP exporter." |
| 174 | + ) |
| 175 | + else: |
| 176 | + exporter = ConsoleSpanExporter() |
| 177 | + self._logger.warning( |
| 178 | + "is_agent365_exporter_enabled() not enabled or token_resolver not set.Falling back to console exporter." |
| 179 | + ) |
169 | 180 |
|
170 | 181 | # Add span processors |
171 | 182 |
|
|
0 commit comments