agenta 0.14.11__tar.gz → 0.14.12a0__tar.gz

{agenta-0.14.11 → agenta-0.14.12a0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: agenta
-Version: 0.14.11
+Version: 0.14.12a0
 Summary: The SDK for agenta is an open-source LLMOps platform.
 Home-page: https://agenta.ai
 Keywords: LLMOps,LLM,evaluation,prompt engineering
@@ -23,7 +23,7 @@ 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)
+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.11 → agenta-0.14.12a0}/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,8 +13,10 @@ 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, tracing
+from .sdk.decorators.llm_entrypoint import entrypoint
+from .sdk.agenta_init import Config, init
 from .sdk.utils.helper.openai_cost import calculate_token_usage
 from .sdk.client import Agenta
 

{agenta-0.14.11 → agenta-0.14.12a0}/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.11 → agenta-0.14.12a0}/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.11 → agenta-0.14.12a0}/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,8 +13,10 @@ 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, tracing
+from .decorators.llm_entrypoint import entrypoint
+from .agenta_init import Config, init
 from .utils.helper.openai_cost import calculate_token_usage
 
 

{agenta-0.14.11 → agenta-0.14.12a0}/agenta/sdk/agenta_init.py

@@ -1,11 +1,9 @@
 import os
 import logging
-from typing import Any, Optional
-
-from .utils.globals import set_global
+from typing import Optional, Any
 
+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
 
 
@@ -13,18 +11,6 @@ logger = logging.getLogger(__name__)
 logger.setLevel(logging.DEBUG)
 
 
-BACKEND_URL_SUFFIX = os.environ.get("BACKEND_URL_SUFFIX", "api")
-CLIENT_API_KEY = os.environ.get("AGENTA_API_KEY")
-CLIENT_HOST = os.environ.get("AGENTA_HOST", "http://localhost")
-
-# initialize the client with the backend url and api key
-backend_url = f"{CLIENT_HOST}/{BACKEND_URL_SUFFIX}"
-client = AgentaApi(
-    base_url=backend_url,
-    api_key=CLIENT_API_KEY if CLIENT_API_KEY else "",
-)
-
-
 class AgentaSingleton:
     """Singleton class to save all the "global variables" for the sdk."""
 
@@ -37,6 +23,16 @@ class AgentaSingleton:
             cls._instance = super(AgentaSingleton, cls).__new__(cls)
         return cls._instance
 
+    @property
+    def client(self):
+        """Builds sdk client instance.
+
+        Returns:
+            AgentaAPI: instance of agenta api backend
+        """
+
+        return AgentaApi(base_url=self.host + "/api", api_key=self.api_key)
+
     def init(
         self,
         app_name: Optional[str] = None,
@@ -49,53 +45,51 @@ class AgentaSingleton:
     ) -> None:
         """Main function to initialize the singleton.
 
-        Initializes the singleton with the given `app_name`, `base_name`, and `host`. If any of these arguments are not provided,
-        the function will look for them in environment variables.
+        Initializes the singleton with the given `app_name`, `base_name`, and `host`.
+        If any of these arguments are not provided, the function will look for them
+        in environment variables.
 
         Args:
-            app_name (Optional[str]): Name of the Agenta application. Defaults to None. If not provided, will look for "AGENTA_APP_NAME" in environment variables.
-            base_name (Optional[str]): Base name for the Agenta setup. Defaults to None. If not provided, will look for "AGENTA_BASE_NAME" in environment variables.
-            host (Optional[str]): Host name of the backend server. Defaults to None. If not provided, will look for "AGENTA_HOST" in environment variables.
+            app_name (Optional[str]): Name of the application. Defaults to None.
+            base_name (Optional[str]): Base name for the setup. Defaults to None.
+            api_key (Optional[str]): API Key to use with the host. Defaults to None.
+            base_id (Optional[str]): Base ID for the setup. Defaults to None.
+            app_id (Optional[str]): App ID. Defaults to None.
+            host (Optional[str]): Host name of the backend server. Defaults to "http://localhost".
             kwargs (Any): Additional keyword arguments.
 
         Raises:
-            ValueError: If `app_name`, `base_name`, or `host` are not specified either as arguments or in the environment variables.
+            ValueError: If `app_name`, `base_name`, or `host` are not specified either as arguments
+            or in the environment variables.
         """
-        if app_name is None:
-            app_name = os.environ.get("AGENTA_APP_NAME")
-        if base_name is None:
-            base_name = os.environ.get("AGENTA_BASE_NAME")
-        if api_key is None:
-            api_key = os.environ.get("AGENTA_API_KEY")
-        if base_id is None:
-            base_id = os.environ.get("AGENTA_BASE_ID")
-        if host is None:
-            host = os.environ.get("AGENTA_HOST", "http://localhost")
-
-        if base_id is None:
-            if app_name is None or base_name is None:
-                print(
-                    f"Warning: Your configuration will not be saved permanently since app_name and base_name are not provided."
+
+        self.app_name = app_name or os.environ.get("AGENTA_APP_NAME")
+        self.base_name = base_name or os.environ.get("AGENTA_BASE_NAME")
+        self.api_key = api_key or os.environ.get("AGENTA_API_KEY")
+        self.base_id = base_id or os.environ.get("AGENTA_BASE_ID")
+        self.app_id = app_id or os.environ.get("AGENTA_APP_ID")
+        self.host = host or os.environ.get("AGENTA_HOST", "http://localhost")
+
+        if not self.app_id and (not self.app_name or not self.base_name):
+            print(
+                f"Warning: Your configuration will not be saved permanently since app_name and base_name are not provided."
+            )
+
+        if not self.base_id and self.app_name and self.base_name:
+            try:
+                self.app_id = self.get_app(self.app_name)
+                self.base_id = self.get_app_base(self.app_id, self.base_name)
+            except Exception as ex:
+                raise APIRequestError(
+                    f"Failed to get base id and/or app_id from the server with error: {ex}"
                 )
-            else:
-                try:
-                    app_id = self.get_app(app_name)
-                    base_id = self.get_app_base(app_id, base_name)
-                except Exception as ex:
-                    raise APIRequestError(
-                        f"Failed to get base id and/or app_id from the server with error: {ex}"
-                    )
 
-        self.base_id = base_id
-        self.host = host
-        self.app_id = os.environ.get("AGENTA_APP_ID") if app_id is None else app_id
         self.variant_id = os.environ.get("AGENTA_VARIANT_ID")
         self.variant_name = os.environ.get("AGENTA_VARIANT_NAME")
-        self.api_key = api_key
-        self.config = Config(base_id=base_id, host=host)
+        self.config = Config(base_id=self.base_id, host=self.host)
 
     def get_app(self, app_name: str) -> str:
-        apps = client.apps.list_apps(app_name=app_name)
+        apps = self.client.apps.list_apps(app_name=app_name)
         if len(apps) == 0:
             raise APIRequestError(f"App with name {app_name} not found")
 
@@ -103,7 +97,7 @@ class AgentaSingleton:
         return app_id
 
     def get_app_base(self, app_id: str, base_name: str) -> str:
-        bases = client.bases.list_bases(app_id=app_id, base_name=base_name)
+        bases = self.client.bases.list_bases(app_id=app_id, base_name=base_name)
         if len(bases) == 0:
             raise APIRequestError(f"No base was found for the app {app_id}")
         return bases[0].base_id
@@ -122,11 +116,23 @@ class Config:
     def __init__(self, base_id, host):
         self.base_id = base_id
         self.host = host
+
         if base_id is None or host is None:
             self.persist = False
         else:
             self.persist = True
 
+    @property
+    def client(self):
+        """Builds sdk client instance.
+
+        Returns:
+            AgentaAPI: instance of agenta api backend
+        """
+
+        sdk_client = SDKClient(api_key=self.api_key, host=self.host)  # type: ignore
+        return sdk_client._build_sdk_client()
+
     def register_default(self, overwrite=False, **kwargs):
         """alias for default"""
         return self.default(overwrite=overwrite, **kwargs)
@@ -157,7 +163,7 @@ class Config:
         if not self.persist:
             return
         try:
-            client.configs.save_config(
+            self.client.configs.save_config(
                 base_id=self.base_id,
                 config_name=config_name,
                 parameters=kwargs,
@@ -168,7 +174,9 @@ class Config:
                 "Failed to push the configuration to the server with error: " + str(ex)
             )
 
-    def pull(self, config_name: str = "default", environment_name: str = None):
+    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
@@ -179,12 +187,12 @@ class Config:
         if self.persist:
             try:
                 if environment_name:
-                    config = client.configs.get_config(
+                    config = self.client.configs.get_config(
                         base_id=self.base_id, environment_name=environment_name
                     )
 
                 else:
-                    config = client.configs.get_config(
+                    config = self.client.configs.get_config(
                         base_id=self.base_id,
                         config_name=config_name,
                     )
@@ -225,20 +233,8 @@ def init(app_name=None, base_name=None, **kwargs):
         app_name: _description_. Defaults to None.
         base_name: _description_. Defaults to None.
     """
+
     singleton = AgentaSingleton()
+
     singleton.init(app_name=app_name, base_name=base_name, **kwargs)
     set_global(setup=singleton.setup, config=singleton.config)
-
-
-def llm_tracing(max_workers: Optional[int] = None) -> Tracing:
-    """Function to start llm tracing."""
-
-    singleton = AgentaSingleton()
-    return Tracing(
-        base_url=singleton.host,
-        app_id=singleton.app_id,  # type: ignore
-        variant_id=singleton.variant_id,  # type: ignore
-        variant_name=singleton.variant_name,
-        api_key=singleton.api_key,
-        max_workers=max_workers,
-    )

agenta-0.14.12a0/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