agenta 0.48.8__tar.gz → 0.48.9__tar.gz

{agenta-0.48.8 → agenta-0.48.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: agenta
-Version: 0.48.8
+Version: 0.48.9
 Summary: The SDK for agenta is an open-source LLMOps platform.
 Keywords: LLMOps,LLM,evaluation,prompt engineering
 Author: Mahmoud Mabrouk

{agenta-0.48.8 → agenta-0.48.9}/agenta/sdk/decorators/tracing.py

@@ -5,7 +5,6 @@ from itertools import chain
 from inspect import iscoroutinefunction, getfullargspec
 
 from opentelemetry import baggage
-from opentelemetry.trace import NonRecordingSpan
 from opentelemetry.context import attach, detach, get_current
 from opentelemetry.baggage import set_baggage, get_all
 
@@ -13,7 +12,6 @@ from agenta.sdk.utils.exceptions import suppress
 from agenta.sdk.context.tracing import tracing_context
 from agenta.sdk.tracing.conventions import parse_span_kind
 
-
 from agenta.sdk.utils.logging import get_module_logger
 
 import agenta as ag
@@ -62,7 +60,7 @@ class instrument:  # pylint: disable=invalid-name
                     name=func.__name__,
                     kind=self.kind,
                     context=ctx,
-                ) as span:
+                ):
                     self._set_link()
 
                     self._pre_instrument(func, *args, **kwargs)
@@ -90,7 +88,7 @@ class instrument:  # pylint: disable=invalid-name
                     name=func.__name__,
                     kind=self.kind,
                     context=ctx,
-                ) as span:
+                ):
                     self._set_link()
 
                     self._pre_instrument(func, *args, **kwargs)

{agenta-0.48.8 → agenta-0.48.9}/agenta/sdk/tracing/processors.py

@@ -34,16 +34,12 @@ class TraceProcessor(SpanProcessor):
         self.references = references or dict()
         self.inline = inline is True
 
-        # --- INLINE
-        if self.inline:
-            self._registry = dict()
-            self._exporter = span_exporter
-            self._spans: Dict[int, List[ReadableSpan]] = dict()
-        # --- INLINE
+        self._registry = dict()
+        self._exporter = span_exporter
+        self._spans: Dict[int, List[ReadableSpan]] = dict()
 
         # --- DISTRIBUTED
-        else:
-            # Use composition instead of inheritance to avoid relying on BatchSpanProcessor internals
+        if not self.inline:
             self._delegate = BatchSpanProcessor(
                 span_exporter,
                 max_queue_size or _DEFAULT_MAX_QUEUE_SIZE,
@@ -69,48 +65,39 @@ class TraceProcessor(SpanProcessor):
                 if _key in [_.value for _ in Reference.__members__.values()]:
                     span.set_attribute(key, baggage[key])
 
-        # --- INLINE
-        if self.inline:
-            if span.context.trace_id not in self._registry:
-                self._registry[span.context.trace_id] = dict()
+        trace_id = span.context.trace_id
+        span_id = span.context.span_id
 
-            self._registry[span.context.trace_id][span.context.span_id] = True
-        # --- INLINE
+        self._registry.setdefault(trace_id, {})
+        self._registry[trace_id][span_id] = True
 
     def on_end(
         self,
         span: ReadableSpan,
     ):
-        # --- INLINE
-        if self.inline:
-            if span.context.trace_id not in self._spans:
-                self._spans[span.context.trace_id] = list()
-
-            self._spans[span.context.trace_id].append(span)
+        trace_id = span.context.trace_id
+        span_id = span.context.span_id
 
-            del self._registry[span.context.trace_id][span.context.span_id]
+        self._spans.setdefault(trace_id, []).append(span)
+        self._registry.setdefault(trace_id, {})
+        self._registry[trace_id].pop(span_id, None)
 
-            if len(self._registry[span.context.trace_id]) == 0:
-                self.export(span.context.trace_id)
-        # --- INLINE
-
-        # --- DISTRIBUTED
-        else:
-            self._delegate.on_end(span)
-        # --- DISTRIBUTED
+        if not self._registry[trace_id]:
+            spans = self._spans.pop(trace_id, [])
+            self._registry.pop(trace_id, None)
 
-    def export(
-        self,
-        trace_id: int,
-    ):
-        # --- INLINE
-        if self.inline:
-            spans = self._spans[trace_id]
+            # --- INLINE
+            if self.inline:
+                self._exporter.export(spans)
+            # --- INLINE
 
-            self._exporter.export(spans)
+            # --- DISTRIBUTED
+            else:
+                for span in spans:
+                    self._delegate.on_end(span)
 
-            del self._spans[trace_id]
-        # --- INLINE
+                self._delegate.force_flush()
+            # --- DISTRIBUTED
 
     def force_flush(
         self,

{agenta-0.48.8 → agenta-0.48.9}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "agenta"
-version = "0.48.8"
+version = "0.48.9"
 description = "The SDK for agenta is an open-source LLMOps platform."
 readme = "README.md"
 authors = ["Mahmoud Mabrouk <mahmoud@agenta.ai>"]