agenta 0.27.0a8__tar.gz → 0.27.0a9__tar.gz

{agenta-0.27.0a8 → agenta-0.27.0a9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: agenta
-Version: 0.27.0a8
+Version: 0.27.0a9
 Summary: The SDK for agenta is an open-source LLMOps platform.
 Home-page: https://agenta.ai
 Keywords: LLMOps,LLM,evaluation,prompt engineering

{agenta-0.27.0a8 → agenta-0.27.0a9}/agenta/sdk/tracing/exporters.py

@@ -38,13 +38,23 @@ class InlineTraceExporter(SpanExporter):
     def force_flush(self, timeout_millis: int = 30000) -> bool:
         return True
 
+    def is_ready(
+        self,
+        trace_id: int,
+    ) -> bool:
+        is_ready = trace_id in self._registry
+
+        return is_ready
+
     def fetch(
         self,
         trace_id: int,
     ) -> List[ReadableSpan]:
+
         trace = self._registry.get(trace_id, [])
 
-        del self._registry[trace_id]
+        if trace_id in self._registry:
+            del self._registry[trace_id]
 
         return trace
 

{agenta-0.27.0a8 → agenta-0.27.0a9}/agenta/sdk/tracing/processors.py

@@ -1,4 +1,4 @@
-from typing import Optional, Dict
+from typing import Optional, Dict, List
 
 from opentelemetry.context import Context
 from opentelemetry.sdk.trace import Span
@@ -7,6 +7,7 @@ from opentelemetry.sdk.trace.export import (
     ReadableSpan,
     BatchSpanProcessor,
     _DEFAULT_MAX_QUEUE_SIZE,
+    _DEFAULT_MAX_EXPORT_BATCH_SIZE,
 )
 
 from agenta.sdk.utils.logging import log
@@ -15,6 +16,7 @@ from agenta.sdk.utils.logging import log
 
 
 class TraceProcessor(BatchSpanProcessor):
+
     def __init__(
         self,
         span_exporter: SpanExporter,
@@ -27,14 +29,15 @@ class TraceProcessor(BatchSpanProcessor):
         super().__init__(
             span_exporter,
             _DEFAULT_MAX_QUEUE_SIZE,
-            60 * 60 * 1000,  # 1 hour
-            _DEFAULT_MAX_QUEUE_SIZE,
+            12 * 60 * 60 * 1000,  # 12 hours
+            _DEFAULT_MAX_EXPORT_BATCH_SIZE,
             500,  # < 1 second (0.5 seconds)
         )
 
         self._registry = dict()
         self._exporter = span_exporter
         self.references = references or dict()
+        self.spans: Dict[int, List[ReadableSpan]] = dict()
 
     def on_start(
         self,
@@ -55,12 +58,32 @@ class TraceProcessor(BatchSpanProcessor):
         self,
         span: ReadableSpan,
     ):
-        super().on_end(span)
+        if self.done:
+            return
+
+        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):
-            self.force_flush()
+        if len(self._registry[span.context.trace_id]) == 0:
+            self.export(span.context.trace_id)
+
+    def export(
+        self,
+        trace_id: int,
+    ):
+        spans = self.spans[trace_id]
+
+        for span in spans:
+            self.queue.appendleft(span)
+
+        with self.condition:
+            self.condition.notify()
+
+        del self.spans[trace_id]
 
     def force_flush(
         self,
@@ -77,7 +100,12 @@ class TraceProcessor(BatchSpanProcessor):
         self,
         trace_id: Optional[int] = None,
     ) -> bool:
-        is_ready = not len(self._registry.get(trace_id, {}))
+        is_ready = True
+
+        try:
+            is_ready = self._exporter.is_ready(trace_id)
+        except:  # pylint: disable=bare-except
+            pass
 
         return is_ready
 

{agenta-0.27.0a8 → agenta-0.27.0a9}/agenta/sdk/tracing/tracing.py

@@ -1,6 +1,5 @@
 from typing import Optional, Any, Dict
 from enum import Enum
-from uuid import UUID
 
 from httpx import get as check
 
@@ -156,15 +155,6 @@ class Tracing(metaclass=Singleton):
 
             for key in refs.keys():
                 if key in [_.value for _ in Reference.__members__.values()]:
-                    # TYPE AND FORMAT CHECKING
-                    if key.endswith(".id"):
-                        try:
-                            refs[key] = str(UUID(refs[key]))
-                        except:  # pylint: disable=bare-except
-                            refs[key] = None
-
-                    refs[key] = str(refs[key])
-
                     # ADD REFERENCE TO THIS SPAN
                     span.set_attribute(
                         key.value if isinstance(key, Enum) else key,

{agenta-0.27.0a8 → agenta-0.27.0a9}/pyproject.toml

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