fix: fix linting issue

This commit is contained in:
Krrish Dholakia 2024-03-21 08:19:09 -07:00
parent e904a84bdb
commit 8e8c4e214e
2 changed files with 9 additions and 11 deletions

View file

@ -199,9 +199,7 @@ class _OPTIONAL_PromptInjectionDetection(CustomLogger):
async def async_moderation_hook(
self,
data: dict,
call_type: (
Literal["completion"] | Literal["embeddings"] | Literal["image_generation"]
),
call_type: Literal["completion", "embeddings", "image_generation"],
):
verbose_proxy_logger.debug(
f"IN ASYNC MODERATION HOOK - self.prompt_injection_params = {self.prompt_injection_params}"