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

@ -1,8 +1,8 @@
#### What this does ####
# On success + failure, log events to lunary.ai
from datetime import datetime, timezone
import traceback
import importlib
import traceback
from datetime import datetime, timezone
import packaging
@ -74,9 +74,9 @@ class LunaryLogger:
try:
import lunary
version = importlib.metadata.version("lunary")
version = importlib.metadata.version("lunary") # type: ignore
# if version < 0.1.43 then raise ImportError
if packaging.version.Version(version) < packaging.version.Version("0.1.43"):
if packaging.version.Version(version) < packaging.version.Version("0.1.43"): # type: ignore
print( # noqa
"Lunary version outdated. Required: >= 0.1.43. Upgrade via 'pip install lunary --upgrade'"
)
@ -97,7 +97,7 @@ class LunaryLogger:
run_id,
model,
print_verbose,
extra=None,
extra={},
input=None,
user_id=None,
response_obj=None,
@ -116,7 +116,7 @@ class LunaryLogger:
if optional_params:
extra = {**extra, **optional_params}
tags = metadata.get("tags", None)
tags = metadata.get("tags", None)
if extra:
extra.pop("extra_body", None)
@ -128,7 +128,7 @@ class LunaryLogger:
if not isinstance(value, (str, int, bool, float)) and param != "tools":
try:
extra[param] = str(value)
except:
except Exception:
pass
if response_obj:
@ -175,6 +175,6 @@ class LunaryLogger:
token_usage=usage,
)
except:
except Exception:
print_verbose(f"Lunary Logging Error - {traceback.format_exc()}")
pass