fix linting

Author: ishaan-jaff

enterprise/enterprise_hooks/google_text_moderation.py

@@ -94,6 +94,7 @@ async def async_moderation_hook(
             "image_generation",
             "moderation",
             "audio_transcription",
+            "responses",
         ],
     ):
         """

enterprise/enterprise_hooks/llama_guard.py

@@ -107,6 +107,7 @@ async def async_moderation_hook(
             "image_generation",
             "moderation",
             "audio_transcription",
+            "responses",
         ],
     ):
         """

enterprise/enterprise_hooks/llm_guard.py

@@ -126,6 +126,7 @@ async def async_moderation_hook(
             "image_generation",
             "moderation",
             "audio_transcription",
+            "responses",
         ],
     ):
         """

litellm/proxy/example_config_yaml/custom_guardrail.py

@@ -66,6 +66,7 @@ async def async_moderation_hook(
             "image_generation",
             "moderation",
             "audio_transcription",
+            "responses",
         ],
     ):
         """

litellm/proxy/guardrails/guardrail_hooks/aim.py

@@ -25,16 +25,22 @@ class AimGuardrailMissingSecrets(Exception):
 
 
 class AimGuardrail(CustomGuardrail):
-    def __init__(self, api_key: Optional[str] = None, api_base: Optional[str] = None, **kwargs):
-        self.async_handler = get_async_httpx_client(llm_provider=httpxSpecialProvider.GuardrailCallback)
+    def __init__(
+        self, api_key: Optional[str] = None, api_base: Optional[str] = None, **kwargs
+    ):
+        self.async_handler = get_async_httpx_client(
+            llm_provider=httpxSpecialProvider.GuardrailCallback
+        )
         self.api_key = api_key or os.environ.get("AIM_API_KEY")
         if not self.api_key:
             msg = (
                 "Couldn't get Aim api key, either set the `AIM_API_KEY` in the environment or "
                 "pass it as a parameter to the guardrail in the config file"
             )
             raise AimGuardrailMissingSecrets(msg)
-        self.api_base = api_base or os.environ.get("AIM_API_BASE") or "https://api.aim.security"
+        self.api_base = (
+            api_base or os.environ.get("AIM_API_BASE") or "https://api.aim.security"
+        )
         super().__init__(**kwargs)
 
     async def async_pre_call_hook(
@@ -68,6 +74,7 @@ async def async_moderation_hook(
             "image_generation",
             "moderation",
             "audio_transcription",
+            "responses",
         ],
     ) -> Union[Exception, str, dict, None]:
         verbose_proxy_logger.debug("Inside AIM Moderation Hook")
@@ -77,9 +84,10 @@ async def async_moderation_hook(
 
     async def call_aim_guardrail(self, data: dict, hook: str) -> None:
         user_email = data.get("metadata", {}).get("headers", {}).get("x-aim-user-email")
-        headers = {"Authorization": f"Bearer {self.api_key}", "x-aim-litellm-hook": hook} | (
-            {"x-aim-user-email": user_email} if user_email else {}
-        )
+        headers = {
+            "Authorization": f"Bearer {self.api_key}",
+            "x-aim-litellm-hook": hook,
+        } | ({"x-aim-user-email": user_email} if user_email else {})
         response = await self.async_handler.post(
             f"{self.api_base}/detect/openai",
             headers=headers,

litellm/proxy/guardrails/guardrail_hooks/custom_guardrail.py

@@ -70,6 +70,7 @@ async def async_moderation_hook(
             "image_generation",
             "moderation",
             "audio_transcription",
+            "responses",
         ],
     ):
         """