From e2c31fd471adede74df2e796b157b366aff89f8c Mon Sep 17 00:00:00 2001 From: Sameer Kankute Date: Wed, 24 Jun 2026 15:48:06 +0530 Subject: [PATCH] style: black formatting for asqav.py and usage_endpoints.py Co-Authored-By: Claude Sonnet 4.6 --- litellm/integrations/asqav/asqav.py | 24 +++++++++++++++++---- litellm/proxy/guardrails/usage_endpoints.py | 4 +++- 2 files changed, 23 insertions(+), 5 deletions(-) diff --git a/litellm/integrations/asqav/asqav.py b/litellm/integrations/asqav/asqav.py index d6680c4b547..ad99910649d 100644 --- a/litellm/integrations/asqav/asqav.py +++ b/litellm/integrations/asqav/asqav.py @@ -370,17 +370,29 @@ class AsqavLogger(CustomLogger): # ------------------------------------------------------------------ def log_success_event( - self, kwargs: dict[str, Any], response_obj: Any, start_time: Any, end_time: Any # noqa: ANN401 + self, + kwargs: dict[str, Any], + response_obj: Any, + start_time: Any, + end_time: Any, # noqa: ANN401 ) -> None: self._build_and_append(kwargs, response_obj, start_time, end_time, "success") def log_failure_event( - self, kwargs: dict[str, Any], response_obj: Any, start_time: Any, end_time: Any # noqa: ANN401 + self, + kwargs: dict[str, Any], + response_obj: Any, + start_time: Any, + end_time: Any, # noqa: ANN401 ) -> None: self._build_and_append(kwargs, response_obj, start_time, end_time, "failure") async def async_log_success_event( - self, kwargs: dict[str, Any], response_obj: Any, start_time: Any, end_time: Any # noqa: ANN401 + self, + kwargs: dict[str, Any], + response_obj: Any, + start_time: Any, + end_time: Any, # noqa: ANN401 ) -> None: await asyncio.to_thread( self._build_and_append, @@ -392,7 +404,11 @@ class AsqavLogger(CustomLogger): ) async def async_log_failure_event( - self, kwargs: dict[str, Any], response_obj: Any, start_time: Any, end_time: Any # noqa: ANN401 + self, + kwargs: dict[str, Any], + response_obj: Any, + start_time: Any, + end_time: Any, # noqa: ANN401 ) -> None: await asyncio.to_thread( self._build_and_append, diff --git a/litellm/proxy/guardrails/usage_endpoints.py b/litellm/proxy/guardrails/usage_endpoints.py index 209e6c7a5d3..0c074eba3fa 100644 --- a/litellm/proxy/guardrails/usage_endpoints.py +++ b/litellm/proxy/guardrails/usage_endpoints.py @@ -175,7 +175,9 @@ def _get_config_loaded_guardrails() -> list[Any]: ] -def _find_config_loaded_guardrail(guardrail_id_or_name: str) -> Optional[Any]: # noqa: ANN401 +def _find_config_loaded_guardrail( + guardrail_id_or_name: str, +) -> Optional[Any]: # noqa: ANN401 for guardrail in _get_config_loaded_guardrails(): gid, display_name = _get_guardrail_attrs(guardrail) if guardrail_id_or_name in (gid, display_name):