agenta 0.48.8__py3-none-any.whl → 0.48.9__py3-none-any.whl

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/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.dist-info → agenta-0.48.9.dist-info}/METADATA

@@ -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.dist-info → agenta-0.48.9.dist-info}/RECORD

@@ -207,7 +207,7 @@ agenta/sdk/context/routing.py,sha256=FEsjw8EttI1SMyUo96ptcUsvHJnhoKwdr1szlkxxJNU
 agenta/sdk/context/tracing.py,sha256=xjErrXP1Nq1AfL-Cif1l-lNEfs12eQ3v_VCRgoKe7nY,743
 agenta/sdk/decorators/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agenta/sdk/decorators/routing.py,sha256=L3o0Q9flBfb2UIfCIx0OWzkP0W-45_GQ3-Ri8phwNlU,25678
-agenta/sdk/decorators/tracing.py,sha256=pABMVFtqD99W3ErqWDFUmymmXiXdzhdAJG01l-CTLVQ,10110
+agenta/sdk/decorators/tracing.py,sha256=6xwN9AomDND_5wL21NXFoZsdwTBuITw2I39GNqNGD-4,10044
 agenta/sdk/litellm/__init__.py,sha256=Bpz1gfHQc0MN1yolWcjifLWznv6GjHggvRGQSpxpihM,37
 agenta/sdk/litellm/litellm.py,sha256=dCjw0H_jD3L3UQ3l9SbLm5dfeIGnel6dPtQYJ78beYM,10202
 agenta/sdk/litellm/mockllm.py,sha256=PG3DF2ZjWHOtCt8WIU3qcSyXBOZspUX1FIyFMA9HuZM,902
@@ -234,7 +234,7 @@ agenta/sdk/tracing/attributes.py,sha256=DwjjOk3mGOvz0jYu8EYr3hhItvawK5GX80_Mfciq
 agenta/sdk/tracing/conventions.py,sha256=JBtznBXZ3aRkGKkLl7cPwdMNh3w1G-H2Ta2YrAxbr38,950
 agenta/sdk/tracing/exporters.py,sha256=Vs1aQLRrWvYgWxjPmlfBWTN4wmUzBREqjTdIV1XcV9o,3251
 agenta/sdk/tracing/inline.py,sha256=ShPAAjk_26I4hgrmC6y0n-I9gxB3Q4VeEYhhsLHInPU,31454
-agenta/sdk/tracing/processors.py,sha256=wweg31mAWiytFPsjPP3zTDqFidjLzyFXNVdXBt1o3Sw,5748
+agenta/sdk/tracing/processors.py,sha256=wZWF8vTEouhbGxbfPGXVQJI53CcXFByqxHiCSIzJV-4,5407
 agenta/sdk/tracing/propagation.py,sha256=EeOqDMqnh_MoEhGd1do_vy_tQBYUcoC8kpLqVoZeqg0,2561
 agenta/sdk/tracing/spans.py,sha256=nqUOjjirBxB8Eacv8Qj4Ra_6rknGi3lbJdNyKmk5ODQ,3707
 agenta/sdk/tracing/tracing.py,sha256=Yv42drWc2tVu4YugHKWYsgTSf1y7Y9boYL8akxKT4eA,9503
@@ -250,6 +250,6 @@ agenta/sdk/utils/logging.py,sha256=j4NzpFk2ilOM10sRBOxCjmansDHSx6HwMV8IAEreRb8,8
 agenta/sdk/utils/preinit.py,sha256=1TAAHhYyYnLLwvzwnf33Qwkou7tI3iITlVt-1kcyGaM,1186
 agenta/sdk/utils/singleton.py,sha256=17Ph7LGnnV8HkPjImruKita2ni03Ari5jr0jqm__4sc,312
 agenta/sdk/utils/timing.py,sha256=nZR-kudVUtKFlHuBhztgSGxj7FVnCB4Uv6sfg-1dkrQ,1556
-agenta-0.48.8.dist-info/METADATA,sha256=QAIDalUoutOgc4zUJCIGgManVrWzh0Bfs91mPIcJgv0,31456
-agenta-0.48.8.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-agenta-0.48.8.dist-info/RECORD,,
+agenta-0.48.9.dist-info/METADATA,sha256=BhII-ix_CJcWaYXCF5aVdHbm9-HlRpqxj3wr6BGavII,31456
+agenta-0.48.9.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+agenta-0.48.9.dist-info/RECORD,,