agenta 0.14.12a0__py3-none-any.whl → 0.14.12a2__py3-none-any.whl

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/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/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
 

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/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/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.dist-info → agenta-0.14.12a2.dist-info}/METADATA

@@ -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.dist-info → agenta-0.14.12a2.dist-info}/RECORD

@@ -1,4 +1,4 @@
-agenta/__init__.py,sha256=yXks3d7zLmyJCM3D_L5rk4GnFKHZX2NJa6j6wjAJUGQ,661
+agenta/__init__.py,sha256=Op8Re2QqGz2vqAzuYRgR7S5ACKh5UdpJEA4KlG8SWvo,688
 agenta/cli/evaluation_commands.py,sha256=fs6492tprPId9p8eGO02Xy-NCBm2RZNJLZWcUxugwd8,474
 agenta/cli/helper.py,sha256=vRxHyeNaltzNIGrfU2vO0H28_rXDzx9QqIZ_S-W6zL4,6212
 agenta/cli/main.py,sha256=Wz0ODhoeKK3Qg_CFUhu6D909szk05tc8ZVBB6H1-w7k,9763
@@ -119,20 +119,20 @@ agenta/client/client.py,sha256=DWOGS9A8u4wu28s9jGOR4eRhf7vo4zT7GyDvrIGu59Y,19648
 agenta/client/exceptions.py,sha256=cxLjjKvZKlUgBxt4Vn9J_SsezJPPNHvrZxnoq-D6zmw,94
 agenta/config.py,sha256=Id-Ie1yf9QRP1YPhRYaYSOruRe6RBrsCXkG9rAa-ZtA,732
 agenta/config.toml,sha256=ptE0P49bwsu3Luyn7OLFmk2buPhj5D-MA-O_ErOGoLg,223
-agenta/docker/docker-assets/Dockerfile.cloud.template,sha256=f8VU9fs0mMDISGlo3kAYzF8YyDAKgK9FJQHl4PJwykw,386
-agenta/docker/docker-assets/Dockerfile.template,sha256=jkZM1dOPUyQiO9q4jn_RR34-A76lhv5CHJmP0OJHyoE,280
+agenta/docker/docker-assets/Dockerfile.cloud.template,sha256=uJuXKvtkMY6f4KaOh3XE5pmuJR7mfZEXJk_8hj2uatc,386
+agenta/docker/docker-assets/Dockerfile.template,sha256=aVA_okx0xXalcTvdQGhSfzSjNpQZVoLJCGYA39-2Nwk,280
 agenta/docker/docker-assets/README.md,sha256=XHxwh2ks_ozrtAU7SLbL3J14SB2holG6buoTxwmMiZM,102
 agenta/docker/docker-assets/entrypoint.sh,sha256=29XK8VQjQsx4hN2j-4JDy-6kQb5y4LCqZEa7PD4eqCQ,74
 agenta/docker/docker-assets/lambda_function.py,sha256=h4UZSSfqwpfsCgERv6frqwm_4JrYu9rLz3I-LxCfeEg,83
 agenta/docker/docker-assets/main.py,sha256=7MI-21n81U7N7A0GxebNi0cmGWtJKcR2sPB6FcH2QfA,251
 agenta/docker/docker_utils.py,sha256=5uHMCzXkCvIsDdEiwbnnn97KkzsFbBvyMwogCsv_Z5U,3509
-agenta/sdk/__init__.py,sha256=Zqst2m-u5HbV_Y6sGcT7suwbP-XanP-5noLpfKlA2nQ,627
-agenta/sdk/agenta_init.py,sha256=KjxMOewpADtlhOAqTbW-T2TQ1oVwKncV1-0ByNwXBpE,8644
+agenta/sdk/__init__.py,sha256=nQS3x2sE_3kuRp6gEhShS_jAZ8hbSBs8i4mkWqdGT6I,618
+agenta/sdk/agenta_init.py,sha256=8D55zEse0YlBhkQGevj2FC8gP1YnMqOo_oEg1CPSyAs,8666
 agenta/sdk/client.py,sha256=trKyBOYFZRk0v5Eptxvh87yPf50Y9CqY6Qgv4Fy-VH4,2142
 agenta/sdk/context.py,sha256=q-PxL05-I84puunUAs9LGsffEXcYhDxhQxjuOz2vK90,901
 agenta/sdk/decorators/base.py,sha256=9aNdX5h8a2mFweuhdO-BQPwXGKY9ONPIdLRhSGAGMfY,217
 agenta/sdk/decorators/llm_entrypoint.py,sha256=qU_TyYAbhoj-Cf-SSAKzo11Le9qpH-Z4bA_kKQ6iJk0,18643
-agenta/sdk/decorators/tracing.py,sha256=w8qBj1P8ft2i7K5NoQBIGEHxfAyfNLiSlvgyRB9m0Mo,2446
+agenta/sdk/decorators/tracing.py,sha256=L0M5oPP6NIZihReBFzYB9JVn60QZghR8ZUTBuok9vyY,2375
 agenta/sdk/router.py,sha256=0sbajvn5C7t18anH6yNo7-oYxldHnYfwcbmQnIXBePw,269
 agenta/sdk/tracing/context_manager.py,sha256=HskDaiORoOhjeN375gm05wYnieQzh5UnoIsnSAHkAyc,252
 agenta/sdk/tracing/llm_tracing.py,sha256=CLijhERu9LYbi-UrUTsUT_jyb_0dT5iO_0o2MpjXk1g,8038
@@ -157,7 +157,7 @@ agenta/templates/simple_prompt/app.py,sha256=kODgF6lhzsaJPdgL5b21bUki6jkvqjWZzWR
 agenta/templates/simple_prompt/env.example,sha256=g9AE5bYcGPpxawXMJ96gh8oenEPCHTabsiOnfQo3c5k,70
 agenta/templates/simple_prompt/requirements.txt,sha256=ywRglRy7pPkw8bljmMEJJ4aOOQKrt9FGKULZ-DGkoBU,23
 agenta/templates/simple_prompt/template.toml,sha256=DQBtRrF4GU8LBEXOZ-GGuINXMQDKGTEG5y37tnvIUIE,60
-agenta-0.14.12a0.dist-info/METADATA,sha256=iS-y1oTXaY-6bHnUAIzCa2u7RmDdIJAuY5Kevj75gBI,26466
-agenta-0.14.12a0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-agenta-0.14.12a0.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
-agenta-0.14.12a0.dist-info/RECORD,,
+agenta-0.14.12a2.dist-info/METADATA,sha256=37JWVWBz4mFIQCZLwACxqhOamzn-Z3VTALgFenew_A4,26466
+agenta-0.14.12a2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+agenta-0.14.12a2.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
+agenta-0.14.12a2.dist-info/RECORD,,