agenta 0.52.6__py3-none-any.whl → 0.63.2__py3-none-any.whl

agenta/sdk/__init__.py

@@ -2,6 +2,50 @@ from typing import Optional, Callable, Any
 
 from .utils.preinit import PreInitObject  # always the first import!
 
+__all__ = [
+    # Decorators
+    "workflow",
+    "application",
+    "evaluator",
+    "instrument",
+    "route",
+    "app",
+    # Initialization
+    "init",
+    "config",
+    # Types
+    "DictInput",
+    "MultipleChoice",
+    "FloatParam",
+    "IntParam",
+    "MultipleChoiceParam",
+    "GroupedMultipleChoiceParam",
+    "TextParam",
+    "MessagesInput",
+    "FileInputURL",
+    "BinaryParam",
+    "Prompt",
+    # Tracing
+    "Tracing",
+    "tracing",
+    "tracer",
+    "get_tracer",
+    "Reference",
+    # Managers
+    "AppManager",
+    "VaultManager",
+    "SecretsManager",
+    "ConfigManager",
+    "VariantManager",
+    "DeploymentManager",
+    # Utilities
+    "calculate_token_usage",
+    # API clients
+    "api",
+    "async_api",
+    "types",
+]
+
 import agenta.client.backend.types as client_types  # pylint: disable=wrong-import-order
 
 from .types import (
@@ -21,9 +65,13 @@ from .types import (
 )
 
 from .tracing import Tracing, get_tracer
-from .decorators.tracing import instrument
-from .decorators.running import workflow, workflows
-from .decorators.serving import entrypoint, app, route
+from agenta.sdk.decorators.tracing import instrument
+from agenta.sdk.decorators.running import (
+    workflow,
+    application,
+    evaluator,
+)
+from agenta.sdk.decorators.serving import route, app
 from .tracing.conventions import Reference
 from .agenta_init import Config, AgentaSingleton, init as _init
 from .utils.costs import calculate_token_usage
@@ -33,6 +81,7 @@ from .managers.secrets import SecretsManager
 from .managers.config import ConfigManager
 from .managers.variant import VariantManager
 from .managers.deployment import DeploymentManager
+from .managers import testsets as testsets
 
 
 config = PreInitObject("agenta.config", Config)

agenta/sdk/agenta_init.py

@@ -9,7 +9,7 @@ from agenta.sdk.utils.logging import get_module_logger
 from agenta.client.client import AgentaApi, AsyncAgentaApi
 
 from agenta.sdk.tracing import Tracing
-from agenta.sdk.context.serving import serving_context
+from agenta.sdk.contexts.routing import RoutingContext
 
 
 log = get_module_logger(__name__)
@@ -27,6 +27,7 @@ class AgentaSingleton:
 
     def __init__(self):
         self.host = None
+        self.api_url = None
         self.api_key = None
 
         self.scope_type = None
@@ -41,6 +42,7 @@ class AgentaSingleton:
         self,
         *,
         host: Optional[str] = None,
+        api_url: Optional[str] = None,
         api_key: Optional[str] = None,
         config_fname: Optional[str] = None,
         redact: Optional[Callable[..., Any]] = None,
@@ -68,7 +70,7 @@ class AgentaSingleton:
 
         """
 
-        log.info("Agenta - SDK version: %s", version("agenta"))
+        log.info("Agenta -  SDK ver: %s", version("agenta"))
 
         config = {}
         if config_fname:
@@ -77,36 +79,59 @@ class AgentaSingleton:
         _host = (
             host
             or getenv("AGENTA_HOST")
-            or config.get("backend_host")
             or config.get("host")
-            or getenv("AGENTA_API_URL", "https://cloud.agenta.ai")
+            or "https://cloud.agenta.ai"
         )
 
+        _api_url = (
+            api_url
+            or getenv("AGENTA_API_INTERNAL_URL")
+            or getenv("AGENTA_API_URL")
+            or config.get("api_url")
+            or None  # NO FALLBACK
+        )
+
+        if _api_url:
+            _api_url = parse_url(url=_api_url)
+            _host = _api_url.rsplit("/api", 1)[0]
+        elif _host:
+            _host = parse_url(url=_host)
+            _api_url = _host + "/api"
+
         try:
-            assert _host and isinstance(
-                _host, str
-            ), "Host is required. Please provide a valid host or set AGENTA_HOST environment variable."
-            self.host = parse_url(url=_host)
+            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."
+            )
+            self.host = _host
+            self.api_url = _api_url
         except AssertionError as e:
             log.error(str(e))
             raise
         except Exception as e:
-            log.error(f"Failed to parse host URL '{_host}': {e}")
+            log.error(f"Failed to parse API URL '{_api_url}': {e}")
             raise
 
-        log.info("Agenta - Host: %s", self.host)
+        self.api_key = (
+            api_key
+            or getenv("AGENTA_API_KEY")
+            or config.get("api_key")
+            or None  # NO FALLBACK
+        )
 
-        self.api_key = api_key or getenv("AGENTA_API_KEY") or config.get("api_key")
+        log.info("Agenta -  API URL: %s", self.api_url)
 
         self.scope_type = (
             scope_type
             or getenv("AGENTA_SCOPE_TYPE")
             or config.get("scope_type")
-            or None
+            or None  # NO FALLBACK
         )
 
         self.scope_id = (
-            scope_id or getenv("AGENTA_SCOPE_ID") or config.get("scope_id") or None
+            scope_id
+            or getenv("AGENTA_SCOPE_ID")
+            or config.get("scope_id")
+            or None  # NO FALLBACK
         )
 
         self.tracing = Tracing(
@@ -120,12 +145,12 @@ class AgentaSingleton:
         )
 
         self.api = AgentaApi(
-            base_url=self.host + "/api",
+            base_url=self.api_url,
             api_key=self.api_key if self.api_key else "",
         )
 
         self.async_api = AsyncAgentaApi(
-            base_url=self.host + "/api",
+            base_url=self.api_url,
             api_key=self.api_key if self.api_key else "",
         )
 
@@ -149,7 +174,7 @@ class Config:
         return self.default_parameters
 
     def __getattr__(self, key):
-        context = serving_context.get()
+        context = RoutingContext.get()
 
         parameters = context.parameters
 
@@ -172,6 +197,7 @@ class Config:
 
 def init(
     host: Optional[str] = None,
+    api_url: Optional[str] = None,
     api_key: Optional[str] = None,
     config_fname: Optional[str] = None,
     redact: Optional[Callable[..., Any]] = None,
@@ -200,6 +226,7 @@ def init(
 
     singleton.init(
         host=host,
+        api_url=api_url,
         api_key=api_key,
         config_fname=config_fname,
         redact=redact,

agenta/sdk/assets.py

@@ -1,5 +1,8 @@
 supported_llm_models = {
-    "Anthropic": [
+    "anthropic": [
+        "anthropic/claude-sonnet-4-5",
+        "anthropic/claude-haiku-4-5",
+        "anthropic/claude-opus-4-1",
         "anthropic/claude-sonnet-4-20250514",
         "anthropic/claude-opus-4-20250514",
         "anthropic/claude-3-7-sonnet-20250219",
@@ -12,29 +15,33 @@ supported_llm_models = {
         "anthropic/claude-2.1",
         "anthropic/claude-2",
     ],
-    "Cohere": [
+    "cohere": [
         "cohere/command-light",
         "cohere/command-r-plus",
         "cohere/command-nightly",
     ],
-    "DeepInfra": [
+    "deepinfra": [
         "deepinfra/meta-llama/Llama-2-70b-chat-hf",
         "deepinfra/meta-llama/Llama-2-13b-chat-hf",
         "deepinfra/codellama/CodeLlama-34b-Instruct-hf",
         "deepinfra/mistralai/Mistral-7B-Instruct-v0.1",
     ],
-    "Gemini": [
+    "gemini": [
+        "gemini/gemini-3-pro-preview",
+        "gemini/gemini-2.5-pro",
+        "gemini/gemini-2.5-pro-preview-05-06",
+        "gemini/gemini-2.5-flash",
+        "gemini/gemini-2.5-flash-preview-09-2025",
         "gemini/gemini-2.5-flash-preview-05-20",
         "gemini/gemini-2.5-flash-preview-04-17",
+        "gemini/gemini-2.5-flash-lite",
+        "gemini/gemini-2.5-flash-lite-preview-09-2025",
+        "gemini/gemini-2.0-flash",
         "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/gemini-2.0-flash-lite-preview-02-05",
     ],
-    "Groq": [
+    "groq": [
         "groq/deepseek-r1-distill-llama-70b",
         "groq/deepseek-r1-distill-llama-70b-specdec",
         "groq/gemma2-9b-it",
@@ -50,14 +57,15 @@ supported_llm_models = {
         "groq/llama3-8b-8192",
         "groq/mixtral-8x7b-32768",
     ],
-    "Mistral": [
+    "mistral": [
         "mistral/mistral-tiny",
         "mistral/mistral-small",
         "mistral/mistral-medium",
         "mistral/mistral-large-latest",
     ],
-    "OpenAI": [
+    "openai": [
         "gpt-5",
+        "gpt-5.1",
         "gpt-5-mini",
         "gpt-5-nano",
         "gpt-4.5-preview",
@@ -72,7 +80,7 @@ supported_llm_models = {
         "gpt-4.1-nano",
         "o4-mini",
     ],
-    "OpenRouter": [
+    "openrouter": [
         "openrouter/qwen/qwen3-235b-a22b",
         "openrouter/qwen/qwen3-32b",
         "openrouter/qwen/qwen3-30b-a3b",
@@ -142,13 +150,13 @@ supported_llm_models = {
         "openrouter/google/gemini-2.0-flash-001",
         "openrouter/perplexity/sonar-reasoning",
     ],
-    "Perplexity AI": [
+    "perplexity": [
         "perplexity/sonar",
         "perplexity/sonar-pro",
         "perplexity/sonar-reasoning",
         "perplexity/sonar-reasoning-pro",
     ],
-    "togetherai": [
+    "together_ai": [
         "together_ai/deepseek-ai/DeepSeek-R1",
         "together_ai/deepseek-ai/DeepSeek-R1-Distill-Llama-70B",
         "together_ai/deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B",

agenta/sdk/context/serving.py

@@ -1,7 +1,7 @@
 from typing import Any, Dict, List, Optional
 
 from contextlib import contextmanager
-from contextvars import ContextVar
+from contextvars import ContextVar, Token
 
 from pydantic import BaseModel
 
@@ -11,17 +11,29 @@ class RoutingContext(BaseModel):
     secrets: Optional[List[Any]] = None
     mock: Optional[str] = None
 
+    @classmethod
+    def get(cls) -> "RoutingContext":
+        try:
+            return routing_context.get()
+        except LookupError:
+            return RoutingContext()
 
-serving_context = ContextVar("serving_context", default=RoutingContext())
+    @classmethod
+    def set(cls, ctx: "RoutingContext") -> Token:
+        return routing_context.set(ctx)
+
+    @classmethod
+    def reset(cls, token: Token) -> None:
+        return routing_context.reset(token)
+
+
+routing_context: ContextVar[RoutingContext] = ContextVar("routing_context")
 
 
 @contextmanager
-def serving_context_manager(
-    *,
-    context: Optional[RoutingContext] = None,
-):
-    token = serving_context.set(context)
+def routing_context_manager(context: RoutingContext):
+    token = RoutingContext.set(context)
     try:
         yield
     finally:
-        serving_context.reset(token)
+        RoutingContext.reset(token)

agenta/sdk/context/tracing.py

@@ -1,7 +1,7 @@
 from typing import Any, Dict, Optional
 
 from contextlib import contextmanager
-from contextvars import ContextVar
+from contextvars import ContextVar, Token
 
 from pydantic import BaseModel
 
@@ -15,42 +15,60 @@ class TracingContext(BaseModel):
     link: Optional[Dict[str, Any]] = None
     type: Optional[str] = None
 
+    @classmethod
+    def get(cls) -> "TracingContext":
+        try:
+            return tracing_context.get()
+        except LookupError:
+            return TracingContext()
 
-tracing_context = ContextVar(
-    "ag.tracing_context",
-    default=TracingContext(),
-)
+    @classmethod
+    def set(cls, ctx: "TracingContext") -> Token:
+        return tracing_context.set(ctx)
+
+    @classmethod
+    def reset(cls, token: Token) -> None:
+        return tracing_context.reset(token)
+
+
+tracing_context: ContextVar[TracingContext] = ContextVar("ag.tracing_context")
 
 
 @contextmanager
-def tracing_context_manager(
-    *,
-    context: Optional[TracingContext] = None,
-):
-    token = tracing_context.set(context)
+def tracing_context_manager(context: TracingContext):
+    token = TracingContext.set(context)
     try:
         yield
     finally:
-        tracing_context.reset(token)
+        TracingContext.reset(token)
 
 
-class TracingExporterContext(BaseModel):
+class OTLPContext(BaseModel):
     credentials: Optional[str] = None
 
+    @classmethod
+    def get(cls) -> "OTLPContext":
+        try:
+            return otlp_context.get()
+        except LookupError:
+            return OTLPContext()
+
+    @classmethod
+    def set(cls, ctx: "OTLPContext") -> Token:
+        return otlp_context.set(ctx)
+
+    @classmethod
+    def reset(cls, token: Token) -> None:
+        return otlp_context.reset(token)
+
 
-tracing_exporter_context = ContextVar(
-    "ag.tracing_exporter_context",
-    default=TracingExporterContext(),
-)
+otlp_context: ContextVar[OTLPContext] = ContextVar("ag.otlp_context")
 
 
 @contextmanager
-def tracing_exporter_context_manager(
-    *,
-    context: Optional[TracingExporterContext] = None,
-):
-    token = tracing_exporter_context.set(context)
+def otlp_context_manager(context: OTLPContext):
+    token = otlp_context.set(context)
     try:
         yield
     finally:
-        tracing_exporter_context.reset(token)
+        otlp_context.reset(token)

agenta/sdk/contexts/routing.py

@@ -0,0 +1,38 @@
+from typing import Optional
+from contextvars import ContextVar, Token
+from contextlib import contextmanager
+
+from pydantic import BaseModel
+
+
+class RoutingContext(BaseModel):
+    parameters: Optional[dict] = None
+    secrets: Optional[list] = None
+    mock: Optional[str] = None
+
+    @classmethod
+    def get(cls) -> "RoutingContext":
+        try:
+            return routing_context.get()
+        except LookupError:
+            return RoutingContext()
+
+    @classmethod
+    def set(cls, ctx: "RoutingContext") -> Token:
+        return routing_context.set(ctx)
+
+    @classmethod
+    def reset(cls, token: Token) -> None:
+        return routing_context.reset(token)
+
+
+routing_context: ContextVar[RoutingContext] = ContextVar("routing_context")
+
+
+@contextmanager
+def routing_context_manager(context: RoutingContext):
+    token = RoutingContext.set(context)
+    try:
+        yield
+    finally:
+        RoutingContext.reset(token)

agenta/sdk/contexts/running.py

@@ -0,0 +1,57 @@
+from typing import Optional, Union, Callable
+from contextvars import Token, ContextVar
+from contextlib import contextmanager
+
+
+from pydantic import BaseModel
+
+from agenta.sdk.models.workflows import (
+    WorkflowServiceInterface,
+    WorkflowServiceConfiguration,
+)
+
+
+class RunningContext(BaseModel):
+    flags: Optional[dict] = None
+    tags: Optional[dict] = None
+    meta: Optional[dict] = None
+
+    aggregate: Optional[Union[bool, Callable]] = None  # stream to batch
+    annotate: Optional[bool] = None  # annotation vs invocation
+
+    interface: Optional[WorkflowServiceInterface] = None
+    configuration: Optional[WorkflowServiceConfiguration] = None
+    parameters: Optional[dict] = None
+    schemas: Optional[dict] = None
+
+    secrets: Optional[list] = None
+    credentials: Optional[str] = None
+
+    handler: Optional[Callable] = None
+
+    @classmethod
+    def get(cls) -> "RunningContext":
+        try:
+            return running_context.get()
+        except LookupError:
+            return RunningContext()
+
+    @classmethod
+    def set(cls, ctx: "RunningContext") -> Token:
+        return running_context.set(ctx)
+
+    @classmethod
+    def reset(cls, token: Token) -> None:
+        return running_context.reset(token)
+
+
+running_context: ContextVar[RunningContext] = ContextVar("running_context")
+
+
+@contextmanager
+def running_context_manager(context: RunningContext):
+    token = RunningContext.set(context)
+    try:
+        yield
+    finally:
+        RunningContext.reset(token)

agenta/sdk/contexts/tracing.py

@@ -0,0 +1,86 @@
+from typing import Optional, Union, Callable
+from contextvars import ContextVar, Token
+from contextlib import contextmanager
+
+from pydantic import BaseModel
+
+
+class TracingContext(BaseModel):
+    traceparent: Optional[dict] = None
+    baggage: Optional[dict] = None
+    #
+    credentials: Optional[str] = None
+    #
+    script: Optional[dict] = None
+    parameters: Optional[dict] = None
+    #
+    flags: Optional[dict] = None
+    tags: Optional[dict] = None
+    meta: Optional[dict] = None
+    #
+    references: Optional[dict] = None
+    links: Optional[dict] = None
+    #
+    type: Optional[str] = None
+    aggregate: Optional[Union[bool, Callable]] = None  # stream to batch
+    annotate: Optional[bool] = None  # annotation vs invocation
+    #
+    link: Optional[dict] = None
+
+    @classmethod
+    def get(cls) -> "TracingContext":
+        try:
+            return tracing_context.get()
+        except LookupError:
+            return TracingContext()
+
+    @classmethod
+    def set(cls, ctx: "TracingContext") -> Token:
+        return tracing_context.set(ctx)
+
+    @classmethod
+    def reset(cls, token: Token) -> None:
+        return tracing_context.reset(token)
+
+
+tracing_context: ContextVar[TracingContext] = ContextVar("ag.tracing_context")
+
+
+@contextmanager
+def tracing_context_manager(context: TracingContext):
+    token = TracingContext.set(context)
+    try:
+        yield
+    finally:
+        TracingContext.reset(token)
+
+
+class OTLPContext(BaseModel):
+    credentials: Optional[str] = None
+
+    @classmethod
+    def get(cls) -> "OTLPContext":
+        try:
+            return otlp_context.get()
+        except LookupError:
+            return OTLPContext()
+
+    @classmethod
+    def set(cls, ctx: "OTLPContext") -> Token:
+        return otlp_context.set(ctx)
+
+    @classmethod
+    def reset(cls, token: Token) -> None:
+        return otlp_context.reset(token)
+
+
+otlp_context: ContextVar[OTLPContext] = ContextVar("ag.otlp_context")
+
+
+@contextmanager
+def otlp_context_manager(context: OTLPContext):
+    token = OTLPContext.set(context)
+    try:
+        yield
+    finally:
+        OTLPContext.reset(token)

agenta/sdk/decorators/__init__.py

@@ -0,0 +1 @@
+from .running import application, evaluator