fix the blocking delay fix

Agenta-AI · Nov 1, 2024 · 8684e40 · 8684e40
1 parent 98a7dc3
commit 8684e40
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/agenta-cli/agenta/sdk/tracing/processors.py b/agenta-cli/agenta/sdk/tracing/processors.py
@@ -50,7 +50,6 @@ def on_start(
 
         if span.context.trace_id not in self._registry:
             self._registry[span.context.trace_id] = dict()
-            self.spans[span.context.trace_id] = list()
 
         self._registry[span.context.trace_id][span.context.span_id] = True
 
@@ -60,7 +59,11 @@ def on_end(
     ):
         super().on_end(span)
 
+        if span.context.trace_id not in self.spans:
+            self.spans[span.context.trace_id] = list()
+
         self.spans[span.context.trace_id].append(span)
+
         del self._registry[span.context.trace_id][span.context.span_id]
 
         if self.is_ready(span.get_span_context().trace_id):