agenta 0.14.12a0__tar.gz → 0.14.12a2__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: agenta
-Version: 0.14.12a0
+Version: 0.14.12a2
 Summary: The SDK for agenta is an open-source LLMOps platform.
 Home-page: https://agenta.ai
 Keywords: LLMOps,LLM,evaluation,prompt engineering

{agenta-0.14.12a0 → agenta-0.14.12a2}/agenta/__init__.py

@@ -14,7 +14,8 @@ from .sdk.types import (
     BinaryParam,
 )
 from .sdk.tracing.llm_tracing import Tracing
-from .sdk.decorators.tracing import instrument, tracing
+from .sdk.decorators import tracing
+from .sdk.decorators.tracing import instrument
 from .sdk.decorators.llm_entrypoint import entrypoint
 from .sdk.agenta_init import Config, init
 from .sdk.utils.helper.openai_cost import calculate_token_usage

{agenta-0.14.12a0 → agenta-0.14.12a2}/agenta/sdk/__init__.py

@@ -14,7 +14,7 @@ from .types import (
     BinaryParam,
 )
 from .tracing.llm_tracing import Tracing
-from .decorators.tracing import instrument, tracing
+from .decorators.tracing import instrument
 from .decorators.llm_entrypoint import entrypoint
 from .agenta_init import Config, init
 from .utils.helper.openai_cost import calculate_token_usage

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

@@ -86,7 +86,7 @@ class AgentaSingleton:
 
         self.variant_id = os.environ.get("AGENTA_VARIANT_ID")
         self.variant_name = os.environ.get("AGENTA_VARIANT_NAME")
-        self.config = Config(base_id=self.base_id, host=self.host)
+        self.config = Config(base_id=self.base_id, host=self.host, api_key=self.api_key)
 
     def get_app(self, app_name: str) -> str:
         apps = self.client.apps.list_apps(app_name=app_name)
@@ -113,9 +113,10 @@ class AgentaSingleton:
 
 
 class Config:
-    def __init__(self, base_id, host):
+    def __init__(self, base_id: str, host: str, api_key: str):
         self.base_id = base_id
         self.host = host
+        self.api_key = api_key
 
         if base_id is None or host is None:
             self.persist = False
@@ -130,8 +131,7 @@ class Config:
             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()
+        return AgentaApi(base_url=self.host + "/api", api_key=self.api_key)
 
     def register_default(self, overwrite=False, **kwargs):
         """alias for default"""

{agenta-0.14.12a0 → agenta-0.14.12a2}/agenta/sdk/decorators/tracing.py

@@ -9,10 +9,6 @@ import agenta as ag
 from agenta.sdk.decorators.base import BaseDecorator
 
 
-# Set global tracing variable to None
-tracing = None
-
-
 class instrument(BaseDecorator):
     """Decorator class for monitoring llm apps functions.
 
@@ -46,9 +42,9 @@ class instrument(BaseDecorator):
             api_key=os.environ["AGENTA_API_KEY"],
         )
 
-        # Update tracing variable to be global
-        global tracing
-        tracing = self.tracing
+        # Set span for global access
+        global span
+        span = self.tracing
 
     def __call__(self, func: Callable[..., Any]):
         @wraps(func)

{agenta-0.14.12a0 → agenta-0.14.12a2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "agenta"
-version = "0.14.12a0"
+version = "0.14.12a2"
 description = "The SDK for agenta is an open-source LLMOps platform."
 readme = "README.md"
 authors = ["Mahmoud Mabrouk <mahmoud@agenta.ai>"]

{agenta-0.14.12a0 → agenta-0.14.12a2}/agenta/docker/docker-assets/Dockerfile.cloud.template

@@ -1,9 +1,9 @@
 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 mangum
-RUN pip install --no-cache-dir --disable-pip-version-check -U agenta
 COPY . ${LAMBDA_TASK_ROOT}
 
 CMD [ "lambda_function.handler" ]

{agenta-0.14.12a0 → agenta-0.14.12a2}/agenta/docker/docker-assets/Dockerfile.template

@@ -4,8 +4,8 @@ WORKDIR /app
 
 COPY . .
 
-RUN pip install --no-cache-dir --disable-pip-version-check -r requirements.txt
 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
 
 EXPOSE 80