Skip to content

Commit 5d51d3a

Browse files
committed
fix(tracing): fix empty local tracing file
1 parent cf26b14 commit 5d51d3a

File tree

3 files changed

+9
-7
lines changed

3 files changed

+9
-7
lines changed

veadk/tracing/telemetry/exporters/inmemory_exporter.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -106,3 +106,6 @@ def __init__(self, name: str = "inmemory_exporter") -> None:
106106

107107
self._exporter = _InMemoryExporter()
108108
self.processor = _InMemorySpanProcessor(self._exporter)
109+
110+
111+
_INMEMORY_EXPORTER_INSTANCE = InMemoryExporter()

veadk/tracing/telemetry/opentelemetry_tracer.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,7 @@
3131
from veadk.tracing.telemetry.exporters.apmplus_exporter import APMPlusExporter
3232
from veadk.tracing.telemetry.exporters.base_exporter import BaseExporter
3333
from veadk.tracing.telemetry.exporters.inmemory_exporter import InMemoryExporter
34+
from veadk.tracing.telemetry.exporters.inmemory_exporter import _INMEMORY_EXPORTER_INSTANCE
3435
from veadk.utils.logger import get_logger
3536
from veadk.utils.patches import patch_google_adk_telemetry
3637

@@ -110,7 +111,7 @@ def _init_global_tracer_provider(self) -> None:
110111
f"Add span processor for exporter `{exporter.__class__.__name__}` to OpentelemetryTracer failed."
111112
)
112113

113-
self._inmemory_exporter = InMemoryExporter()
114+
self._inmemory_exporter = _INMEMORY_EXPORTER_INSTANCE
114115
if self._inmemory_exporter.processor:
115116
# make sure the in memory exporter processor is added at index 0
116117
# because we use this to record all spans

veadk/tracing/telemetry/telemetry.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@
2828
LLMAttributesParams,
2929
ToolAttributesParams,
3030
)
31+
from veadk.tracing.telemetry.exporters.inmemory_exporter import _INMEMORY_EXPORTER_INSTANCE
3132
from veadk.utils.logger import get_logger
3233

3334
logger = get_logger(__name__)
@@ -64,13 +65,10 @@ def set_common_attributes(
6465
)
6566
return
6667

67-
if isinstance(invocation_context.agent, Agent) and invocation_context.agent.tracers:
68+
if isinstance(invocation_context.agent, Agent):
6869
try:
69-
from veadk.tracing.telemetry.opentelemetry_tracer import OpentelemetryTracer
70-
71-
tracer: OpentelemetryTracer = invocation_context.agent.tracers[0] # type: ignore
72-
spans = tracer._inmemory_exporter.processor.spans # # type: ignore
73-
70+
spans = _INMEMORY_EXPORTER_INSTANCE.processor.spans # # type: ignore
71+
7472
spans_in_current_trace = [
7573
span
7674
for span in spans

0 commit comments

Comments
 (0)