agenta 0.27.7a2__py3-none-any.whl → 0.27.8a2__py3-none-any.whl

agenta/client/backend/evaluations/client.py

@@ -26,7 +26,6 @@ class EvaluationsClient:
     def fetch_evaluation_ids(
         self,
         *,
-        app_id: str,
         resource_type: str,
         resource_ids: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
         request_options: typing.Optional[RequestOptions] = None,
@@ -35,7 +34,6 @@ class EvaluationsClient:
         Fetches evaluation ids for a given resource type and id.
 
         Arguments:
-        app_id (str): The ID of the app for which to fetch evaluations.
         resource_type (str): The type of resource for which to fetch evaluations.
         resource_ids List[ObjectId]: The IDs of resource for which to fetch evaluations.
 
@@ -47,8 +45,6 @@ class EvaluationsClient:
 
         Parameters
         ----------
-        app_id : str
-
         resource_type : str
 
         resource_ids : typing.Optional[typing.Union[str, typing.Sequence[str]]]
@@ -70,7 +66,6 @@ class EvaluationsClient:
             base_url="https://yourhost.com/path/to/api",
         )
         client.evaluations.fetch_evaluation_ids(
-            app_id="app_id",
             resource_type="resource_type",
         )
         """
@@ -78,7 +73,6 @@ class EvaluationsClient:
             "evaluations/by_resource",
             method="GET",
             params={
-                "app_id": app_id,
                 "resource_type": resource_type,
                 "resource_ids": resource_ids,
             },
@@ -714,7 +708,6 @@ class AsyncEvaluationsClient:
     async def fetch_evaluation_ids(
         self,
         *,
-        app_id: str,
         resource_type: str,
         resource_ids: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
         request_options: typing.Optional[RequestOptions] = None,
@@ -723,7 +716,6 @@ class AsyncEvaluationsClient:
         Fetches evaluation ids for a given resource type and id.
 
         Arguments:
-        app_id (str): The ID of the app for which to fetch evaluations.
         resource_type (str): The type of resource for which to fetch evaluations.
         resource_ids List[ObjectId]: The IDs of resource for which to fetch evaluations.
 
@@ -735,7 +727,6 @@ class AsyncEvaluationsClient:
 
         Parameters
         ----------
-        app_id : str
 
         resource_type : str
 
@@ -763,7 +754,6 @@ class AsyncEvaluationsClient:
 
         async def main() -> None:
             await client.evaluations.fetch_evaluation_ids(
-                app_id="app_id",
                 resource_type="resource_type",
             )
 
@@ -774,7 +764,6 @@ class AsyncEvaluationsClient:
             "evaluations/by_resource",
             method="GET",
             params={
-                "app_id": app_id,
                 "resource_type": resource_type,
                 "resource_ids": resource_ids,
             },

agenta/client/backend/testsets/client.py

@@ -28,7 +28,6 @@ class TestsetsClient:
         file: core.File,
         upload_type: typing.Optional[str] = OMIT,
         testset_name: typing.Optional[str] = OMIT,
-        app_id: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> TestSetSimpleResponse:
         """
@@ -51,8 +50,6 @@ class TestsetsClient:
 
         testset_name : typing.Optional[str]
 
-        app_id : typing.Optional[str]
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -77,7 +74,6 @@ class TestsetsClient:
             data={
                 "upload_type": upload_type,
                 "testset_name": testset_name,
-                "app_id": app_id,
             },
             files={
                 "file": file,
@@ -173,18 +169,16 @@ class TestsetsClient:
 
     def create_testset(
         self,
-        app_id: str,
         *,
         name: str,
         csvdata: typing.Sequence[typing.Dict[str, typing.Optional[typing.Any]]],
         request_options: typing.Optional[RequestOptions] = None,
     ) -> TestSetSimpleResponse:
         """
-        Create a testset with given name and app_name, save the testset to MongoDB.
+        Create a testset with given name, save the testset to MongoDB.
 
         Args:
         name (str): name of the test set.
-        app_name (str): name of the application.
         testset (Dict[str, str]): test set data.
 
         Returns:
@@ -192,8 +186,6 @@ class TestsetsClient:
 
         Parameters
         ----------
-        app_id : str
-
         name : str
 
         csvdata : typing.Sequence[typing.Dict[str, typing.Optional[typing.Any]]]
@@ -215,13 +207,12 @@ class TestsetsClient:
             base_url="https://yourhost.com/path/to/api",
         )
         client.testsets.create_testset(
-            app_id="app_id",
             name="name",
             csvdata=[{"key": "value"}],
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"testsets/{jsonable_encoder(app_id)}",
+            "testsets",
             method="POST",
             json={
                 "name": name,
@@ -405,7 +396,7 @@ class TestsetsClient:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
     def get_testsets(
-        self, *, app_id: str, request_options: typing.Optional[RequestOptions] = None
+        self, *, request_options: typing.Optional[RequestOptions] = None
     ) -> typing.List[TestSetOutputResponse]:
         """
         Get all testsets.
@@ -420,8 +411,6 @@ class TestsetsClient:
 
         Parameters
         ----------
-        app_id : str
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -438,16 +427,11 @@ class TestsetsClient:
             api_key="YOUR_API_KEY",
             base_url="https://yourhost.com/path/to/api",
         )
-        client.testsets.get_testsets(
-            app_id="app_id",
-        )
+        client.testsets.get_testsets()
         """
         _response = self._client_wrapper.httpx_client.request(
             "testsets",
             method="GET",
-            params={
-                "app_id": app_id,
-            },
             request_options=request_options,
         )
         try:
@@ -557,7 +541,6 @@ class AsyncTestsetsClient:
         file: core.File,
         upload_type: typing.Optional[str] = OMIT,
         testset_name: typing.Optional[str] = OMIT,
-        app_id: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> TestSetSimpleResponse:
         """
@@ -580,8 +563,6 @@ class AsyncTestsetsClient:
 
         testset_name : typing.Optional[str]
 
-        app_id : typing.Optional[str]
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -614,7 +595,6 @@ class AsyncTestsetsClient:
             data={
                 "upload_type": upload_type,
                 "testset_name": testset_name,
-                "app_id": app_id,
             },
             files={
                 "file": file,
@@ -718,18 +698,16 @@ class AsyncTestsetsClient:
 
     async def create_testset(
         self,
-        app_id: str,
         *,
         name: str,
         csvdata: typing.Sequence[typing.Dict[str, typing.Optional[typing.Any]]],
         request_options: typing.Optional[RequestOptions] = None,
     ) -> TestSetSimpleResponse:
         """
-        Create a testset with given name and app_name, save the testset to MongoDB.
+        Create a testset with given name, save the testset to MongoDB.
 
         Args:
         name (str): name of the test set.
-        app_name (str): name of the application.
         testset (Dict[str, str]): test set data.
 
         Returns:
@@ -737,8 +715,6 @@ class AsyncTestsetsClient:
 
         Parameters
         ----------
-        app_id : str
-
         name : str
 
         csvdata : typing.Sequence[typing.Dict[str, typing.Optional[typing.Any]]]
@@ -765,7 +741,6 @@ class AsyncTestsetsClient:
 
         async def main() -> None:
             await client.testsets.create_testset(
-                app_id="app_id",
                 name="name",
                 csvdata=[{"key": "value"}],
             )
@@ -774,7 +749,7 @@ class AsyncTestsetsClient:
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"testsets/{jsonable_encoder(app_id)}",
+            "testsets",
             method="POST",
             json={
                 "name": name,
@@ -974,7 +949,7 @@ class AsyncTestsetsClient:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
     async def get_testsets(
-        self, *, app_id: str, request_options: typing.Optional[RequestOptions] = None
+        self, *, request_options: typing.Optional[RequestOptions] = None
     ) -> typing.List[TestSetOutputResponse]:
         """
         Get all testsets.
@@ -989,8 +964,6 @@ class AsyncTestsetsClient:
 
         Parameters
         ----------
-        app_id : str
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -1012,9 +985,7 @@ class AsyncTestsetsClient:
 
 
         async def main() -> None:
-            await client.testsets.get_testsets(
-                app_id="app_id",
-            )
+            await client.testsets.get_testsets()
 
 
         asyncio.run(main())
@@ -1022,9 +993,6 @@ class AsyncTestsetsClient:
         _response = await self._client_wrapper.httpx_client.request(
             "testsets",
             method="GET",
-            params={
-                "app_id": app_id,
-            },
             request_options=request_options,
         )
         try:

agenta/sdk/agenta_init.py

@@ -1,4 +1,3 @@
-import logging
 import toml
 from os import getenv
 from typing import Optional, Callable, Any
@@ -11,10 +10,6 @@ from agenta.sdk.tracing import Tracing
 from agenta.client.exceptions import APIRequestError
 
 
-logger = logging.getLogger(__name__)
-logger.setLevel(logging.DEBUG)
-
-
 class AgentaSingleton:
     """Singleton class to save all the "global variables" for the sdk."""
 
@@ -164,7 +159,7 @@ class Config:
         try:
             self.push(config_name="default", overwrite=overwrite, **kwargs)
         except Exception as ex:
-            logger.warning(
+            log.warning(
                 "Unable to push the default configuration to the server. %s", str(ex)
             )
 
@@ -185,7 +180,7 @@ class Config:
                 overwrite=overwrite,
             )
         except Exception as ex:
-            logger.warning(
+            log.warning(
                 "Failed to push the configuration to the server with error: %s", ex
             )
 
@@ -212,14 +207,14 @@ class Config:
                         config_name=config_name,
                     )
             except Exception as ex:
-                logger.warning(
+                log.warning(
                     "Failed to pull the configuration from the server with error: %s",
                     str(ex),
                 )
         try:
             self.set(**{"current_version": config.current_version, **config.parameters})
         except Exception as ex:
-            logger.warning("Failed to set the configuration with error: %s", str(ex))
+            log.warning("Failed to set the configuration with error: %s", str(ex))
 
     def all(self):
         """Returns all the parameters for the app variant"""

agenta/sdk/decorators/routing.py

@@ -143,9 +143,7 @@ class entrypoint:
                 _MIDDLEWARES = False
 
         except:  # pylint: disable=bare-except
-            log.error("------------------------------------")
-            log.error("Agenta SDK - failed to secure route: %s", route_path)
-            log.error("------------------------------------")
+            log.warning("Agenta SDK - failed to secure route: %s", route_path)
         ### --- Update Middleware --- #
 
         DEFAULT_PATH = "generate"
@@ -357,9 +355,7 @@ class entrypoint:
         *args,
         **func_params,
     ):
-        log.info("---------------------------")
-        log.info(f"Agenta SDK - running route: {repr(self.route_path or '/')}")
-        log.info("---------------------------")
+        log.info("Agenta SDK - handling route: %s", repr(self.route_path or "/"))
 
         tracing_context.set(routing_context.get())
 
@@ -377,7 +373,7 @@ class entrypoint:
 
     async def handle_success(self, result: Any, inline_trace: bool):
         data = None
-        trace = dict()
+        trace = None
 
         with suppress():
             data = self.patch_result(result)
@@ -392,21 +388,17 @@ class entrypoint:
         return BaseResponse(data=data, tree=trace)
 
     def handle_failure(self, error: Exception):
-        log.error("--------------------------------------------------")
-        log.error("Agenta SDK - handling application exception below:")
-        log.error("--------------------------------------------------")
-        log.error(format_exc().strip("\n"))
-        log.error("--------------------------------------------------")
+        log.warning("--------------------------------------------------")
+        log.warning("Agenta SDK - handling application exception below:")
+        log.warning("--------------------------------------------------")
+        log.warning(format_exc().strip("\n"))
+        log.warning("--------------------------------------------------")
 
         status_code = error.status_code if hasattr(error, "status_code") else 500
         message = str(error)
         stacktrace = format_exception(error, value=error, tb=error.__traceback__)  # type: ignore
         detail = {"message": message, "stacktrace": stacktrace}
 
-        log.error(f"----------------------------------")
-        log.error(f"Agenta SDK - exiting with failure: {status_code}")
-        log.error(f"----------------------------------")
-
         raise HTTPException(status_code=status_code, detail=detail)
 
     def patch_result(self, result: Any):
@@ -683,10 +675,7 @@ class entrypoint:
 
         loop = get_event_loop()
 
-        with routing_context_manager(
-            config=args_config_params,
-            environment="terminal",
-        ):
+        with routing_context_manager(config=args_config_params):
             result = loop.run_until_complete(
                 self.execute_function(
                     func,
@@ -695,30 +684,23 @@ class entrypoint:
                 )
             )
 
-        SHOW_DETAILS = True
-        SHOW_DATA = False
-        SHOW_TRACE = False
-
         if result.trace:
             log.info("\n========= Result =========\n")
 
             log.info(f"trace_id: {result.trace['trace_id']}")
-            if SHOW_DETAILS:
-                log.info(f"latency:  {result.trace.get('latency')}")
-                log.info(f"cost:     {result.trace.get('cost')}")
-                log.info(f"usage:   {list(result.trace.get('usage', {}).values())}")
-
-            if SHOW_DATA:
-                log.info(" ")
-                log.info(f"data:")
-                log.info(dumps(result.data, indent=2))
-
-            if SHOW_TRACE:
-                log.info(" ")
-                log.info(f"trace:")
-                log.info(f"----------------")
-                log.info(dumps(result.trace.get("spans", []), indent=2))
-                log.info(f"----------------")
+            log.info(f"latency:  {result.trace.get('latency')}")
+            log.info(f"cost:     {result.trace.get('cost')}")
+            log.info(f"usage:   {list(result.trace.get('usage', {}).values())}")
+
+            log.info(" ")
+            log.info("data:")
+            log.info(dumps(result.data, indent=2))
+
+            log.info(" ")
+            log.info("trace:")
+            log.info("----------------")
+            log.info(dumps(result.trace.get("spans", []), indent=2))
+            log.info("----------------")
 
             log.info("\n==========================\n")
 

agenta/sdk/litellm/litellm.py

@@ -60,7 +60,11 @@ def litellm_handler():
             )
 
             if not self.span:
-                log.error("LiteLLM callback error: span not found.")
+                log.warning("Agenta SDK - litellm tracing failed")
+                return
+
+            if not self.span.is_recording():
+                log.error("Agenta SDK - litellm span not recording.")
                 return
 
             self.span.set_attributes(
@@ -86,43 +90,11 @@ def litellm_handler():
             end_time,
         ):
             if not self.span:
-                log.error("LiteLLM callback error: span not found.")
+                log.warning("Agenta SDK - litellm tracing failed")
                 return
 
-            try:
-                result = []
-                for choice in response_obj.choices:
-                    message = choice.message.__dict__
-                    result.append(message)
-
-                outputs = {"completion": result}
-                self.span.set_attributes(
-                    attributes={"outputs": outputs},
-                    namespace="data",
-                )
-
-            except Exception as e:
-                pass
-
-            self.span.set_attributes(
-                attributes={"total": kwargs.get("response_cost")},
-                namespace="metrics.unit.costs",
-            )
-
-            self.span.set_attributes(
-                attributes=(
-                    {
-                        "prompt": response_obj.usage.prompt_tokens,
-                        "completion": response_obj.usage.completion_tokens,
-                        "total": response_obj.usage.total_tokens,
-                    }
-                ),
-                namespace="metrics.unit.tokens",
-            )
-
-            self.span.set_status(status="OK")
-
-            self.span.end()
+            if not self.span.is_recording():
+                return
 
         def log_success_event(
             self,
@@ -131,8 +103,14 @@ def litellm_handler():
             start_time,
             end_time,
         ):
+            if kwargs.get("stream"):
+                return
+
             if not self.span:
-                log.error("LiteLLM callback error: span not found.")
+                log.warning("Agenta SDK - litellm tracing failed")
+                return
+
+            if not self.span.is_recording():
                 return
 
             try:
@@ -178,7 +156,10 @@ def litellm_handler():
             end_time,
         ):
             if not self.span:
-                log.error("LiteLLM callback error: span not found.")
+                log.warning("Agenta SDK - litellm tracing failed")
+                return
+
+            if not self.span.is_recording():
                 return
 
             self.span.record_exception(kwargs["exception"])
@@ -195,43 +176,11 @@ def litellm_handler():
             end_time,
         ):
             if not self.span:
-                log.error("LiteLLM callback error: span not found.")
+                log.warning("Agenta SDK - litellm tracing failed")
                 return
 
-            try:
-                result = []
-                for choice in response_obj.choices:
-                    message = choice.message.__dict__
-                    result.append(message)
-
-                outputs = {"completion": result}
-                self.span.set_attributes(
-                    attributes={"outputs": outputs},
-                    namespace="data",
-                )
-
-            except Exception as e:
-                pass
-
-            self.span.set_attributes(
-                attributes={"total": kwargs.get("response_cost")},
-                namespace="metrics.unit.costs",
-            )
-
-            self.span.set_attributes(
-                attributes=(
-                    {
-                        "prompt": response_obj.usage.prompt_tokens,
-                        "completion": response_obj.usage.completion_tokens,
-                        "total": response_obj.usage.total_tokens,
-                    }
-                ),
-                namespace="metrics.unit.tokens",
-            )
-
-            self.span.set_status(status="OK")
-
-            self.span.end()
+            if not self.span.is_recording():
+                return
 
         async def async_log_success_event(
             self,
@@ -241,7 +190,10 @@ def litellm_handler():
             end_time,
         ):
             if not self.span:
-                log.error("LiteLLM callback error: span not found.")
+                log.warning("Agenta SDK - litellm tracing failed")
+                return
+
+            if not self.span.is_recording():
                 return
 
             try:
@@ -287,7 +239,10 @@ def litellm_handler():
             end_time,
         ):
             if not self.span:
-                log.error("LiteLLM callback error: span not found.")
+                log.warning("Agenta SDK - litellm tracing failed")
+                return
+
+            if not self.span.is_recording():
                 return
 
             self.span.record_exception(kwargs["exception"])

agenta/sdk/middleware/auth.py

@@ -133,10 +133,10 @@ class AuthorizationMiddleware(BaseHTTPMiddleware):
             return await call_next(request)
 
         except:  # pylint: disable=bare-except
-            log.error("------------------------------------------------------")
-            log.error("Agenta SDK - handling auth middleware exception below:")
-            log.error("------------------------------------------------------")
-            log.error(format_exc().strip("\n"))
-            log.error("------------------------------------------------------")
+            log.warning("------------------------------------------------------")
+            log.warning("Agenta SDK - handling auth middleware exception below:")
+            log.warning("------------------------------------------------------")
+            log.warning(format_exc().strip("\n"))
+            log.warning("------------------------------------------------------")
 
             return Deny()

agenta/sdk/tracing/context.py

@@ -14,11 +14,11 @@ def tracing_context_manager():
     token = tracing_context.set(_tracing_context)
     try:
         yield
-    except Exception as e:
-        log.error("----------------------------------------------")
-        log.error("Agenta SDK - handling tracing exception below:")
-        log.error("----------------------------------------------")
-        log.error(format_exc().strip("\n"))
-        log.error("----------------------------------------------")
+    except:  # pylint: disable=bare-except
+        log.warning("----------------------------------------------")
+        log.warning("Agenta SDK - handling tracing exception below:")
+        log.warning("----------------------------------------------")
+        log.warning(format_exc().strip("\n"))
+        log.warning("----------------------------------------------")
     finally:
         tracing_context.reset(token)

agenta/sdk/tracing/inline.py

@@ -1118,13 +1118,15 @@ def calculate_costs(span_idx: Dict[str, SpanDTO]):
             and span.meta
             and span.metrics
         ):
+            model = span.meta.get("response.model")
+            prompt_tokens = span.metrics.get("unit.tokens.prompt", 0.0)
+            completion_tokens = span.metrics.get("unit.tokens.completion", 0.0)
+
             try:
                 costs = cost_calculator.cost_per_token(
-                    model=span.meta.get("response.model"),
-                    prompt_tokens=span.metrics.get("unit.tokens.prompt", 0.0),
-                    completion_tokens=span.metrics.get("unit.tokens.completion", 0.0),
-                    call_type=span.node.type.name.lower(),
-                    response_time_ms=span.time.span // 1_000,
+                    model=model,
+                    prompt_tokens=prompt_tokens,
+                    completion_tokens=completion_tokens,
                 )
 
                 if not costs:
@@ -1137,5 +1139,8 @@ def calculate_costs(span_idx: Dict[str, SpanDTO]):
                 span.metrics["unit.costs.completion"] = completion_cost
                 span.metrics["unit.costs.total"] = total_cost
 
-            except:
-                pass
+            except:  # pylint: disable=bare-except
+                print("Failed to calculate costs:")
+                print(
+                    f"model={model}, prompt_tokens={prompt_tokens}, completion_tokens={completion_tokens}"
+                )

agenta/sdk/tracing/processors.py

@@ -91,9 +91,7 @@ class TraceProcessor(BatchSpanProcessor):
         ret = super().force_flush(timeout_millis)
 
         if not ret:
-            log.error("--------------------------------------------")
-            log.error("Agenta SDK - skipping export due to timeout.")
-            log.error("--------------------------------------------")
+            log.warning("Agenta SDK - skipping export due to timeout.")
 
     def is_ready(
         self,

agenta/sdk/tracing/tracing.py

@@ -90,7 +90,6 @@ class Tracing(metaclass=Singleton):
                 self.otlp_url,
             )
             log.info("--------------------------------------------")
-
             check(
                 self.otlp_url,
                 headers=self.headers,
@@ -106,13 +105,10 @@ class Tracing(metaclass=Singleton):
             )
 
             self.tracer_provider.add_span_processor(_otlp)
-
             log.info("Success: traces will be exported.")
             log.info("--------------------------------------------")
-
         except:  # pylint: disable=bare-except
-            log.warning("Failure: traces will not be exported.")
-            log.warning("--------------------------------------------")
+            log.warning("Agenta SDK - traces will not be exported.")
 
         # GLOBAL TRACER PROVIDER -- INSTRUMENTATION LIBRARIES
         set_tracer_provider(self.tracer_provider)

agenta/sdk/utils/exceptions.py

@@ -17,11 +17,11 @@ class suppress(AbstractContextManager):  # pylint: disable=invalid-name
         if exc_type is None:
             return True
         else:
-            log.error("-------------------------------------------------")
-            log.error("Agenta SDK - suppressing tracing exception below:")
-            log.error("-------------------------------------------------")
-            log.error(format_exc().strip("\n"))
-            log.error("-------------------------------------------------")
+            log.warning("-------------------------------------------------")
+            log.warning("Agenta SDK - suppressing tracing exception below:")
+            log.warning("-------------------------------------------------")
+            log.warning(format_exc().strip("\n"))
+            log.warning("-------------------------------------------------")
             return True
 
 
@@ -34,8 +34,11 @@ def handle_exceptions():
             try:
                 return await func(*args, **kwargs)
             except Exception as e:
-                log.error("--- HANDLING EXCEPTION ---")
-                log.error("--------------------------")
+                log.warning("------------------------------------------")
+                log.warning("Agenta SDK - intercepting exception below:")
+                log.warning("------------------------------------------")
+                log.warning(format_exc().strip("\n"))
+                log.warning("------------------------------------------")
                 raise e
 
         @wraps(func)
@@ -43,8 +46,11 @@ def handle_exceptions():
             try:
                 return func(*args, **kwargs)
             except Exception as e:
-                log.error("--- HANDLING EXCEPTION ---")
-                log.error("--------------------------")
+                log.warning("------------------------------------------")
+                log.warning("Agenta SDK - intercepting exception below:")
+                log.warning("------------------------------------------")
+                log.warning(format_exc().strip("\n"))
+                log.warning("------------------------------------------")
                 raise e
 
         return async_wrapper if is_coroutine_function else sync_wrapper

agenta/sdk/utils/logging.py

@@ -1,8 +1,12 @@
 import logging
+from os import getenv
 
 
 class Logger:
-    def __init__(self, name="agenta.logger", level=logging.DEBUG):
+    def __init__(self, name="agenta.logger", level=logging.WARNING):
+        if getenv("AGENTA_DEBUG"):
+            level = logging.DEBUG
+
         self.logger = logging.getLogger(name)
         self.logger.setLevel(level)
 

{agenta-0.27.7a2.dist-info → agenta-0.27.8a2.dist-info}/METADATA

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

{agenta-0.27.7a2.dist-info → agenta-0.27.8a2.dist-info}/RECORD

@@ -38,7 +38,7 @@ agenta/client/backend/environments/client.py,sha256=JG980MafNEcMX60gCtLdPQb8Ja6R
 agenta/client/backend/errors/__init__.py,sha256=pbbVUFtB9LCocA1RMWMMF_RKjsy5YkOKX5BAuE49w6g,170
 agenta/client/backend/errors/unprocessable_entity_error.py,sha256=FvR7XPlV3Xx5nu8HNlmLhBRdk4so_gCHjYT5PyZe6sM,313
 agenta/client/backend/evaluations/__init__.py,sha256=9mUnTDeA1TxYvkj1l01A1prqsJV0ERRY2tzkY1fA4MQ,64
-agenta/client/backend/evaluations/client.py,sha256=T7sDKXeuiihQqxW_TIGrcSShCxduZ2QqjK80nTbTg0w,46698
+agenta/client/backend/evaluations/client.py,sha256=GgBq5iXFR89VMt6SSbSpk6zDseR79brzVtpfJM1XxxA,46339
 agenta/client/backend/evaluators/__init__.py,sha256=9mUnTDeA1TxYvkj1l01A1prqsJV0ERRY2tzkY1fA4MQ,64
 agenta/client/backend/evaluators/client.py,sha256=fdfb0ZFLP-kI5jXr6F3O-wKWvWBGUXhqRXIoetyN-zQ,40405
 agenta/client/backend/observability/__init__.py,sha256=9mUnTDeA1TxYvkj1l01A1prqsJV0ERRY2tzkY1fA4MQ,64
@@ -49,7 +49,7 @@ agenta/client/backend/observability_v_1/types/__init__.py,sha256=vHpFg4KBBFowe8E
 agenta/client/backend/observability_v_1/types/format.py,sha256=U-b6HwkOirqvyU7WWDfXyDkYlxBV90lRe1YKds4qg_Q,157
 agenta/client/backend/observability_v_1/types/query_traces_response.py,sha256=0-4nNfKLnciOR7E1859HupWqcwunJe53OLgXRlxxeb8,452
 agenta/client/backend/testsets/__init__.py,sha256=9mUnTDeA1TxYvkj1l01A1prqsJV0ERRY2tzkY1fA4MQ,64
-agenta/client/backend/testsets/client.py,sha256=awjS8DjX08FnEgr6LtMHroK9ysMkLqKAZHh2_1-rldE,35201
+agenta/client/backend/testsets/client.py,sha256=nmH7xcA4ByXTNHr8m24HbYGKmCdHrFOzlXRGV2QO7L0,34343
 agenta/client/backend/types/__init__.py,sha256=fjXpRS1ZZAJPTRYybJcuAMJ9JnkmmvDbghY3GGLYm34,7521
 agenta/client/backend/types/agenta_node_dto.py,sha256=DLZltD_ueVOfdiSZ1u2i3mu1C3kshxhcggQFFfybBdY,1747
 agenta/client/backend/types/agenta_node_dto_nodes_value.py,sha256=ifG7dBYLphFoCgQ70ivvDXkTSlxX8w5x_9S-Glh8FlI,180
@@ -174,42 +174,41 @@ 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=kO1q2_IR0fEAo4M-2Pt_v-zC7GxxnkLogjKFhU869Ps,3555
 agenta/sdk/__init__.py,sha256=Rz5ZqSbz5wIl-v4Jf4kkaTy6ofCYg5CcybyfgeAgaP8,1948
-agenta/sdk/agenta_init.py,sha256=YjJjwBUci77KTP27lZTPJMVvjmskng4CqbfQVbcMBJQ,10967
+agenta/sdk/agenta_init.py,sha256=vH7FOY7XshKqHfAV7P54Owb-7E0cuxuMMCRobd1PT1U,10870
 agenta/sdk/assets.py,sha256=Zv4i8MVUSB3jMODQon1mzJtYxuntmrCNjLGk8f-2fls,2856
 agenta/sdk/client.py,sha256=trKyBOYFZRk0v5Eptxvh87yPf50Y9CqY6Qgv4Fy-VH4,2142
 agenta/sdk/context/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agenta/sdk/context/routing.py,sha256=ycUgmJZyWhL4bHjKtUSAsTlt_0Fujr_6OpoaEH1lAN0,683
 agenta/sdk/context/tracing.py,sha256=UmmW15UFFsvxS0myS6aD9wBk5iNepNlQi4tEQ_ejfYM,96
 agenta/sdk/decorators/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-agenta/sdk/decorators/routing.py,sha256=PAZ0fw8sQf5PQ_QYBWCxpqnGN9mjaydADw5MjqMchfI,37244
+agenta/sdk/decorators/routing.py,sha256=zOkjs9TH7jF0DCvCjV-F9uzUPRDiHRakgS_vkpXBch4,36579
 agenta/sdk/decorators/tracing.py,sha256=vL5e6TVX6TQwO0t9raZwnzXHV3vElVT0pHS1vD-vzEo,8523
 agenta/sdk/litellm/__init__.py,sha256=Bpz1gfHQc0MN1yolWcjifLWznv6GjHggvRGQSpxpihM,37
-agenta/sdk/litellm/litellm.py,sha256=bzVm-388BleZ2QH_JWpfZ2fB_7-HuoSqdvLo4EDPcEM,8788
+agenta/sdk/litellm/litellm.py,sha256=97eDU9MUTW6Wv2e7QYhCWLrRYwhaRqG6LVcedp3qNrk,7186
 agenta/sdk/managers/__init__.py,sha256=SN-LRwG0pRRDV3u2Q4JiiSTigN3-mYpzGNM35RzT4mc,238
 agenta/sdk/managers/config.py,sha256=AuFfHYOkmilDdcAJt2nlw_WlA3ho4Htjf29FKTZGElM,11716
 agenta/sdk/managers/deployment.py,sha256=SEokjZeh6n7HRKZ92Y0WncdG49hIFx-Z3B3HAl2kmUg,1174
 agenta/sdk/managers/shared.py,sha256=e53jckQq5PIMpjdxADOonUj7o8aGfzmSvdeH5f43rGs,21497
 agenta/sdk/managers/variant.py,sha256=A5ga3mq3b0weUTXa9HO72MGaspthGcu1uK9K5OnP738,4172
 agenta/sdk/middleware/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-agenta/sdk/middleware/auth.py,sha256=EGoj45mLe2YpHbNSX6nCzxsIC2-B-S3CdlRF3DBju3I,4093
+agenta/sdk/middleware/auth.py,sha256=H_UwEq5QioSkHZ3ipCtE3lrl7mxs5h9ULQvZHEtV8d4,4103
 agenta/sdk/middleware/cache.py,sha256=C_LEzFbulbCBIKtcut2T4qJzh90q4369WCsApDg3Vm0,902
 agenta/sdk/router.py,sha256=mOguvtOwl2wmyAgOuWTsf98pQwpNiUILKIo67W_hR3A,119
 agenta/sdk/tracing/__init__.py,sha256=rQNe5-zT5Kt7_CDhq-lnUIi1EYTBVzVf_MbfcIxVD98,41
 agenta/sdk/tracing/attributes.py,sha256=zh8JQZSeYCLBeIRSopKJx6QQ-WEgw08Cr64DS_WOcT8,3833
-agenta/sdk/tracing/context.py,sha256=PSJdhcaOXSMAuGUBySpLKPKyx8duF3TJzhUEk2ufqPc,777
+agenta/sdk/tracing/context.py,sha256=IpNENDGRrXWjs-vti5XheqwybQs0QdD-ii4aK0enNrM,803
 agenta/sdk/tracing/conventions.py,sha256=JBtznBXZ3aRkGKkLl7cPwdMNh3w1G-H2Ta2YrAxbr38,950
 agenta/sdk/tracing/exporters.py,sha256=YvTke0RaxeOLqWOuhC5EFzYwFY39kcoBtDLfcyla3j4,1604
-agenta/sdk/tracing/inline.py,sha256=M4apMWajHQBgTQFb8ZF_wgqwboLnAOUiZnHPVL3ixFA,31197
-agenta/sdk/tracing/processors.py,sha256=tjoz_uXm_2yD1Ozvz_CnYnsG82R6vmnYpjMhx2ztALs,3117
+agenta/sdk/tracing/inline.py,sha256=ncIvMKWpy1o-5l3i7YThj_pE3Oq6xvQfwyUfGsfUxzc,31405
+agenta/sdk/tracing/processors.py,sha256=tPf3Hx1TmnWSljvX28kYFOFrp4ImIXhlwAaR39sf5qU,2979
 agenta/sdk/tracing/spans.py,sha256=nqUOjjirBxB8Eacv8Qj4Ra_6rknGi3lbJdNyKmk5ODQ,3707
-agenta/sdk/tracing/tracing.py,sha256=HYuGp5fQozLKWXLXonfCT8pqb7x1gwzzbZk87mfhu3Y,6983
+agenta/sdk/tracing/tracing.py,sha256=RUSnQIWwfOFwhWbiyJfgC0wYtvcV1PusXTwNslV5WZo,6912
 agenta/sdk/types.py,sha256=_lGsGSEaZJrUT4cVcT3zSpgEqex2jFaPtfpFeUEetbc,7247
 agenta/sdk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agenta/sdk/utils/costs.py,sha256=i8C7ud__pThLS55XkN4YW8czXtGeXr2mx7jjcOFeiXg,5955
-agenta/sdk/utils/debug.py,sha256=DxiCAeXxxrcEZT2CjlNA6BMvujGP4nzQ-rfb-_mLMck,2114
-agenta/sdk/utils/exceptions.py,sha256=AU0csJtuzswjkr8O8K-PmuOEr0vRNymKPZVRnRR_0Ts,1646
+agenta/sdk/utils/exceptions.py,sha256=UPO5aqNkpxevE7I_gW6O2B0FbtqqwgQobp5b6RQy_Ik,2132
 agenta/sdk/utils/globals.py,sha256=2HhyzWn55BbYNCZ3rT8dAxk1GGXuGQPbtq_THjaHbBw,372
-agenta/sdk/utils/logging.py,sha256=FKzAO5eHjR_qvpEnc4tKXUA6WftmwdwYwuJK6CbSc84,379
+agenta/sdk/utils/logging.py,sha256=eFzEFuYpggfIhEKv09JZRqcDzkmZ482a_E2G-X0FK7Y,473
 agenta/sdk/utils/preinit.py,sha256=YlJL7RLfel0R7DFp-jK7OV-z4ZIQJM0oupYlk7g8b5o,1278
 agenta/sdk/utils/singleton.py,sha256=17Ph7LGnnV8HkPjImruKita2ni03Ari5jr0jqm__4sc,312
 agenta/templates/compose_email/README.md,sha256=ss7vZPpI1Hg0VmYtFliwq_r5LnqbCy_S5OQDXg8UoIA,308
@@ -227,7 +226,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.27.7a2.dist-info/METADATA,sha256=f2CT_p3gHpLMIdNZVYH3GSWEPU2x1wf_ocqohombbOw,31619
-agenta-0.27.7a2.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-agenta-0.27.7a2.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
-agenta-0.27.7a2.dist-info/RECORD,,
+agenta-0.27.8a2.dist-info/METADATA,sha256=fwGIpk-muffeEzl-HQyz1O3m6_bpaa0OvRR8FcmItVU,31619
+agenta-0.27.8a2.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+agenta-0.27.8a2.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
+agenta-0.27.8a2.dist-info/RECORD,,

agenta/sdk/utils/debug.py

@@ -1,68 +0,0 @@
-from inspect import iscoroutinefunction
-from functools import wraps
-
-from agenta.sdk.utils.logging import log
-
-DEBUG = False
-SHIFT = 7
-
-
-def debug(shift=1, req=False, res=False, chars=[">", "<"]):
-    def log_decorator(f):
-        is_async = iscoroutinefunction(f)
-
-        @wraps(f)
-        async def async_log_wrapper(*args, **kwargs):
-            if DEBUG:
-                log.debug(
-                    " ".join(
-                        [
-                            chars[0] * shift + " " * (SHIFT - shift),
-                            f.__name__ + " ()",
-                            str(args) if req else "",
-                            str(kwargs) if req else "",
-                        ]
-                    )
-                )
-            result = await f(*args, **kwargs)
-            if DEBUG:
-                log.debug(
-                    " ".join(
-                        [
-                            chars[1] * shift + " " * (SHIFT - shift),
-                            f.__name__ + " <-",
-                            str(result) if res else "",
-                        ]
-                    )
-                )
-            return result
-
-        @wraps(f)
-        def log_wrapper(*args, **kwargs):
-            if DEBUG:
-                log.debug(
-                    " ".join(
-                        [
-                            chars[0] * shift + " " * (SHIFT - shift),
-                            f.__name__ + " ()",
-                            str(args) if req else "",
-                            str(kwargs) if req else "",
-                        ]
-                    )
-                )
-            result = f(*args, **kwargs)
-            if DEBUG:
-                log.debug(
-                    " ".join(
-                        [
-                            chars[1] * shift + " " * (SHIFT - shift),
-                            f.__name__ + " <-",
-                            str(result) if res else "",
-                        ]
-                    )
-                )
-            return result
-
-        return async_log_wrapper if is_async else log_wrapper
-
-    return log_decorator