agenta 0.13.8__tar.gz → 0.14.1__tar.gz

{agenta-0.13.8 → agenta-0.14.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: agenta
-Version: 0.13.8
+Version: 0.14.1
 Summary: The SDK for agenta is an open-source LLMOps platform.
 Home-page: https://agenta.ai
 Keywords: LLMOps,LLM,evaluation,prompt engineering
@@ -15,6 +15,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.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,<7.0.0)
 Requires-Dist: fastapi (>=0.96.1)

{agenta-0.13.8 → agenta-0.14.1}/agenta/__init__.py

@@ -17,6 +17,6 @@ from .sdk.types import (
 from .sdk.tracing.decorators import span
 from .sdk.agenta_init import Config, init, llm_tracing
 from .sdk.utils.helper.openai_cost import calculate_token_usage
-
+from .sdk.client import Agenta
 
 config = PreInitObject("agenta.config", Config)

agenta-0.14.1/agenta/sdk/client.py

@@ -0,0 +1,56 @@
+import os
+
+from cachetools import TTLCache, cached
+
+from agenta.client.backend.client import AgentaApi
+
+
+class Agenta:
+    """Client class for interacting with the Agenta API."""
+
+    def __init__(self, api_key: str = None, host: str = None):
+        """
+        Initializes the Agenta client with API key and host.
+
+        Raises:
+            EnvironmentError: If AGENTA_API_KEY is not set.
+        """
+        if not api_key and not os.environ.get("AGENTA_API_KEY"):
+            raise EnvironmentError(
+                "Required environment variables AGENTA_API_KEY is not set."
+            )
+        self.api_key = api_key if api_key else os.environ.get("AGENTA_API_KEY")
+        self.host = (
+            host if host else os.environ.get("AGENTA_HOST", "https://cloud.agenta.ai")
+        )
+        self.cache = TTLCache(maxsize=1024, ttl=300)
+        backend_url = f"{self.host}/api"
+        self.client = AgentaApi(base_url=backend_url, api_key=self.api_key)
+
+    def get_config(self, base_id: str, environment: str, cache_timeout: int = 300):
+        """
+        Fetches and caches the configuration for a specified base ID and environment.
+
+        Args:
+            base_id (str): The unique identifier for the base.
+            environment (str): The environment name (e.g., 'production', 'development').
+            cache_timeout (int): The TTL for the cache in seconds. Defaults to 300 seconds.
+
+        Returns:
+            dict: The configuration data retrieved from the Agenta API.
+
+        Raises:
+            EnvironmentError: If the required AGENTA_API_KEY is not set in the environment variables.
+        """
+        if cache_timeout != self.cache.ttl:
+            self.cache = TTLCache(
+                maxsize=1024, ttl=cache_timeout
+            )  # TODO: We need to modify this to use a dynamic TTLCache implementation in the future
+
+        @cached(cache=self.cache)
+        def fetch_config(base_id: str, environment: str = "production"):
+            return self.client.configs.get_config(
+                base_id=base_id, environment_name=environment
+            )
+
+        return fetch_config(base_id, environment)

{agenta-0.13.8 → agenta-0.14.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "agenta"
-version = "0.13.8"
+version = "0.14.1"
 description = "The SDK for agenta is an open-source LLMOps platform."
 readme = "README.md"
 authors = ["Mahmoud Mabrouk <mahmoud@agenta.ai>"]
@@ -30,6 +30,7 @@ posthog = "^3.1.0"
 pydantic = "1.10.13"
 httpx = "^0.27.0"
 pymongo = "^4.6.3"
+cachetools = "^5.3.3"
 
 [tool.poetry.dev-dependencies]
 pytest = "^6.2"