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

@ -16,11 +16,11 @@ repos:
name: Check if files match
entry: python3 ci_cd/check_files_match.py
language: system
- repo: local
hooks:
- id: mypy
name: mypy
entry: python3 -m mypy --ignore-missing-imports
language: system
types: [python]
files: ^litellm/
# - repo: local
# hooks:
# - id: mypy
# name: mypy
# entry: python3 -m mypy --ignore-missing-imports
# language: system
# types: [python]
# files: ^litellm/

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}"