mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-26 11:14:04 +00:00
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:
parent
4fa8991a90
commit
94a05ca5d0
263 changed files with 1687 additions and 3320 deletions
|
@ -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
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue