Litellm ruff linting enforcement (#5992)

* ci(config.yml): add a 'check_code_quality' step

Addresses https://github.com/BerriAI/litellm/issues/5991

* ci(config.yml): check why circle ci doesn't pick up this test

* ci(config.yml): fix to run 'check_code_quality' tests

* fix(__init__.py): fix unprotected import

* fix(__init__.py): don't remove unused imports

* build(ruff.toml): update ruff.toml to ignore unused imports

* fix: fix: ruff + pyright - fix linting + type-checking errors

* fix: fix linting errors

* fix(lago.py): fix module init error

* fix: fix linting errors

* ci(config.yml): cd into correct dir for checks

* fix(proxy_server.py): fix linting error

* fix(utils.py): fix bare except

causes ruff linting errors

* fix: ruff - fix remaining linting errors

* fix(clickhouse.py): use standard logging object

* fix(__init__.py): fix unprotected import

* fix: ruff - fix linting errors

* fix: fix linting errors

* ci(config.yml): cleanup code qa step (formatting handled in local_testing)

* fix(_health_endpoints.py): fix ruff linting errors

* ci(config.yml): just use ruff in check_code_quality pipeline for now

* build(custom_guardrail.py): include missing file

* style(embedding_handler.py): fix ruff check
This commit is contained in:
Krish Dholakia 2024-10-01 16:44:20 -07:00 committed by GitHub
parent 3fc4ae0d65
commit d57be47b0f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
263 changed files with 1687 additions and 3320 deletions

View file

@ -79,7 +79,7 @@ class _OPTIONAL_PromptInjectionDetection(CustomLogger):
if (
self.prompt_injection_params is not None
and self.prompt_injection_params.llm_api_check == True
and self.prompt_injection_params.llm_api_check is True
):
if self.llm_router is None:
raise Exception(
@ -146,7 +146,7 @@ class _OPTIONAL_PromptInjectionDetection(CustomLogger):
- check if user id part of call
- check if user id part of blocked list
"""
self.print_verbose(f"Inside Prompt Injection Detection Pre-Call Hook")
self.print_verbose("Inside Prompt Injection Detection Pre-Call Hook")
try:
assert call_type in [
"completion",
@ -156,7 +156,7 @@ class _OPTIONAL_PromptInjectionDetection(CustomLogger):
"moderation",
"audio_transcription",
]
except Exception as e:
except Exception:
self.print_verbose(
f"Call Type - {call_type}, not in accepted list - ['completion','embeddings','image_generation','moderation','audio_transcription']"
)
@ -244,7 +244,7 @@ class _OPTIONAL_PromptInjectionDetection(CustomLogger):
# 3. check if llm api check turned on
if (
self.prompt_injection_params.llm_api_check == True
self.prompt_injection_params.llm_api_check is True
and self.prompt_injection_params.llm_api_name is not None
and self.llm_router is not None
):
@ -270,7 +270,7 @@ class _OPTIONAL_PromptInjectionDetection(CustomLogger):
if self.prompt_injection_params.llm_api_fail_call_string in response.choices[0].message.content: # type: ignore
is_prompt_attack = True
if is_prompt_attack == True:
if is_prompt_attack is True:
raise HTTPException(
status_code=400,
detail={