agenta 0.47.0__py3-none-any.whl → 0.48.1__py3-none-any.whl

agenta/sdk/assets.py

@@ -24,14 +24,15 @@ supported_llm_models = {
         "deepinfra/mistralai/Mistral-7B-Instruct-v0.1",
     ],
     "Gemini": [
+        "gemini/gemini-2.5-flash-preview-05-20",
         "gemini/gemini-2.5-flash-preview-04-17",
-        "gemini/gemini-2.5-pro-exp-03-25",
         "gemini/gemini-2.0-flash-001",
+        "gemini/gemini-2.5-pro-preview-05-06",
         "gemini/gemini-2.0-flash-lite-preview-02-05",
         "gemini/gemini-1.5-pro-latest",
+        "gemini/gemini-2.0-flash-lite",
         "gemini/gemini-1.5-flash",
         "gemini/gemini-1.5-flash-8b",
-        "gemini/gemma-3-27b-it",
     ],
     "Groq": [
         "groq/deepseek-r1-distill-llama-70b",

agenta/sdk/tracing/processors.py

@@ -20,7 +20,7 @@ from agenta.sdk.tracing.conventions import Reference
 log = get_module_logger(__name__)
 
 
-class TraceProcessor(BatchSpanProcessor):
+class TraceProcessor(SpanProcessor):
     def __init__(
         self,
         span_exporter: SpanExporter,
@@ -31,14 +31,6 @@ class TraceProcessor(BatchSpanProcessor):
         max_export_batch_size: int = None,
         export_timeout_millis: float = None,
     ):
-        super().__init__(
-            span_exporter,
-            _DEFAULT_MAX_QUEUE_SIZE,
-            5 * 1000 if inline else _DEFAULT_SCHEDULE_DELAY_MILLIS,  # < 5 seconds
-            _DEFAULT_MAX_EXPORT_BATCH_SIZE,
-            500 if inline else _DEFAULT_EXPORT_TIMEOUT_MILLIS,  # < 1 second
-        )
-
         self.references = references or dict()
         self.inline = inline is True
 
@@ -49,6 +41,18 @@ class TraceProcessor(BatchSpanProcessor):
             self._spans: Dict[int, List[ReadableSpan]] = dict()
         # --- INLINE
 
+        # --- DISTRIBUTED
+        else:
+            # Use composition instead of inheritance to avoid relying on BatchSpanProcessor internals
+            self._delegate = BatchSpanProcessor(
+                span_exporter,
+                max_queue_size or _DEFAULT_MAX_QUEUE_SIZE,
+                schedule_delay_millis or _DEFAULT_SCHEDULE_DELAY_MILLIS,
+                max_export_batch_size or _DEFAULT_MAX_EXPORT_BATCH_SIZE,
+                export_timeout_millis or _DEFAULT_EXPORT_TIMEOUT_MILLIS,
+            )
+        # --- DISTRIBUTED
+
     def on_start(
         self,
         span: Span,
@@ -79,9 +83,6 @@ class TraceProcessor(BatchSpanProcessor):
     ):
         # --- INLINE
         if self.inline:
-            if self.done:
-                return
-
             if span.context.trace_id not in self._spans:
                 self._spans[span.context.trace_id] = list()
 
@@ -95,7 +96,7 @@ class TraceProcessor(BatchSpanProcessor):
 
         # --- DISTRIBUTED
         else:
-            super().on_end(span)
+            self._delegate.on_end(span)
         # --- DISTRIBUTED
 
     def export(
@@ -106,11 +107,7 @@ class TraceProcessor(BatchSpanProcessor):
         if self.inline:
             spans = self._spans[trace_id]
 
-            for span in spans:
-                self.queue.appendleft(span)
-
-            with self.condition:
-                self.condition.notify()
+            self._exporter.export(spans)
 
             del self._spans[trace_id]
         # --- INLINE
@@ -119,11 +116,35 @@ class TraceProcessor(BatchSpanProcessor):
         self,
         timeout_millis: int = None,
     ) -> bool:
-        ret = super().force_flush(timeout_millis)
+        # --- INLINE
+        if self.inline:
+            try:
+                ret = self._exporter.force_flush(timeout_millis)
+            except:  # pylint: disable=bare-except
+                ret = True
+        # --- INLINE
+
+        # --- DISTRIBUTED
+        else:
+            ret = self._delegate.force_flush(timeout_millis)
+        # --- DISTRIBUTED
 
         if not ret:
             log.warning("Agenta - Skipping export due to timeout.")
 
+        return ret
+
+    def shutdown(self) -> None:
+        # --- INLINE
+        if self.inline:
+            self._exporter.shutdown()
+        # --- INLINE
+
+        # --- DISTRIBUTED
+        else:
+            self._delegate.shutdown()
+        # --- DISTRIBUTED
+
     def is_ready(
         self,
         trace_id: Optional[int] = None,

agenta/sdk/utils/preinit.py

@@ -1,10 +1,4 @@
 from typing import Any, Optional
-from dotenv import load_dotenv
-
-# from .context import setup_db
-
-load_dotenv()
-# setup_db()
 
 
 class PreInitObject:

{agenta-0.47.0.dist-info → agenta-0.48.1.dist-info}/METADATA

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

{agenta-0.47.0.dist-info → agenta-0.48.1.dist-info}/RECORD

@@ -200,7 +200,7 @@ agenta/config.py,sha256=0VrTqduB4g8Mt_Ll7ffFcEjKF5qjTUIxmUtTPW2ygWw,653
 agenta/config.toml,sha256=sIORbhnyct2R9lJrquxhNL4pHul3O0R7iaipCoja5MY,193
 agenta/sdk/__init__.py,sha256=m5UxAteSVzJtYhSsmUmdGFsyFdZO6jlfvOpJi_4mK_E,2118
 agenta/sdk/agenta_init.py,sha256=-0m0IH6K2Nic8Vftt5ENdRxpChhsaihKLtMWn8jzoJs,6610
-agenta/sdk/assets.py,sha256=viDrSsQfVnMEVkzxRVcQ3WsOuhD5HPfaslLwm9uV0MA,8271
+agenta/sdk/assets.py,sha256=h-0OD9K6GOMUDkWOO30Je4hCMhMg2UzI1ccgkN3RICc,8331
 agenta/sdk/context/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agenta/sdk/context/exporting.py,sha256=16X8fgMhl58gehSlqANX97FiKxx4TkGiG4d2B0-7ZX0,516
 agenta/sdk/context/routing.py,sha256=FEsjw8EttI1SMyUo96ptcUsvHJnhoKwdr1szlkxxJNU,598
@@ -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=vDzQItQnDG_bL5rWRZjh-8DOLcfhXJcMVCAbStiTVDA,5102
+agenta/sdk/tracing/processors.py,sha256=wweg31mAWiytFPsjPP3zTDqFidjLzyFXNVdXBt1o3Sw,5748
 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
@@ -247,9 +247,9 @@ agenta/sdk/utils/exceptions.py,sha256=Dq3lh5Ug7IhF4nvYRXXlwhgxEuByOnT1w0uVMZx1ea
 agenta/sdk/utils/globals.py,sha256=9ixKS8aI6ZWgsjT8WngvTg4dsnP2cKErNJngHxzdK9U,256
 agenta/sdk/utils/helpers.py,sha256=gfvQ-2-M_tZonFJtSdC0F7CNFcHyUG1CTgGQNwxRt2g,1169
 agenta/sdk/utils/logging.py,sha256=j4NzpFk2ilOM10sRBOxCjmansDHSx6HwMV8IAEreRb8,8386
-agenta/sdk/utils/preinit.py,sha256=YlJL7RLfel0R7DFp-jK7OV-z4ZIQJM0oupYlk7g8b5o,1278
+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.47.0.dist-info/METADATA,sha256=2LjpV1wQBIt5iJKrEUpF-2-uA9zTtNENFrDgem725DE,31433
-agenta-0.47.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-agenta-0.47.0.dist-info/RECORD,,
+agenta-0.48.1.dist-info/METADATA,sha256=ZMcNgq1u-xqeKbmDqAs9bdxt7_ztPw7TNd_5g2QXWWg,31433
+agenta-0.48.1.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+agenta-0.48.1.dist-info/RECORD,,