agenta 0.27.0a7__py3-none-any.whl → 0.27.0a9__py3-none-any.whl

agenta/sdk/agenta_init.py

@@ -88,7 +88,7 @@ class AgentaSingleton:
         )
 
         self.tracing = Tracing(
-            url=f"{self.host}/api/observability/v1/oltp/traces",  # type: ignore
+            url=f"{self.host}/api/observability/v1/otlp/traces",  # type: ignore
         )
 
         self.tracing.configure(

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/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/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.0a7.dist-info → agenta-0.27.0a9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: agenta
-Version: 0.27.0a7
+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.0a7.dist-info → agenta-0.27.0a9.dist-info}/RECORD

@@ -133,7 +133,7 @@ agenta/docker/docker-assets/lambda_function.py,sha256=h4UZSSfqwpfsCgERv6frqwm_4J
 agenta/docker/docker-assets/main.py,sha256=7MI-21n81U7N7A0GxebNi0cmGWtJKcR2sPB6FcH2QfA,251
 agenta/docker/docker_utils.py,sha256=kO1q2_IR0fEAo4M-2Pt_v-zC7GxxnkLogjKFhU869Ps,3555
 agenta/sdk/__init__.py,sha256=IopYEonqvpdGXsAYh2wQ4Ai_2YzsY7aF88wTzrclV4M,1208
-agenta/sdk/agenta_init.py,sha256=3TVZFGrcD8GzhVRQSA36PepQLr313RQCVVsdinzP_ok,10308
+agenta/sdk/agenta_init.py,sha256=vQcaWvvrMR6xMNJvX6V-9eNIvx3_JsjPA6xcxiahX30,10308
 agenta/sdk/assets.py,sha256=Zv4i8MVUSB3jMODQon1mzJtYxuntmrCNjLGk8f-2fls,2856
 agenta/sdk/client.py,sha256=trKyBOYFZRk0v5Eptxvh87yPf50Y9CqY6Qgv4Fy-VH4,2142
 agenta/sdk/config_manager.py,sha256=HFOJpJKBkhlA0C-KPuxb4-bHNZeZqdpmx_beoX4lQw8,7997
@@ -150,11 +150,11 @@ agenta/sdk/tracing/__init__.py,sha256=rQNe5-zT5Kt7_CDhq-lnUIi1EYTBVzVf_MbfcIxVD9
 agenta/sdk/tracing/attributes.py,sha256=zh8JQZSeYCLBeIRSopKJx6QQ-WEgw08Cr64DS_WOcT8,3833
 agenta/sdk/tracing/context.py,sha256=PSJdhcaOXSMAuGUBySpLKPKyx8duF3TJzhUEk2ufqPc,777
 agenta/sdk/tracing/conventions.py,sha256=JBtznBXZ3aRkGKkLl7cPwdMNh3w1G-H2Ta2YrAxbr38,950
-agenta/sdk/tracing/exporters.py,sha256=LzNGjOARwRrJ5gTFjdG9ziZlqnA0WeMjpkUfpsl8BSs,1383
+agenta/sdk/tracing/exporters.py,sha256=VLebUsoq29mcBccpwXvNHOZWwi2Kq3oZvUsKMsb9CGE,1569
 agenta/sdk/tracing/inline.py,sha256=0ZLy89k5wkmTUaz69vqc3vP51ZdqmLGMj3Qz2o3hieE,32856
-agenta/sdk/tracing/processors.py,sha256=uqH8AKdRJkv-MY5FcTD-Qrrr0ARMlAAsImPPKolteQ0,2444
+agenta/sdk/tracing/processors.py,sha256=oyZkIbFBlBLEyl757TY1yykdDm40vH2z1_t-itOC0j0,3116
 agenta/sdk/tracing/spans.py,sha256=nqUOjjirBxB8Eacv8Qj4Ra_6rknGi3lbJdNyKmk5ODQ,3707
-agenta/sdk/tracing/tracing.py,sha256=S-7rt1LMHCF3YzV1XzAnhdkeocCffj9CCwlHScU7mqU,7350
+agenta/sdk/tracing/tracing.py,sha256=laInX27JUBCS_KiOqMax2Ja5UJ9X72tY5bufLbcYebo,6990
 agenta/sdk/types.py,sha256=B7C7Jpeq0CpmLwmN1dmpggJsZX8WpxyyM42PkdSsx3s,5669
 agenta/sdk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agenta/sdk/utils/costs.py,sha256=i8C7ud__pThLS55XkN4YW8czXtGeXr2mx7jjcOFeiXg,5955
@@ -179,7 +179,7 @@ agenta/templates/simple_prompt/app.py,sha256=kODgF6lhzsaJPdgL5b21bUki6jkvqjWZzWR
 agenta/templates/simple_prompt/env.example,sha256=g9AE5bYcGPpxawXMJ96gh8oenEPCHTabsiOnfQo3c5k,70
 agenta/templates/simple_prompt/requirements.txt,sha256=ywRglRy7pPkw8bljmMEJJ4aOOQKrt9FGKULZ-DGkoBU,23
 agenta/templates/simple_prompt/template.toml,sha256=DQBtRrF4GU8LBEXOZ-GGuINXMQDKGTEG5y37tnvIUIE,60
-agenta-0.27.0a7.dist-info/METADATA,sha256=ztCgqJSX5gG6RVtrgR_84TjG87fDPd8Hcjtgin-c2m4,31738
-agenta-0.27.0a7.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-agenta-0.27.0a7.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
-agenta-0.27.0a7.dist-info/RECORD,,
+agenta-0.27.0a9.dist-info/METADATA,sha256=IgUZsGVZc83d5z7nmSivXC7lrpE1lJaV3P2D6LduATI,31738
+agenta-0.27.0a9.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+agenta-0.27.0a9.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
+agenta-0.27.0a9.dist-info/RECORD,,