agenta 0.48.0__tar.gz → 0.48.2__tar.gz

{agenta-0.48.0 → agenta-0.48.2}/PKG-INFO

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

{agenta-0.48.0 → agenta-0.48.2}/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-0.48.0 → agenta-0.48.2}/agenta/sdk/middleware/auth.py

@@ -20,7 +20,7 @@ log = get_module_logger(__name__)
 AGENTA_RUNTIME_PREFIX = getenv("AGENTA_RUNTIME_PREFIX", "")
 
 
-_CACHE_ENABLED = getenv("AGENTA_MIDDLEWARE_CACHE_ENABLED", "false").lower() in TRUTHY
+_CACHE_ENABLED = getenv("AGENTA_SERVICE_MIDDLEWARE_CACHE_ENABLED", "true").lower() in TRUTHY
 
 _ALWAYS_ALLOW_LIST = [f"{AGENTA_RUNTIME_PREFIX}/health"]
 

{agenta-0.48.0 → agenta-0.48.2}/agenta/sdk/middleware/config.py

@@ -17,7 +17,7 @@ from agenta.sdk.utils.exceptions import suppress
 import agenta as ag
 
 
-_CACHE_ENABLED = getenv("AGENTA_MIDDLEWARE_CACHE_ENABLED", "false").lower() in TRUTHY
+_CACHE_ENABLED = getenv("AGENTA_SERVICE_MIDDLEWARE_CACHE_ENABLED", "true").lower() in TRUTHY
 
 _cache = TTLLRUCache()
 

{agenta-0.48.0 → agenta-0.48.2}/agenta/sdk/middleware/vault.py

@@ -24,7 +24,7 @@ _PROVIDER_KINDS = []
 for provider_kind in StandardProviderKind.__args__[0].__args__:  # type: ignore
     _PROVIDER_KINDS.append(provider_kind)
 
-_CACHE_ENABLED = getenv("AGENTA_MIDDLEWARE_CACHE_ENABLED", "false").lower() in TRUTHY
+_CACHE_ENABLED = getenv("AGENTA_SERVICE_MIDDLEWARE_CACHE_ENABLED", "true").lower() in TRUTHY
 
 _cache = TTLLRUCache()
 

{agenta-0.48.0 → agenta-0.48.2}/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-0.48.0 → agenta-0.48.2}/agenta/sdk/utils/cache.py

@@ -5,7 +5,7 @@ from collections import OrderedDict
 from threading import Lock
 
 CACHE_CAPACITY = int(getenv("AGENTA_MIDDLEWARE_CACHE_CAPACITY", "512"))
-CACHE_TTL = int(getenv("AGENTA_MIDDLEWARE_CACHE_TTL", str(5 * 60)))  # 5 minutes
+CACHE_TTL = int(getenv("AGENTA_MIDDLEWARE_CACHE_TTL", str(60)))  # 1 minute
 
 
 class TTLLRUCache:

{agenta-0.48.0 → agenta-0.48.2}/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.48.0 → agenta-0.48.2}/pyproject.toml

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