agenta 0.68.0__py3-none-any.whl → 0.70.1__py3-none-any.whl

agenta/__init__.py

@@ -1,52 +1,50 @@
 from typing import Any, Callable, Optional
 
-from .sdk.utils.preinit import PreInitObject
-
-from agenta.client import AgentaApi, AsyncAgentaApi
 import agenta.client.backend.types as client_types  # pylint: disable=wrong-import-order
+from agenta.client import AgentaApi, AsyncAgentaApi
 
-from .sdk.types import (
-    MCField,
-    DictInput,
-    MultipleChoice,
-    FloatParam,
-    IntParam,
-    MultipleChoiceParam,
-    GroupedMultipleChoiceParam,
-    MessagesInput,
-    TextParam,
-    FileInputURL,
-    BinaryParam,
-    Prompt,
-    PromptTemplate,
-)
+from .sdk import assets as assets
 
-from .sdk.agenta_init import Config, AgentaSingleton, init as _init
-from .sdk.utils.logging import get_module_logger
-from .sdk.utils.costs import calculate_token_usage
-from .sdk.tracing import Tracing, get_tracer
-from .sdk.tracing.conventions import Reference
-from .sdk.decorators.tracing import instrument
+# evaluations
+from .sdk import testsets as testsets
+from .sdk import tracer
+from .sdk.agenta_init import AgentaSingleton, Config
+from .sdk.agenta_init import init as _init
+from .sdk.context.running import workflow_mode_enabled
 from .sdk.decorators.running import (
-    workflow,
     application,
     evaluator,
+    workflow,
 )
-from .sdk.decorators.serving import route, app
-from .sdk.context.running import workflow_mode_enabled
+from .sdk.decorators.serving import app, route
+from .sdk.decorators.tracing import instrument
 from .sdk.litellm import litellm as callbacks
 from .sdk.managers.apps import AppManager
-from .sdk.managers.vault import VaultManager
-from .sdk.managers.secrets import SecretsManager
 from .sdk.managers.config import ConfigManager
-from .sdk.managers.variant import VariantManager
 from .sdk.managers.deployment import DeploymentManager
-from .sdk import assets as assets
-from .sdk import tracer
-
-# evaluations
-from .sdk import testsets as testsets
-
+from .sdk.managers.secrets import SecretsManager
+from .sdk.managers.variant import VariantManager
+from .sdk.managers.vault import VaultManager
+from .sdk.tracing import Tracing, get_tracer
+from .sdk.tracing.conventions import Reference
+from .sdk.types import (
+    BinaryParam,
+    DictInput,
+    FileInputURL,
+    FloatParam,
+    GroupedMultipleChoiceParam,
+    IntParam,
+    MCField,
+    MessagesInput,
+    MultipleChoice,
+    MultipleChoiceParam,
+    Prompt,
+    PromptTemplate,
+    TextParam,
+)
+from .sdk.utils.costs import calculate_token_usage
+from .sdk.utils.logging import get_module_logger
+from .sdk.utils.preinit import PreInitObject
 
 config = PreInitObject("agenta.config", Config)
 DEFAULT_AGENTA_SINGLETON_INSTANCE = AgentaSingleton()
@@ -88,3 +86,35 @@ def init(
 
     tracing = DEFAULT_AGENTA_SINGLETON_INSTANCE.tracing  # type: ignore
     tracer = get_tracer(tracing)
+
+
+def get_trace_url(trace_id: Optional[str] = None) -> str:
+    """
+    Build a URL to view the current trace in the Agenta UI.
+
+    Automatically extracts the trace ID from the current tracing context.
+    Can also accept an explicit trace_id if needed.
+
+    Args:
+        trace_id: Optional trace ID (hex string format). If not provided,
+                  it will be automatically extracted from the current trace context.
+
+    Returns:
+        The full URL to view the trace in the observability dashboard
+
+    Raises:
+        RuntimeError: If the SDK is not initialized, no active trace context exists,
+                      or scope info cannot be fetched
+
+    Example:
+        >>> import agenta as ag
+        >>> ag.init(api_key="xxx")
+        >>>
+        >>> @ag.instrument()
+        >>> def my_function():
+        >>>     # Get URL for the current trace
+        >>>     url = ag.tracing.get_trace_url()
+        >>>     print(url)
+        >>>     return "result"
+    """
+    return DEFAULT_AGENTA_SINGLETON_INSTANCE.tracing.get_trace_url(trace_id)

agenta/sdk/agenta_init.py

@@ -1,16 +1,15 @@
-import toml
-from os import getenv
-from typing import Optional, Callable, Any
 from importlib.metadata import version
+from os import getenv
+from typing import Any, Callable, Optional
 
-from agenta.sdk.utils.helpers import parse_url
-from agenta.sdk.utils.globals import set_global
-from agenta.sdk.utils.logging import get_module_logger
+import requests
+import toml
 from agenta.client.client import AgentaApi, AsyncAgentaApi
-
-from agenta.sdk.tracing import Tracing
 from agenta.sdk.contexts.routing import RoutingContext
-
+from agenta.sdk.tracing import Tracing
+from agenta.sdk.utils.globals import set_global
+from agenta.sdk.utils.helpers import parse_url
+from agenta.sdk.utils.logging import get_module_logger
 
 log = get_module_logger(__name__)
 
@@ -19,6 +18,7 @@ class AgentaSingleton:
     """Singleton class to save all the "global variables" for the sdk."""
 
     _instance = None
+    _initialized = False
     config = None
     tracing = None
 
@@ -26,6 +26,11 @@ class AgentaSingleton:
     async_api = None
 
     def __init__(self):
+        # Only initialize once
+        if AgentaSingleton._initialized:
+            return
+
+        AgentaSingleton._initialized = True
         self.host = None
         self.api_url = None
         self.api_key = None
@@ -33,6 +38,11 @@ class AgentaSingleton:
         self.scope_type = None
         self.scope_id = None
 
+        # Cached scope information for URL building
+        self.organization_id: Optional[str] = None
+        self.workspace_id: Optional[str] = None
+        self.project_id: Optional[str] = None
+
     def __new__(cls):
         if not cls._instance:
             cls._instance = super(AgentaSingleton, cls).__new__(cls)
@@ -70,6 +80,10 @@ class AgentaSingleton:
 
         """
 
+        # Idempotency check: if already initialized, skip re-initialization
+        if self.tracing and self.api and self.async_api:
+            return
+
         log.info("Agenta -     SDK ver: %s", version("agenta"))
 
         config = {}
@@ -100,7 +114,7 @@ class AgentaSingleton:
 
         try:
             assert _api_url and isinstance(_api_url, str), (
-                "API URL is required. Please provide a valid API URL or set AGENTA_API_URL environment variable."
+                "API URL is required. Please set AGENTA_API_URL environment variable or pass api_url parameter in ag.init()."
             )
             self.host = _host
             self.api_url = _api_url
@@ -118,6 +132,11 @@ class AgentaSingleton:
             or None  # NO FALLBACK
         )
 
+        if self.api_key is None:
+            log.error(
+                "API key is required. Please set AGENTA_API_KEY environment variable or pass api_key parameter in ag.init()."
+            )
+
         log.info("Agenta -     API URL: %s", self.api_url)
 
         self.scope_type = (
@@ -159,6 +178,65 @@ class AgentaSingleton:
             api_key=self.api_key,
         )
 
+        # Reset cached scope info on re-init
+        self.organization_id = None
+        self.workspace_id = None
+        self.project_id = None
+
+    def resolve_scopes(self) -> Optional[tuple[str, str, str]]:
+        """Fetch and cache workspace_id and project_id from the API."""
+        if (
+            self.organization_id is not None
+            and self.workspace_id is not None
+            and self.project_id is not None
+        ):
+            return
+
+        if self.api_url is None or self.api_key is None:
+            log.error("API URL or API key is not set. Please call ag.init() first.")
+            return
+
+        try:
+            response = requests.get(
+                f"{self.api_url}/projects/current",
+                headers={"Authorization": f"ApiKey {self.api_key}"},
+                timeout=10,
+            )
+            response.raise_for_status()
+
+            project_info = response.json()
+
+            if not project_info:
+                log.error(
+                    "No project context found. Please ensure your API key is valid."
+                )
+
+            self.organization_id = project_info.get("organization_id")
+            self.workspace_id = project_info.get("workspace_id")
+            self.project_id = project_info.get("project_id")
+
+            if (
+                not self.organization_id
+                and not self.workspace_id
+                or not self.project_id
+            ):
+                log.error(
+                    "Could not determine organization/workspace/project from API response."
+                )
+
+        except Exception as e:
+            log.error(f"Failed to fetch scope information: {e}")
+            return
+
+        if self.organization_id and self.workspace_id and self.project_id:
+            return (
+                self.organization_id,
+                self.workspace_id,
+                self.project_id,
+            )
+
+        return None
+
 
 class Config:
     def __init__(

agenta/sdk/assets.py

@@ -28,6 +28,7 @@ supported_llm_models = {
     ],
     "gemini": [
         "gemini/gemini-3-pro-preview",
+        "gemini/gemini-3-flash-preview",
         "gemini/gemini-2.5-pro",
         "gemini/gemini-2.5-pro-preview-05-06",
         "gemini/gemini-2.5-flash",
@@ -64,21 +65,26 @@ supported_llm_models = {
         "mistral/mistral-large-latest",
     ],
     "openai": [
-        "gpt-5",
+        "gpt-5.2-pro",
+        "gpt-5.2-chat-latest",
+        "gpt-5.2",
+        "gpt-5.1-chat-latest",
         "gpt-5.1",
-        "gpt-5-mini",
+        "gpt-5-pro",
         "gpt-5-nano",
+        "gpt-5-mini",
+        "gpt-5",
+        "o4-mini",
         "gpt-4.5-preview",
-        "gpt-3.5-turbo-1106",
-        "gpt-3.5-turbo",
-        "gpt-4",
-        "gpt-4o",
+        "gpt-4.1-nano",
+        "gpt-4.1-mini",
+        "gpt-4.1",
         "gpt-4o-mini",
+        "gpt-4o",
         "gpt-4-1106-preview",
-        "gpt-4.1",
-        "gpt-4.1-mini",
-        "gpt-4.1-nano",
-        "o4-mini",
+        "gpt-4",
+        "gpt-3.5-turbo-1106",
+        "gpt-3.5-turbo",
     ],
     "openrouter": [
         "openrouter/qwen/qwen3-235b-a22b",

agenta/sdk/litellm/litellm.py

@@ -166,24 +166,27 @@ def litellm_handler():
                 namespace="metrics.unit.costs",
             )
 
+            # Handle both dict and object attribute access for usage, and safely handle None
+            usage = getattr(response_obj, "usage", None)
+            if isinstance(usage, dict):
+                prompt_tokens = usage.get("prompt_tokens")
+                completion_tokens = usage.get("completion_tokens")
+                total_tokens = usage.get("total_tokens")
+            elif usage is not None:
+                prompt_tokens = getattr(usage, "prompt_tokens", None)
+                completion_tokens = getattr(usage, "completion_tokens", None)
+                total_tokens = getattr(usage, "total_tokens", None)
+            else:
+                prompt_tokens = completion_tokens = total_tokens = None
+
             span.set_attributes(
                 attributes=(
                     {
-                        "prompt": (
-                            float(response_obj.usage.prompt_tokens)
-                            if response_obj.usage.prompt_tokens
-                            else None
-                        ),
-                        "completion": (
-                            float(response_obj.usage.completion_tokens)
-                            if response_obj.usage.completion_tokens
-                            else None
-                        ),
-                        "total": (
-                            float(response_obj.usage.total_tokens)
-                            if response_obj.usage.total_tokens
-                            else None
-                        ),
+                        "prompt": float(prompt_tokens) if prompt_tokens else None,
+                        "completion": float(completion_tokens)
+                        if completion_tokens
+                        else None,
+                        "total": float(total_tokens) if total_tokens else None,
                     }
                 ),
                 namespace="metrics.unit.tokens",
@@ -300,24 +303,29 @@ def litellm_handler():
                 namespace="metrics.unit.costs",
             )
 
+            # Handle both dict and object attribute access for usage
+            usage = getattr(response_obj, "usage", None)
+            if usage is None:
+                prompt_tokens = None
+                completion_tokens = None
+                total_tokens = None
+            elif isinstance(usage, dict):
+                prompt_tokens = usage.get("prompt_tokens")
+                completion_tokens = usage.get("completion_tokens")
+                total_tokens = usage.get("total_tokens")
+            else:
+                prompt_tokens = getattr(usage, "prompt_tokens", None)
+                completion_tokens = getattr(usage, "completion_tokens", None)
+                total_tokens = getattr(usage, "total_tokens", None)
+
             span.set_attributes(
                 attributes=(
                     {
-                        "prompt": (
-                            float(response_obj.usage.prompt_tokens)
-                            if response_obj.usage.prompt_tokens
-                            else None
-                        ),
-                        "completion": (
-                            float(response_obj.usage.completion_tokens)
-                            if response_obj.usage.completion_tokens
-                            else None
-                        ),
-                        "total": (
-                            float(response_obj.usage.total_tokens)
-                            if response_obj.usage.total_tokens
-                            else None
-                        ),
+                        "prompt": float(prompt_tokens) if prompt_tokens else None,
+                        "completion": float(completion_tokens)
+                        if completion_tokens
+                        else None,
+                        "total": float(total_tokens) if total_tokens else None,
                     }
                 ),
                 namespace="metrics.unit.tokens",

agenta/sdk/middleware/auth.py

@@ -17,14 +17,12 @@ import agenta as ag
 
 log = get_module_logger(__name__)
 
-AGENTA_RUNTIME_PREFIX = getenv("AGENTA_RUNTIME_PREFIX", "")
-
 
 _CACHE_ENABLED = (
     getenv("AGENTA_SERVICE_MIDDLEWARE_CACHE_ENABLED", "true").lower() in TRUTHY
 )
 
-_ALWAYS_ALLOW_LIST = [f"{AGENTA_RUNTIME_PREFIX}/health"]
+_ALWAYS_ALLOW_LIST = ["/health"]
 
 _cache = TTLLRUCache()
 
@@ -64,7 +62,7 @@ class AuthHTTPMiddleware(BaseHTTPMiddleware):
 
     async def dispatch(self, request: Request, call_next: Callable):
         try:
-            if request.url.path in _ALWAYS_ALLOW_LIST:
+            if _strip_service_prefix(request.url.path) in _ALWAYS_ALLOW_LIST:
                 request.state.auth = {}
 
             else:
@@ -253,3 +251,20 @@ class AuthHTTPMiddleware(BaseHTTPMiddleware):
                 status_code=500,
                 content=f"Could not verify credentials: unexpected error - {str(exc)}. Please try again later or contact support if the issue persists.",
             ) from exc
+
+
+def _strip_service_prefix(path: str) -> str:
+    if not path.startswith("/services/"):
+        return path
+
+    parts = path.split("/", 3)
+    if len(parts) < 4:
+        return "/"
+
+    service_name = parts[2]
+    remainder = parts[3]
+
+    if not service_name or not remainder or remainder.startswith("/"):
+        return path
+
+    return f"/{remainder}"

agenta/sdk/middleware/otel.py

@@ -3,9 +3,6 @@ from typing import Callable
 from starlette.middleware.base import BaseHTTPMiddleware
 from fastapi import Request, FastAPI
 
-from opentelemetry.baggage.propagation import W3CBaggagePropagator
-from opentelemetry.trace.propagation.tracecontext import TraceContextTextMapPropagator
-
 from agenta.sdk.utils.exceptions import suppress
 from agenta.sdk.tracing.propagation import extract
 

agenta/sdk/middleware/vault.py

@@ -21,11 +21,9 @@ import agenta as ag
 log = get_module_logger(__name__)
 
 
-AGENTA_RUNTIME_PREFIX = getenv("AGENTA_RUNTIME_PREFIX", "")
-
 _ALWAYS_ALLOW_LIST = [
-    f"{AGENTA_RUNTIME_PREFIX}/health",
-    f"{AGENTA_RUNTIME_PREFIX}/openapi.json",
+    "/health",
+    "/openapi.json",
 ]
 
 _PROVIDER_KINDS = [
@@ -116,7 +114,7 @@ class VaultMiddleware(BaseHTTPMiddleware):
         allow_secrets = True
 
         try:
-            if not request.url.path in _ALWAYS_ALLOW_LIST:
+            if _strip_service_prefix(request.url.path) not in _ALWAYS_ALLOW_LIST:
                 await self._allow_local_secrets(credentials)
 
             for provider_kind in _PROVIDER_KINDS:
@@ -331,3 +329,20 @@ class VaultMiddleware(BaseHTTPMiddleware):
                 status_code=500,
                 content=f"Could not verify credentials: unexpected error - {str(exc)}. Please try again later or contact support if the issue persists.",
             ) from exc
+
+
+def _strip_service_prefix(path: str) -> str:
+    if not path.startswith("/services/"):
+        return path
+
+    parts = path.split("/", 3)
+    if len(parts) < 4:
+        return "/"
+
+    service_name = parts[2]
+    remainder = parts[3]
+
+    if not service_name or not remainder or remainder.startswith("/"):
+        return path
+
+    return f"/{remainder}"

agenta/sdk/middlewares/running/vault.py

@@ -127,7 +127,7 @@ class VaultMiddleware:
         request: WorkflowServiceRequest,
         call_next: Callable[[WorkflowServiceRequest], Any],
     ):
-        api_url = f"{ag.DEFAULT_AGENTA_SINGLETON_INSTANCE.host}/api"
+        api_url = ag.DEFAULT_AGENTA_SINGLETON_INSTANCE.api_url
 
         with suppress():
             ctx = RunningContext.get()

agenta/sdk/models/shared.py

@@ -88,7 +88,7 @@ class Slug(BaseModel):
     def check_url_safety(cls, v):
         if v is not None:
             if not match(r"^[a-zA-Z0-9_-]+$", v):
-                raise ValueError("slug must be URL-safe.")
+                raise ValueError("'slug' must be URL-safe.")
         return v
 
 

agenta/sdk/tracing/exporters.py

@@ -122,6 +122,7 @@ class OTLPExporter(OTLPSpanExporter):
                     #     "[SPAN]  [EXPORT]",
                     #     trace_id=UUID(int=trace_id).hex,
                     #     span_id=UUID(int=span_id).hex[-16:],
+                    #     span_attributes=_span.attributes,
                     # )
 
                 serialized_spans.append(super().export(_spans))

agenta/sdk/tracing/processors.py

@@ -1,7 +1,5 @@
 from typing import Optional, Dict, List
 from threading import Lock
-from json import dumps
-from uuid import UUID
 
 from opentelemetry.baggage import get_all as get_baggage
 from opentelemetry.context import Context
@@ -14,8 +12,7 @@ from opentelemetry.sdk.trace.export import (
 from opentelemetry.trace import SpanContext
 
 from agenta.sdk.utils.logging import get_module_logger
-from agenta.sdk.tracing.conventions import Reference
-
+from agenta.sdk.models.tracing import BaseModel
 from agenta.sdk.contexts.tracing import TracingContext
 
 log = get_module_logger(__name__)
@@ -65,15 +62,36 @@ class TraceProcessor(SpanProcessor):
         # )
 
         for key in self.references.keys():
-            span.set_attribute(f"ag.refs.{key}", self.references[key])
+            ref = self.references[key]
+            if isinstance(ref, BaseModel):
+                try:
+                    ref = ref.model_dump(mode="json", exclude_none=True)
+                except Exception:  # pylint: disable=bare-except
+                    pass
+            if isinstance(ref, dict):
+                for field, value in ref.items():
+                    span.set_attribute(f"ag.refs.{key}.{field}", str(value))
 
         baggage = get_baggage(parent_context)
 
         for key in baggage.keys():
-            if key.startswith("ag.refs."):
-                _key = key.replace("ag.refs.", "")
-                if _key in [_.value for _ in Reference.__members__.values()]:
-                    span.set_attribute(key, baggage[key])
+            if key.startswith("ag."):
+                value = baggage[key]
+
+                if key.startswith("ag.refs."):
+                    ref = value
+                    if isinstance(value, BaseModel):
+                        try:
+                            ref = value.model_dump(mode="json", exclude_none=True)  # type: ignore
+                        except Exception:  # pylint: disable=bare-except
+                            pass
+                    if isinstance(ref, dict):
+                        for field, val in ref.items():
+                            span.set_attribute(f"{key}.{field}", str(val))
+                else:
+                    # Not a reference - only set if it's a valid attribute type
+                    if isinstance(value, (str, bool, int, float, bytes)):
+                        span.set_attribute(key, value)
 
         context = TracingContext.get()
 
@@ -105,10 +123,11 @@ class TraceProcessor(SpanProcessor):
         if not self.inline:
             if context.links:
                 for key, link in context.links.items():
-                    try:
-                        link = link.model_dump(mode="json", exclude_none=True)
-                    except:  # pylint: disable=bare-except
-                        pass
+                    if isinstance(link, BaseModel):
+                        try:
+                            link = link.model_dump(mode="json", exclude_none=True)
+                        except Exception:
+                            pass
                     if not isinstance(link, dict):
                         continue
                     if not link.get("trace_id") or not link.get("span_id"):
@@ -127,30 +146,14 @@ class TraceProcessor(SpanProcessor):
 
         if context.references:
             for key, ref in context.references.items():
-                try:
-                    ref = ref.model_dump(mode="json", exclude_none=True)
-                except:  # pylint: disable=bare-except
-                    pass
-                if not isinstance(ref, dict):
-                    continue
-                if not ref.get("id") and not ref.get("slug") and not ref.get("version"):
-                    continue
-
-                if ref.get("id"):
-                    span.set_attribute(
-                        f"ag.refs.{key}.id",
-                        str(ref.get("id")),
-                    )
-                if ref.get("slug"):
-                    span.set_attribute(
-                        f"ag.refs.{key}.slug",
-                        str(ref.get("slug")),
-                    )
-                if ref.get("version"):
-                    span.set_attribute(
-                        f"ag.refs.{key}.version",
-                        str(ref.get("version")),
-                    )
+                if isinstance(ref, BaseModel):
+                    try:
+                        ref = ref.model_dump(mode="json", exclude_none=True)
+                    except Exception:
+                        pass
+                if isinstance(ref, dict):
+                    for field, value in ref.items():
+                        span.set_attribute(f"ag.refs.{key}.{field}", str(value))
 
         trace_id = span.context.trace_id
         span_id = span.context.span_id

agenta/sdk/tracing/tracing.py

@@ -31,6 +31,7 @@ from agenta.sdk.tracing.conventions import Reference, is_valid_attribute_key
 from agenta.sdk.tracing.propagation import extract, inject
 from agenta.sdk.utils.cache import TTLLRUCache
 
+import agenta as ag
 
 log = get_module_logger(__name__)
 
@@ -215,6 +216,42 @@ class Tracing(metaclass=Singleton):
                         namespace="metrics",
                     )
 
+    def store_session(
+        self,
+        session_id: Optional[str] = None,
+        span: Optional[Span] = None,
+    ):
+        """Set session attributes on the current span.
+
+        Args:
+            session_id: Unique identifier for the session
+            span: Optional span to set attributes on (defaults to current span)
+        """
+        with suppress():
+            if span is None:
+                span = self.get_current_span()
+
+            if session_id:
+                span.set_attribute("id", session_id, namespace="session")
+
+    def store_user(
+        self,
+        user_id: Optional[str] = None,
+        span: Optional[Span] = None,
+    ):
+        """Set user attributes on the current span.
+
+        Args:
+            user_id: Unique identifier for the user
+            span: Optional span to set attributes on (defaults to current span)
+        """
+        with suppress():
+            if span is None:
+                span = self.get_current_span()
+
+            if user_id:
+                span.set_attribute("id", user_id, namespace="user")
+
     def is_inline_trace_ready(
         self,
         trace_id: Optional[int] = None,
@@ -314,6 +351,58 @@ class Tracing(metaclass=Singleton):
 
         return None
 
+    def get_trace_url(
+        self,
+        trace_id: Optional[str] = None,
+    ) -> str:
+        """
+        Build a URL to view a trace in the Agenta UI.
+
+        Automatically extracts the trace ID from the current tracing context
+        if not explicitly provided.
+
+        Args:
+            trace_id: Optional trace ID (hex string format). If not provided,
+                      it will be automatically extracted from the current trace context.
+
+        Returns:
+            The full URL to view the trace in the observability dashboard
+
+        Raises:
+            RuntimeError: If the SDK is not initialized, no active trace context exists,
+                          or scope info cannot be fetched
+        """
+        if trace_id is None:
+            span_ctx = self.get_span_context()
+            if span_ctx is None or not span_ctx.is_valid:
+                raise RuntimeError(
+                    "No active trace context found. "
+                    "Make sure you call this within an instrumented function or span."
+                )
+
+            trace_id = f"{span_ctx.trace_id:032x}"
+
+        if not ag or not ag.DEFAULT_AGENTA_SINGLETON_INSTANCE:
+            raise RuntimeError(
+                "Agenta SDK is not initialized. Please call ag.init() first."
+            )
+
+        api_url = ag.DEFAULT_AGENTA_SINGLETON_INSTANCE.api_url
+        web_url = api_url.replace("/api", "") if api_url else None
+
+        (organization_id, workspace_id, project_id) = (
+            ag.DEFAULT_AGENTA_SINGLETON_INSTANCE.resolve_scopes()
+        )
+
+        if not web_url or not workspace_id or not project_id:
+            raise RuntimeError(
+                "Could not determine workspace/project context. Please call ag.init() first."
+            )
+
+        return (
+            f"{web_url}/w/{workspace_id}/p/{project_id}/observability?trace={trace_id}"
+        )
+
 
 def get_tracer(
     tracing: Tracing,

agenta/sdk/workflows/runners/daytona.py

@@ -88,7 +88,6 @@ class DaytonaRunner(CodeRunner):
                 target=target,
             )
             self.daytona = Daytona(config)
-            # log.debug("Daytona client initialized")
 
         except Exception as e:
             raise RuntimeError(f"Failed to initialize Daytona client: {e}")
@@ -107,8 +106,6 @@ class DaytonaRunner(CodeRunner):
                     "Set it to the Daytona sandbox ID or snapshot name you want to use."
                 )
 
-            # log.debug(f"Creating sandbox from snapshot: {snapshot_id}")
-
             from daytona import CreateSandboxFromSnapshotParams
 
             sandbox = self.daytona.create(
@@ -118,9 +115,6 @@ class DaytonaRunner(CodeRunner):
                 )
             )
 
-            # log.debug(
-            #     f"Sandbox created: {sandbox.id if hasattr(sandbox, 'id') else sandbox}"
-            # )
             return sandbox
 
         except Exception as e:

agenta/sdk/workflows/runners/registry.py

@@ -1,7 +1,17 @@
 import os
+from typing import TYPE_CHECKING
+
 from agenta.sdk.workflows.runners.base import CodeRunner
 from agenta.sdk.workflows.runners.local import LocalRunner
-from agenta.sdk.workflows.runners.daytona import DaytonaRunner
+
+if TYPE_CHECKING:
+    from agenta.sdk.workflows.runners.daytona import DaytonaRunner
+
+
+def _get_daytona_runner() -> "DaytonaRunner":
+    from agenta.sdk.workflows.runners.daytona import DaytonaRunner
+
+    return DaytonaRunner()
 
 
 def get_runner() -> CodeRunner:
@@ -21,7 +31,14 @@ def get_runner() -> CodeRunner:
     runner_type = os.getenv("AGENTA_SERVICES_SANDBOX_RUNNER", "local").lower()
 
     if runner_type == "daytona":
-        return DaytonaRunner()
+        try:
+            return _get_daytona_runner()
+        except ImportError as exc:
+            raise ValueError(
+                "Daytona runner requires the 'daytona' package. "
+                "Install optional dependencies or set "
+                "AGENTA_SERVICES_SANDBOX_RUNNER=local."
+            ) from exc
     elif runner_type == "local":
         return LocalRunner()
     else:

{agenta-0.68.0.dist-info → agenta-0.70.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agenta
-Version: 0.68.0
+Version: 0.70.1
 Summary: The SDK for agenta is an open-source LLMOps platform.
 Keywords: LLMOps,LLM,evaluation,prompt engineering
 Author: Mahmoud Mabrouk
@@ -16,7 +16,7 @@ Classifier: Programming Language :: Python :: 3.14
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Topic :: Software Development :: Libraries
 Requires-Dist: daytona (>=0.121.0,<0.122.0)
-Requires-Dist: fastapi (>=0.122,<0.123)
+Requires-Dist: fastapi (>=0.125)
 Requires-Dist: httpx (>=0.28,<0.29)
 Requires-Dist: importlib-metadata (>=8,<9)
 Requires-Dist: jinja2 (>=3,<4)
@@ -26,6 +26,7 @@ Requires-Dist: opentelemetry-api (>=1,<2)
 Requires-Dist: opentelemetry-exporter-otlp-proto-http (>=1,<2)
 Requires-Dist: opentelemetry-instrumentation (>=0.59b0,<0.60)
 Requires-Dist: opentelemetry-sdk (>=1,<2)
+Requires-Dist: orjson (>=3,<4)
 Requires-Dist: pydantic (>=2,<3)
 Requires-Dist: python-dotenv (>=1,<2)
 Requires-Dist: python-jsonpath (>=2,<3)

{agenta-0.68.0.dist-info → agenta-0.70.1.dist-info}/RECORD

@@ -1,4 +1,4 @@
-agenta/__init__.py,sha256=ltMd-tuCDxglMfUT51-XlvSqwoE9QsoYi5JSQ4WKO9Q,2597
+agenta/__init__.py,sha256=OtejSy4-gUHKpLOLgQ_yHYROKaJ57u7DHuBMDTDrmJM,3677
 agenta/client/Readme.md,sha256=ZQJ_nBVYfVJizfMaD_3WPdXPBfi8okrV7i8LAUAfdu0,7604
 agenta/client/__init__.py,sha256=dXPiqGSFUGv0XIZtKXcmoMea1oRFOiTi5247VVk56Ig,10832
 agenta/client/backend/__init__.py,sha256=KmQImSlBeXCum1aWUOKN0415fAs0-CGoWgWcKhr0wmI,10559
@@ -306,8 +306,8 @@ agenta/client/types.py,sha256=wBGDVktTL2EblEKW23Y-VrFp7V_JHLPMHltt2jEkF0Q,129
 agenta/config.py,sha256=0VrTqduB4g8Mt_Ll7ffFcEjKF5qjTUIxmUtTPW2ygWw,653
 agenta/config.toml,sha256=sIORbhnyct2R9lJrquxhNL4pHul3O0R7iaipCoja5MY,193
 agenta/sdk/__init__.py,sha256=7QUpZ409HcLB22A80qaZydzhs6afPnCvG0Tfq6PE4fk,3011
-agenta/sdk/agenta_init.py,sha256=Vm14_nzyObAmwQFQvAANLsOjqy5AwYIFa6MrdnJ0aqY,7286
-agenta/sdk/assets.py,sha256=51uSUp-qlFLB-nLSrDDTDXOQhM-2yGIuODgALYt1i9Y,8699
+agenta/sdk/agenta_init.py,sha256=KiZpk0cfzvKNUtO5S0I_FQghcy8AKRkSxDtVjlZaNkw,9950
+agenta/sdk/assets.py,sha256=c7fbv-ZJdoY8xcHwTJfgT9UeSSamfZuhimfPNYyQZ_k,8865
 agenta/sdk/context/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agenta/sdk/context/running.py,sha256=3gEuUdQrJwcuN93MlXFZ6aHXNxUW6dUk_EudgaxOkCU,907
 agenta/sdk/context/serving.py,sha256=jGRd6v4wWNNoDFM7PJ4ct0MXcMDAVY6puPuseSeIL60,960
@@ -343,7 +343,7 @@ agenta/sdk/evaluations/results.py,sha256=3pe2c0oI1Wc1wFCKFeQnbe_iwtNn9U0MO_c4U3H
 agenta/sdk/evaluations/runs.py,sha256=8euS2Zfzcw9a7SDOnhK4DymuYNIjIXQ1iWlRuuf0q78,3398
 agenta/sdk/evaluations/scenarios.py,sha256=XlsVa_M8FmnSvVniF_FEZUhZDLYIotx4V0SRmPEzWS8,1024
 agenta/sdk/litellm/__init__.py,sha256=Bpz1gfHQc0MN1yolWcjifLWznv6GjHggvRGQSpxpihM,37
-agenta/sdk/litellm/litellm.py,sha256=E7omr9kz0yn8CUK5O0g0QUlDA4bD5fllYtHK9RL2bXE,10646
+agenta/sdk/litellm/litellm.py,sha256=5UJ9PcxwocgskHjrwKDW1XCz5y4OxUHq2dL2EwTXkq0,11288
 agenta/sdk/litellm/mockllm.py,sha256=R32eoGvXokxNk0WihgP8rIdd5mTwCPXF--U5NDisnJE,2255
 agenta/sdk/litellm/mocks/__init__.py,sha256=dqTLo5sFH6IkjyunWPuBPpahfZ-cbUHOkRloTWTq7BY,5262
 agenta/sdk/managers/__init__.py,sha256=SN-LRwG0pRRDV3u2Q4JiiSTigN3-mYpzGNM35RzT4mc,238
@@ -359,13 +359,13 @@ agenta/sdk/managers/testsets.py,sha256=m8U6LA-ZuS_8sm_sIFR2rUXFScNncMZ52Ps1hSPt-
 agenta/sdk/managers/variant.py,sha256=A5ga3mq3b0weUTXa9HO72MGaspthGcu1uK9K5OnP738,4172
 agenta/sdk/managers/vault.py,sha256=wqDVFPuUi-Zida9zVhuHeW6y63K2kez1-4Kk5U0dY48,334
 agenta/sdk/middleware/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-agenta/sdk/middleware/auth.py,sha256=z-FEr9UbcxM0snf_2KlW1O7GvQFMf2ZET-wxSs4CDkI,10233
+agenta/sdk/middleware/auth.py,sha256=UUwqEfsrNt49U1hs-Qfjsjkl5t7-I25VWhS7dsrHqJs,10528
 agenta/sdk/middleware/config.py,sha256=0eKRkARI0fkXwmq3NNL4x7iOue0_xN3RQCgoHk8guTc,8723
 agenta/sdk/middleware/cors.py,sha256=q3r7lGkrIdMcT_vuhsburMcjG7pyl7w0ycxrIrGJ2e8,921
 agenta/sdk/middleware/inline.py,sha256=ee8E4XBGcRSrHTvblqX1yRXuTN_sxLm7lY1jnywrBG8,901
 agenta/sdk/middleware/mock.py,sha256=bCUN9iJBxePyN9MBwBpJs-_iCNkUQeUjIIu3WElS1oQ,759
-agenta/sdk/middleware/otel.py,sha256=lHzhGUv4fq2RPuTPH2keJ16v-_cBUrLrTqWzHUmEVdI,1041
-agenta/sdk/middleware/vault.py,sha256=0fMsNY0HOxm4tmkgchxGe6EfaOMdvS-gzs3E7zoAhQI,12200
+agenta/sdk/middleware/otel.py,sha256=M0cIFdRXVx17g3Nh2bvPjjVBBjoKwyI0MrGtJ4aGjjA,886
+agenta/sdk/middleware/vault.py,sha256=xPrR8XUAMG7pgJ4SKMJxAlrfQGgeeIHFoJ-gYfqVWj0,12471
 agenta/sdk/middlewares/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agenta/sdk/middlewares/routing/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agenta/sdk/middlewares/routing/auth.py,sha256=WmZLxSu4lj9ltI-SpF31MAfAp7uJh3-E9T-U2L4Y58g,10476
@@ -374,12 +374,12 @@ agenta/sdk/middlewares/routing/otel.py,sha256=w-J7PpoBSyWYAFeoqAHxywjMuYMaW_eJ-t
 agenta/sdk/middlewares/running/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agenta/sdk/middlewares/running/normalizer.py,sha256=8D7iJkYDbqWiQEGISZOInZCyEjqyOlXETTq3dvDWFac,11384
 agenta/sdk/middlewares/running/resolver.py,sha256=I4nX7jpsq7oKSwQnlsoLm9kh94G2qxBdablJob1l_fs,5013
-agenta/sdk/middlewares/running/vault.py,sha256=DqeWyViDDRJycXtRKEO1S7ihlBfCnPFtXSfj_6trW98,3845
+agenta/sdk/middlewares/running/vault.py,sha256=C4bKrQOdSc1L16peAm9adOw-hOlEnbN4CQSLtp_C4L4,3839
 agenta/sdk/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agenta/sdk/models/blobs.py,sha256=g8zV6V3UcuskEV6tK_tvt8GDWqx1okfE3Oy4n59mgU0,576
 agenta/sdk/models/evaluations.py,sha256=twHFZQlAT2xYSqefJdpwtDLSu5GpsDmswV_o04ACyFs,3003
 agenta/sdk/models/git.py,sha256=ol5H3lu6s3Lx2l7K3glbCqcvAnPXsDsYa-OsSR0CupM,2415
-agenta/sdk/models/shared.py,sha256=ynHFXsOgkpvzHJuBcmPp2lMj7ia6x0e9LvNAiKVpPSo,4020
+agenta/sdk/models/shared.py,sha256=zZz0dqSjOhtfGdXq7_zMe35AuUdb2moIoyZafQxUhyM,4022
 agenta/sdk/models/testsets.py,sha256=cHCZEWqnDMGqKKH9yVuM5FroG10YFKJ4nF5xZniE-Ds,3415
 agenta/sdk/models/tracing.py,sha256=adxOguxuU_uFRL1sGWPr31xCHs9qnfkD92xYcgHEDdA,5079
 agenta/sdk/models/workflows.py,sha256=VhPPK4V15pkrbI3otA9RxmoXvTeVd_KHVmYK86iw_MU,18997
@@ -387,12 +387,12 @@ agenta/sdk/router.py,sha256=mOguvtOwl2wmyAgOuWTsf98pQwpNiUILKIo67W_hR3A,119
 agenta/sdk/tracing/__init__.py,sha256=rQNe5-zT5Kt7_CDhq-lnUIi1EYTBVzVf_MbfcIxVD98,41
 agenta/sdk/tracing/attributes.py,sha256=Brqle9hGV3DaEJjeYCBq7MDlbvHMAIwmkUj2Lni8dF4,5563
 agenta/sdk/tracing/conventions.py,sha256=JBtznBXZ3aRkGKkLl7cPwdMNh3w1G-H2Ta2YrAxbr38,950
-agenta/sdk/tracing/exporters.py,sha256=NYqw80liPrs91ywkkQC13530pPmpUZguPgTEDO0JKFs,5118
+agenta/sdk/tracing/exporters.py,sha256=6KEI3ESEJaMqktnq0DyTljWRXSVNBv6qcHezfVI9apA,5178
 agenta/sdk/tracing/inline.py,sha256=UKt10JGKdS6gVDIpExng3UC8vegAcuA2KxlzyvSdUZ0,31886
-agenta/sdk/tracing/processors.py,sha256=A7rsaicpFq9xZgyhU3hV5ZQoz6X33gB81G9IhB-x3Xg,8597
+agenta/sdk/tracing/processors.py,sha256=AXCVPLgvUXRHuJwaHvG6K0TkP3pUnSeVk3en6wu3n6s,9019
 agenta/sdk/tracing/propagation.py,sha256=Zu_z5In8eOhy0tkYzQOI09T4OwdjGMP74nhzvElvyFE,2593
 agenta/sdk/tracing/spans.py,sha256=r-R68d12BjvilHgbqN-1xp26qxdVRzxRcFUO-IB_u94,3780
-agenta/sdk/tracing/tracing.py,sha256=mogsWlTwz-pYvzpst4xb4kjuGRtywzrU9GO9T9stvZw,9236
+agenta/sdk/tracing/tracing.py,sha256=dP2LShcNBtWqGQKQBAGA52dDqNN15IcKcvXz_EwHigI,12200
 agenta/sdk/types.py,sha256=41yIQagl5L_7WFInjiACHwuNfCQqDrrDOusD17kJGWs,28469
 agenta/sdk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agenta/sdk/utils/cache.py,sha256=Er1Hvu1QVLGl99HkUHZ2lKBg3f6PnpkD1uZRvK9r3u4,1429
@@ -416,11 +416,11 @@ agenta/sdk/workflows/handlers.py,sha256=z_DtfgiejsToO4kqXvXBnO36hd0rk97Y2J2hYlIf
 agenta/sdk/workflows/interfaces.py,sha256=I5Bfil0awdL1TAb_vHqW5n5BHxSBOTuDMhOi4RnUt8A,36315
 agenta/sdk/workflows/runners/__init__.py,sha256=HoYaKf9G03WEUbY7B1uX4O_6xE5dfliNCG1nEuWp1ks,87
 agenta/sdk/workflows/runners/base.py,sha256=WgX0OgbLL5PHeGqLNAvrV7NC3FHDWVfU7v9EBj8MIW0,857
-agenta/sdk/workflows/runners/daytona.py,sha256=g09014ocerh7X-sLOJ01i5ko6YysdXCPfnrOfn9MWQQ,9791
+agenta/sdk/workflows/runners/daytona.py,sha256=PqpbXh3HD_SZoS5lMYBW2lE7i0pepTzTDfHir5dNhv0,9531
 agenta/sdk/workflows/runners/local.py,sha256=SJ1msO35mQ4XzlqZi9fE25QJu-PDnYru8a66pMo-5vs,3636
-agenta/sdk/workflows/runners/registry.py,sha256=bHM7hTiFawdOM30RwCHeAwFN8C0zGF0at1zsI-57tlw,1067
+agenta/sdk/workflows/runners/registry.py,sha256=QKU_6IXMcbdq_kxoQlOhjVBJ9nBq07QqHIBxFqSI7Uk,1556
 agenta/sdk/workflows/sandbox.py,sha256=O1Opeg4hc9jygAzyF5cCsStmMjYgrahA_aF0JdGbBO0,1734
 agenta/sdk/workflows/utils.py,sha256=UDG5or8qqiSCpqi0Fphjxkkhu4MdbiCkHn_yIQcTd0c,11664
-agenta-0.68.0.dist-info/METADATA,sha256=B8evMuWFY51eFJX03VPfmeyJ-ns-4wFnBSxVtcmWj1A,31572
-agenta-0.68.0.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-agenta-0.68.0.dist-info/RECORD,,
+agenta-0.70.1.dist-info/METADATA,sha256=-w_Gfj3Sz_2mDtULdrTUoviZQoL2botY1pg-pYjrk9c,31596
+agenta-0.70.1.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+agenta-0.70.1.dist-info/RECORD,,