agenta 0.15.0a1__py3-none-any.whl → 0.15.0a3__py3-none-any.whl

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/sdk/agenta_init.py

@@ -12,6 +12,19 @@ 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."""
 
@@ -56,9 +69,6 @@ class AgentaSingleton:
             ValueError: If `app_name`, `base_name`, or `host` are not specified either as arguments or in the environment variables.
         """
 
-        self.api_key = api_key or os.environ.get("AGENTA_API_KEY")
-        self.host = host or os.environ.get("AGENTA_HOST", "http://localhost")
-
         app_id = app_id or os.environ.get("AGENTA_APP_ID")
         if not app_id:
             raise ValueError("App ID must be specified.")
@@ -79,12 +89,14 @@ class AgentaSingleton:
 
         self.app_id = app_id
         self.base_id = base_id
+        self.host = host
+        self.api_key = api_key or ""
         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, api_key=self.api_key)  # type: ignore
+        self.config = Config(base_id=self.base_id, host=self.host)  # type: ignore
 
     def get_app_base(self, app_id: str, base_name: str) -> str:
-        bases = self.client.bases.list_bases(app_id=app_id, base_name=base_name)
+        bases = 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
@@ -100,26 +112,15 @@ class AgentaSingleton:
 
 
 class Config:
-    def __init__(self, base_id: str, host: str, api_key: str):
+    def __init__(self, base_id: str, host: 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
         else:
             self.persist = True
 
-    @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 register_default(self, overwrite=False, **kwargs):
         """alias for default"""
         return self.default(overwrite=overwrite, **kwargs)
@@ -150,7 +151,7 @@ class Config:
         if not self.persist:
             return
         try:
-            self.client.configs.save_config(
+            client.configs.save_config(
                 base_id=self.base_id,
                 config_name=config_name,
                 parameters=kwargs,
@@ -174,12 +175,12 @@ class Config:
         if self.persist:
             try:
                 if environment_name:
-                    config = self.client.configs.get_config(
+                    config = client.configs.get_config(
                         base_id=self.base_id, environment_name=environment_name
                     )
 
                 else:
-                    config = self.client.configs.get_config(
+                    config = client.configs.get_config(
                         base_id=self.base_id,
                         config_name=config_name,
                     )
@@ -244,7 +245,7 @@ def init(
         app_id=singleton.app_id,  # type: ignore
         variant_id=singleton.variant_id,  # type: ignore
         variant_name=singleton.variant_name,
-        api_key=singleton.api_key,
+        api_key=api_key,
         max_workers=max_workers,
     )
     set_global(setup=singleton.setup, config=singleton.config, tracing=tracing)

agenta/sdk/decorators/llm_entrypoint.py

@@ -82,7 +82,7 @@ class entrypoint(BaseDecorator):
 
             # Set the configuration and environment of the LLM app parent span at run-time
             agenta.tracing.set_span_attribute(
-               {"config": config_params, "environment": "playground"}
+                {"config": config_params, "environment": "playground"}
             )
 
             llm_result = await self.execute_function(
@@ -105,7 +105,7 @@ class entrypoint(BaseDecorator):
 
             # Set the configuration and environment of the LLM app parent span at run-time
             agenta.tracing.set_span_attribute(
-               {"config": config_params, "environment": kwargs["environment"]}
+                {"config": config_params, "environment": kwargs["environment"]}
             )
 
             llm_result = await self.execute_function(
@@ -391,8 +391,8 @@ class entrypoint(BaseDecorator):
 
         # Set the configuration and environment of the LLM app parent span at run-time
         agenta.tracing.set_span_attribute(
-               {"config": agenta.config.all(), "environment": "bash"}
-            )
+            {"config": agenta.config.all(), "environment": "bash"}
+        )
 
         loop = asyncio.get_event_loop()
         result = loop.run_until_complete(

agenta/sdk/tracing/llm_tracing.py

@@ -76,7 +76,9 @@ class Tracing(metaclass=SingletonMeta):
         self.active_trace_id: Optional[str] = None
         self.pending_spans: List[CreateSpan] = []
         self.tags: List[str] = []
-        self.trace_config_cache: Dict[str, Any] = {}  # used to save the trace configuration before starting the first span
+        self.trace_config_cache: Dict[
+            str, Any
+        ] = {}  # used to save the trace configuration before starting the first span
         self.span_dict: Dict[str, CreateSpan] = {}  # type: ignore
 
     @property
@@ -95,7 +97,9 @@ class Tracing(metaclass=SingletonMeta):
         self,
         attributes: Dict[str, Any] = {},
     ):
-        if self.active_span is None:  # This is the case where entrypoint wants to save the trace information but the parent span has not been initialized yet
+        if (
+            self.active_span is None
+        ):  # This is the case where entrypoint wants to save the trace information but the parent span has not been initialized yet
             for key, value in attributes.items():
                 self.trace_config_cache[key] = value
         else:

{agenta-0.15.0a1.dist-info → agenta-0.15.0a3.dist-info}/METADATA

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

{agenta-0.15.0a1.dist-info → agenta-0.15.0a3.dist-info}/RECORD

@@ -119,7 +119,7 @@ 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=uJuXKvtkMY6f4KaOh3XE5pmuJR7mfZEXJk_8hj2uatc,386
+agenta/docker/docker-assets/Dockerfile.cloud.template,sha256=Mn9g7bTQSnPXcCLZo-iVirV3cveCQHC_F7IECQufmME,389
 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
@@ -127,15 +127,15 @@ agenta/docker/docker-assets/lambda_function.py,sha256=h4UZSSfqwpfsCgERv6frqwm_4J
 agenta/docker/docker-assets/main.py,sha256=7MI-21n81U7N7A0GxebNi0cmGWtJKcR2sPB6FcH2QfA,251
 agenta/docker/docker_utils.py,sha256=o1Dn2oCr7qpg7SYk9l9RlZHwxfpWNzJpdpj8IGE0lz0,3508
 agenta/sdk/__init__.py,sha256=oHgl-qoEyi3d2VI_Kv-rIMSx9zgs6b5MP62PLq5GqYI,762
-agenta/sdk/agenta_init.py,sha256=2iU8POiedmWL34_DgeXk38n0kN3MWlrGOruRsvd66Yw,8843
+agenta/sdk/agenta_init.py,sha256=CGttzScnB64IgyUQ-atz9RgWyWcAiKx5FXfowVuRsmg,8827
 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=aW78CMpAVQ5YWQuDAgfjlqyXY5LHUmS4LQukJqhxNB0,19531
+agenta/sdk/decorators/llm_entrypoint.py,sha256=rdtehr_OmLz-fIZqHWfKxir6GhwkDeI1Of3BcQlkzpc,19526
 agenta/sdk/decorators/tracing.py,sha256=-QrchKPjFjYbIau5bw1hlUyOcX4kdOzjVQ79KE4iwmw,2800
 agenta/sdk/router.py,sha256=0sbajvn5C7t18anH6yNo7-oYxldHnYfwcbmQnIXBePw,269
 agenta/sdk/tracing/context_manager.py,sha256=HskDaiORoOhjeN375gm05wYnieQzh5UnoIsnSAHkAyc,252
-agenta/sdk/tracing/llm_tracing.py,sha256=RxMVO5xynk3FWpl5CFAQFEohUlhb1WoUqSze1BgtL-4,8140
+agenta/sdk/tracing/llm_tracing.py,sha256=JbrcHBwmHA_7EVgWGM3fatz4jYeKJJDXyu8qH3Q2LUE,8186
 agenta/sdk/tracing/logger.py,sha256=4zG9c51p8xPdKA5SL8MOgBfkpCnBSuV6JfWiXO0A7oc,473
 agenta/sdk/tracing/tasks_manager.py,sha256=ROrWIaqS2J2HHiJtRWiHKlLY8CCsqToP5VeXu7mamck,3748
 agenta/sdk/types.py,sha256=Mn0yBlHh_Yr_5oQXUfsYI3V7sJAVWkJgkxEOBDOOMS0,5852
@@ -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.15.0a1.dist-info/METADATA,sha256=MVIjIlEt-YrAzqRdO5mg7JzGx_R9BD-60QkNmiK39Rw,26465
-agenta-0.15.0a1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-agenta-0.15.0a1.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
-agenta-0.15.0a1.dist-info/RECORD,,
+agenta-0.15.0a3.dist-info/METADATA,sha256=f8bhz2NNW-TAfzwpy9mEMSPYA9UX2M3Sm9D43woToM4,26465
+agenta-0.15.0a3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+agenta-0.15.0a3.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
+agenta-0.15.0a3.dist-info/RECORD,,