agenta 0.26.0a0__py3-none-any.whl → 0.27.0__py3-none-any.whl

agenta/docker/docker-assets/Dockerfile.cloud.template

@@ -1,4 +1,4 @@
-FROM public.ecr.aws/s2t9a1r1/agentaai/lambda_templates_public:main
+FROM public.ecr.aws/h3w6n5z0/agentaai/lambda_templates_public:main
 
 COPY requirements.txt ${LAMBDA_TASK_ROOT}
 RUN pip install --no-cache-dir --disable-pip-version-check -U agenta

agenta/sdk/__init__.py

@@ -1,7 +1,10 @@
+from typing import Optional
+
 from .utils.preinit import PreInitObject  # always the first import!
-from .context import get_contexts, save_context
+
+import agenta.client.backend.types as client_types  # pylint: disable=wrong-import-order
+
 from .types import (
-    Context,
     DictInput,
     MultipleChoice,
     FloatParam,
@@ -13,15 +16,49 @@ from .types import (
     MessagesInput,
     FileInputURL,
     BinaryParam,
+    Prompt,
 )
 
-from .tracing.llm_tracing import Tracing
+from .tracing import Tracing, get_tracer
 from .decorators.tracing import instrument
-from .decorators.llm_entrypoint import entrypoint, app, route
-from .agenta_init import Config, AgentaSingleton, init
-from .utils.helper.openai_cost import calculate_token_usage
-from .config_manager import ConfigManager
+from .tracing.conventions import Reference
+from .decorators.routing import entrypoint, app, route
+from .agenta_init import Config, AgentaSingleton, init as _init
+from .utils.costs import calculate_token_usage
+from .managers.config import ConfigManager
+from .managers.variant import VariantManager
+from .managers.deployment import DeploymentManager
 
 config = PreInitObject("agenta.config", Config)
 DEFAULT_AGENTA_SINGLETON_INSTANCE = AgentaSingleton()
+
+types = client_types
+
+api = None
+async_api = None
+
 tracing = DEFAULT_AGENTA_SINGLETON_INSTANCE.tracing  # type: ignore
+tracer = get_tracer(tracing)
+
+
+def init(
+    host: Optional[str] = None,
+    app_id: Optional[str] = None,
+    api_key: Optional[str] = None,
+    config_fname: Optional[str] = None,
+):
+    global api, async_api, tracing, tracer
+
+    _init(
+        host=host,
+        api_key=api_key,
+        config_fname=config_fname,
+        # DEPRECATING
+        app_id=app_id,
+    )
+
+    api = DEFAULT_AGENTA_SINGLETON_INSTANCE.api  # type: ignore
+    async_api = DEFAULT_AGENTA_SINGLETON_INSTANCE.async_api  # type: ignore
+
+    tracing = DEFAULT_AGENTA_SINGLETON_INSTANCE.tracing  # type: ignore
+    tracer = get_tracer(tracing)

agenta/sdk/agenta_init.py

@@ -2,10 +2,12 @@ import os
 import logging
 import toml
 from typing import Optional
+from importlib.metadata import version
 
+from agenta.sdk.utils.logging import log
 from agenta.sdk.utils.globals import set_global
-from agenta.client.backend.client import AgentaApi
-from agenta.sdk.tracing.llm_tracing import Tracing
+from agenta.client.backend.client import AgentaApi, AsyncAgentaApi
+from agenta.sdk.tracing import Tracing
 from agenta.client.exceptions import APIRequestError
 
 
@@ -17,9 +19,11 @@ class AgentaSingleton:
     """Singleton class to save all the "global variables" for the sdk."""
 
     _instance = None
-    setup = None
     config = None
-    tracing: Optional[Tracing] = None
+    tracing = None
+
+    api = None
+    async_api = None
 
     def __new__(cls):
         if not cls._instance:
@@ -28,12 +32,15 @@ class AgentaSingleton:
 
     def init(
         self,
-        app_id: Optional[str] = None,
+        *,
         host: Optional[str] = None,
         api_key: Optional[str] = None,
         config_fname: Optional[str] = None,
+        # DEPRECATING
+        app_id: Optional[str] = None,
     ) -> None:
-        """Main function to initialize the singleton.
+        """
+        Main function to initialize the singleton.
 
         Initializes the singleton with the given `app_id`, `host`, and `api_key`. The order of precedence for these variables is:
         1. Explicit argument provided in the function call.
@@ -54,47 +61,89 @@ class AgentaSingleton:
         Raises:
             ValueError: If `app_id` is not specified either as an argument, in the config file, or in the environment variables.
         """
+
+        log.info("---------------------------")
+        log.info("Agenta SDK - using version: %s", version("agenta"))
+        log.info("---------------------------")
+
         config = {}
         if config_fname:
             config = toml.load(config_fname)
 
-        self.app_id = app_id or config.get("app_id") or os.environ.get("AGENTA_APP_ID")
         self.host = (
             host
+            or os.environ.get("AGENTA_HOST")
             or config.get("backend_host")
-            or os.environ.get("AGENTA_HOST", "https://cloud.agenta.ai")
+            or config.get("host")
+            or "https://cloud.agenta.ai"
         )
+
+        self.app_id = app_id or config.get("app_id") or os.environ.get("AGENTA_APP_ID")
+        # if not self.app_id:
+        #     raise ValueError(
+        #         "App ID must be specified. You can provide it in one of the following ways:\n"
+        #         "1. As an argument when calling ag.init(app_id='your_app_id').\n"
+        #         "2. In the configuration file specified by config_fname.\n"
+        #         "3. As an environment variable 'AGENTA_APP_ID'."
+        #     )
+
         self.api_key = (
-            api_key or config.get("api_key") or os.environ.get("AGENTA_API_KEY")
+            api_key or os.environ.get("AGENTA_API_KEY") or config.get("api_key")
+        )
+
+        self.tracing = Tracing(
+            url=f"{self.host}/api/observability/v1/otlp/traces",  # type: ignore
+        )
+
+        self.tracing.configure(
+            api_key=self.api_key,
+            # DEPRECATING
+            app_id=self.app_id,
+        )
+
+        self.api = AgentaApi(
+            base_url=self.host + "/api",
+            api_key=api_key if api_key else "",
+        )
+
+        self.async_api = AsyncAgentaApi(
+            base_url=self.host + "/api",
+            api_key=api_key if api_key else "",
         )
 
-        if not self.app_id:
-            raise ValueError(
-                "App ID must be specified. You can provide it in one of the following ways:\n"
-                "1. As an argument when calling ag.init(app_id='your_app_id').\n"
-                "2. In the configuration file specified by config_fname.\n"
-                "3. As an environment variable 'AGENTA_APP_ID'."
-            )
         self.base_id = os.environ.get("AGENTA_BASE_ID")
-        if self.base_id is None:
-            print(
-                "Warning: Your configuration will not be saved permanently since base_id is not provided."
-            )
 
-        self.config = Config(base_id=self.base_id, host=self.host, api_key=self.api_key)  # type: ignore
+        self.config = Config(
+            host=self.host,
+            base_id=self.base_id,
+            api_key=self.api_key,
+        )
 
 
 class Config:
-    def __init__(self, base_id: str, host: str, api_key: Optional[str] = ""):
-        self.base_id = base_id
+    def __init__(
+        self,
+        host: str,
+        base_id: Optional[str] = None,
+        api_key: Optional[str] = "",
+    ):
         self.host = host
 
+        self.base_id = base_id
+
+        if self.base_id is None:
+            # print(
+            #    "Warning: Your configuration will not be saved permanently since base_id is not provided.\n"
+            # )
+            pass
+
         if base_id is None or host is None:
             self.persist = False
         else:
             self.persist = True
             self.client = AgentaApi(
-                base_url=self.host + "/api", api_key=api_key if api_key else ""
+                base_url=self.host + "/api",
+                api_key=api_key if api_key else "",
             )
 
     def register_default(self, overwrite=False, **kwargs):
@@ -208,11 +257,11 @@ class Config:
 
 
 def init(
-    app_id: Optional[str] = None,
     host: Optional[str] = None,
     api_key: Optional[str] = None,
     config_fname: Optional[str] = None,
-    max_workers: Optional[int] = None,
+    # DEPRECATED
+    app_id: Optional[str] = None,
 ):
     """Main function to initialize the agenta sdk.
 
@@ -238,12 +287,15 @@ def init(
 
     singleton = AgentaSingleton()
 
-    singleton.init(app_id=app_id, host=host, api_key=api_key, config_fname=config_fname)
+    singleton.init(
+        host=host,
+        api_key=api_key,
+        config_fname=config_fname,
+        # DEPRECATED
+        app_id=app_id,
+    )
 
-    tracing = Tracing(
-        host=singleton.host,  # type: ignore
-        app_id=singleton.app_id,  # type: ignore
-        api_key=singleton.api_key,
-        max_workers=max_workers,
+    set_global(
+        config=singleton.config,
+        tracing=singleton.tracing,
     )
-    set_global(setup=singleton.setup, config=singleton.config, tracing=tracing)

agenta/sdk/context/routing.py

@@ -0,0 +1,26 @@
+from contextlib import contextmanager
+from contextvars import ContextVar
+from typing import Any, Dict, Optional
+
+routing_context = ContextVar("routing_context", default={})
+
+
+@contextmanager
+def routing_context_manager(
+    *,
+    config: Optional[Dict[str, Any]] = None,
+    application: Optional[Dict[str, Any]] = None,
+    variant: Optional[Dict[str, Any]] = None,
+    environment: Optional[Dict[str, Any]] = None,
+):
+    context = {
+        "config": config,
+        "application": application,
+        "variant": variant,
+        "environment": environment,
+    }
+    token = routing_context.set(context)
+    try:
+        yield
+    finally:
+        routing_context.reset(token)

agenta/sdk/context/tracing.py

@@ -0,0 +1,3 @@
+from contextvars import ContextVar
+
+tracing_context = ContextVar("tracing_context", default={})