phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 11:43:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: fix linting issue

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

16
.pre-commit-config.yaml
View file

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

4
litellm/proxy/hooks/prompt_injection_detection.py
View file

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