agenta 0.14.14a1__tar.gz → 0.15.0__tar.gz

{agenta-0.14.14a1 → agenta-0.15.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: agenta
-Version: 0.14.14a1
+Version: 0.15.0
 Summary: The SDK for agenta is an open-source LLMOps platform.
 Home-page: https://agenta.ai
 Keywords: LLMOps,LLM,evaluation,prompt engineering
@@ -18,12 +18,12 @@ Classifier: Topic :: Software Development :: Libraries
 Requires-Dist: cachetools (>=5.3.3,<6.0.0)
 Requires-Dist: click (>=8.1.3,<9.0.0)
 Requires-Dist: docker (>=6.1.1,<8.0.0)
-Requires-Dist: fastapi (>=0.111.0,<0.112.0)
+Requires-Dist: fastapi (>=0.96.1)
 Requires-Dist: httpx (>=0.24,<0.28)
 Requires-Dist: importlib-metadata (>=6.7,<8.0)
 Requires-Dist: ipdb (>=0.13)
 Requires-Dist: posthog (>=3.1.0,<4.0.0)
-Requires-Dist: pydantic (>=2.7.1,<3.0.0)
+Requires-Dist: pydantic (==1.10.13)
 Requires-Dist: pymongo (>=4.6.3,<5.0.0)
 Requires-Dist: python-dotenv (>=1.0.0,<2.0.0)
 Requires-Dist: python-multipart (>=0.0.6,<0.0.10)

{agenta-0.14.14a1 → agenta-0.15.0}/agenta/__init__.py

@@ -1,5 +1,4 @@
 from .sdk.utils.preinit import PreInitObject
-from .sdk.agenta_decorator import app, entrypoint
 from .sdk.context import get_contexts, save_context
 from .sdk.types import (
     Context,
@@ -14,9 +13,13 @@ from .sdk.types import (
     FileInputURL,
     BinaryParam,
 )
-from .sdk.tracing.decorators import span
-from .sdk.agenta_init import Config, init, llm_tracing
+from .sdk.tracing.llm_tracing import Tracing
+from .sdk.decorators.tracing import instrument
+from .sdk.decorators.llm_entrypoint import entrypoint, app
+from .sdk.agenta_init import Config, AgentaSingleton, init
 from .sdk.utils.helper.openai_cost import calculate_token_usage
 from .sdk.client import Agenta
 
 config = PreInitObject("agenta.config", Config)
+DEFAULT_AGENTA_SINGLETON_INSTANCE = AgentaSingleton()
+tracing = DEFAULT_AGENTA_SINGLETON_INSTANCE.tracing  # type: ignore

{agenta-0.14.14a1 → agenta-0.15.0}/agenta/client/backend/client.py

@@ -56,9 +56,9 @@ class AgentaApi:
         self._client_wrapper = SyncClientWrapper(
             base_url=base_url,
             api_key=api_key,
-            httpx_client=httpx.Client(timeout=timeout)
-            if httpx_client is None
-            else httpx_client,
+            httpx_client=(
+                httpx.Client(timeout=timeout) if httpx_client is None else httpx_client
+            ),
         )
         self.observability = ObservabilityClient(client_wrapper=self._client_wrapper)
         self.apps = AppsClient(client_wrapper=self._client_wrapper)
@@ -1037,9 +1037,11 @@ class AsyncAgentaApi:
         self._client_wrapper = AsyncClientWrapper(
             base_url=base_url,
             api_key=api_key,
-            httpx_client=httpx.AsyncClient(timeout=timeout)
-            if httpx_client is None
-            else httpx_client,
+            httpx_client=(
+                httpx.AsyncClient(timeout=timeout)
+                if httpx_client is None
+                else httpx_client
+            ),
         )
         self.observability = AsyncObservabilityClient(
             client_wrapper=self._client_wrapper

{agenta-0.14.14a1 → agenta-0.15.0}/agenta/client/backend/types/create_span.py

@@ -53,6 +53,6 @@ class CreateSpan(pydantic.BaseModel):
         return super().dict(**kwargs_with_defaults)
 
     class Config:
-        frozen = True
+        frozen = False
         smart_union = True
         json_encoders = {dt.datetime: serialize_datetime}

{agenta-0.14.14a1 → agenta-0.15.0}/agenta/docker/docker-assets/Dockerfile.cloud.template

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

{agenta-0.14.14a1 → agenta-0.15.0}/agenta/sdk/__init__.py

@@ -1,6 +1,4 @@
 from .utils.preinit import PreInitObject  # always the first import!
-from . import agenta_decorator, context, types, utils  # noqa: F401
-from .agenta_decorator import app, entrypoint
 from .context import get_contexts, save_context
 from .types import (
     Context,
@@ -15,9 +13,13 @@ from .types import (
     FileInputURL,
     BinaryParam,
 )
-from .tracing.decorators import span
-from .agenta_init import Config, init, llm_tracing
+from .tracing.llm_tracing import Tracing
+from .decorators.tracing import instrument
+from .decorators.llm_entrypoint import entrypoint, app
+from .agenta_init import Config, AgentaSingleton, init
 from .utils.helper.openai_cost import calculate_token_usage
 
 
 config = PreInitObject("agenta.config", Config)
+DEFAULT_AGENTA_SINGLETON_INSTANCE = AgentaSingleton()
+tracing = DEFAULT_AGENTA_SINGLETON_INSTANCE.tracing  # type: ignore

agenta-0.15.0/agenta/sdk/agenta_init.py

@@ -0,0 +1,256 @@
+import os
+import logging
+import toml
+from typing import Optional
+
+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.exceptions import APIRequestError
+
+
+logger = logging.getLogger(__name__)
+logger.setLevel(logging.DEBUG)
+
+
+class AgentaSingleton:
+    """Singleton class to save all the "global variables" for the sdk."""
+
+    _instance = None
+    setup = None
+    config = None
+    tracing: Optional[Tracing] = None
+
+    def __new__(cls):
+        if not cls._instance:
+            cls._instance = super(AgentaSingleton, cls).__new__(cls)
+        return cls._instance
+
+    @property
+    def client(self):
+        """API Backend client.
+
+        Returns:
+            AgentaAPI: instance of agenta api backend
+        """
+
+        return AgentaApi(base_url=self.host + "/api", api_key=self.api_key)
+
+    def init(
+        self,
+        app_id: Optional[str] = None,
+        host: Optional[str] = None,
+        api_key: Optional[str] = None,
+        config_fname: Optional[str] = None,
+    ) -> None:
+        """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.
+        2. Value from the configuration file specified by `config_fname`.
+        3. Environment variables.
+
+        Examples:
+        ag.init(app_id="xxxx", api_key="xxx")
+        ag.init(config_fname="config.toml")
+        ag.init() #assuming env vars are set
+
+        Args:
+            app_id (Optional[str]): ID of the Agenta application. Defaults to None. If not provided, will look for "app_id" in the config file, then "AGENTA_APP_ID" in environment variables.
+            host (Optional[str]): Host name of the backend server. Defaults to None. If not provided, will look for "backend_host" in the config file, then "AGENTA_HOST" in environment variables.
+            api_key (Optional[str]): API Key to use with the host of the backend server. Defaults to None. If not provided, will look for "api_key" in the config file, then "AGENTA_API_KEY" in environment variables.
+            config_fname (Optional[str]): Path to the configuration file (relative or absolute). Defaults to None.
+
+        Raises:
+            ValueError: If `app_id` is not specified either as an argument, in the config file, or in the environment variables.
+        """
+        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 config.get("backend_host")
+            or os.environ.get("AGENTA_HOST", "https://cloud.agenta.ai")
+        )
+        self.api_key = (
+            api_key or config.get("api_key") or os.environ.get("AGENTA_API_KEY")
+        )
+
+        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)  # type: ignore
+
+
+class Config:
+    def __init__(self, base_id: str, host: str, api_key: str = ""):
+        self.base_id = base_id
+        self.host = host
+
+        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)
+
+    def register_default(self, overwrite=False, **kwargs):
+        """alias for default"""
+        return self.default(overwrite=overwrite, **kwargs)
+
+    def default(self, overwrite=False, **kwargs):
+        """Saves the default parameters to the app_name and base_name in case they are not already saved.
+        Args:
+            overwrite: Whether to overwrite the existing configuration or not
+            **kwargs: A dict containing the parameters
+        """
+        self.set(
+            **kwargs
+        )  # In case there is no connectivity, we still can use the default values
+        try:
+            self.push(config_name="default", overwrite=overwrite, **kwargs)
+        except Exception as ex:
+            logger.warning(
+                "Unable to push the default configuration to the server. %s", str(ex)
+            )
+
+    def push(self, config_name: str, overwrite=True, **kwargs):
+        """Pushes the parameters for the app variant to the server
+        Args:
+            config_name: Name of the configuration to push to
+            overwrite: Whether to overwrite the existing configuration or not
+            **kwargs: A dict containing the parameters
+        """
+        if not self.persist:
+            return
+        try:
+            self.client.configs.save_config(
+                base_id=self.base_id,
+                config_name=config_name,
+                parameters=kwargs,
+                overwrite=overwrite,
+            )
+        except Exception as ex:
+            logger.warning(
+                "Failed to push the configuration to the server with error: %s", ex
+            )
+
+    def pull(
+        self, config_name: str = "default", environment_name: Optional[str] = None
+    ):
+        """Pulls the parameters for the app variant from the server and sets them to the config"""
+        if not self.persist and (
+            config_name != "default" or environment_name is not None
+        ):
+            raise ValueError(
+                "Cannot pull the configuration from the server since the app_name and base_name are not provided."
+            )
+        if self.persist:
+            try:
+                if environment_name:
+                    config = self.client.configs.get_config(
+                        base_id=self.base_id, environment_name=environment_name
+                    )
+
+                else:
+                    config = self.client.configs.get_config(
+                        base_id=self.base_id,
+                        config_name=config_name,
+                    )
+            except Exception as ex:
+                logger.warning(
+                    "Failed to pull the configuration from the server with error: %s",
+                    str(ex),
+                )
+        try:
+            self.set(**{"current_version": config.current_version, **config.parameters})
+        except Exception as ex:
+            logger.warning("Failed to set the configuration with error: %s", str(ex))
+
+    def all(self):
+        """Returns all the parameters for the app variant"""
+        return {
+            k: v
+            for k, v in self.__dict__.items()
+            if k
+            not in [
+                "app_name",
+                "base_name",
+                "host",
+                "base_id",
+                "api_key",
+                "persist",
+                "client",
+            ]
+        }
+
+    # function to set the parameters for the app variant
+    def set(self, **kwargs):
+        """Sets the parameters for the app variant
+
+        Args:
+            **kwargs: A dict containing the parameters
+        """
+        for key, value in kwargs.items():
+            setattr(self, key, value)
+
+    def dump(self):
+        """Returns all the information about the current version in the configuration.
+
+        Raises:
+            NotImplementedError: _description_
+        """
+
+        raise NotImplementedError()
+
+
+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,
+):
+    """Main function to initialize the agenta sdk.
+
+    Initializes agenta 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.
+    2. Value from the configuration file specified by `config_fname`.
+    3. Environment variables.
+
+    - `app_id` is a required parameter (to be specified in one of the above ways)
+    - `host` is optional and defaults to "https://cloud.agenta.ai"
+    - `api_key` is optional and defaults to "". It is required only when using cloud or enterprise version of agenta.
+
+
+    Args:
+        app_id (Optional[str]): ID of the Agenta application. Defaults to None. If not provided, will look for "app_id" in the config file, then "AGENTA_APP_ID" in environment variables.
+        host (Optional[str]): Host name of the backend server. Defaults to None. If not provided, will look for "backend_host" in the config file, then "AGENTA_HOST" in environment variables.
+        api_key (Optional[str]): API Key to use with the host of the backend server. Defaults to None. If not provided, will look for "api_key" in the config file, then "AGENTA_API_KEY" in environment variables.
+        config_fname (Optional[str]): Path to the configuration file. Defaults to None.
+
+    Raises:
+        ValueError: If `app_id` is not specified either as an argument, in the config file, or in the environment variables.
+    """
+
+    singleton = AgentaSingleton()
+
+    singleton.init(app_id=app_id, host=host, api_key=api_key, config_fname=config_fname)
+    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(setup=singleton.setup, config=singleton.config, tracing=tracing)

agenta-0.15.0/agenta/sdk/decorators/base.py

@@ -0,0 +1,10 @@
+# Stdlib Imports
+from typing import Any, Callable
+
+
+class BaseDecorator:
+    def __init__(self):
+        pass
+
+    def __call__(self, func: Callable[..., Any]) -> Callable[..., Any]:
+        raise NotImplementedError