scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-10 03:28:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 85

fix: resolve mypy type errors in audit_logs.py

Browse source
This commit is contained in:
Ryan Crabbe 2026-03-21 15:42:01 -07:00
parent f494ab513f
commit 3e27ff1b78
1 changed files with 3 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
litellm/proxy/management_helpers/audit_logs.py
View file

@ -51,9 +51,7 @@ def _build_audit_log_payload(
if request_data.updated_at is not None:
updated_at = request_data.updated_at.isoformat()
table_name = request_data.table_name
if isinstance(table_name, LitellmTableNames):
table_name = table_name.value
table_name_str: str = request_data.table_name.value if isinstance(request_data.table_name, LitellmTableNames) else str(request_data.table_name)
return StandardAuditLogPayload(
id=request_data.id,
@ -61,7 +59,7 @@ def _build_audit_log_payload(
changed_by=request_data.changed_by or "",
changed_by_api_key=request_data.changed_by_api_key or "",
action=request_data.action,
table_name=str(table_name),
table_name=table_name_str,
object_id=request_data.object_id,
before_value=request_data.before_value,
updated_values=request_data.updated_values,
@ -91,7 +89,7 @@ async def _dispatch_audit_log_to_callbacks(
for callback in litellm.audit_log_callbacks:
try:
resolved = callback
resolved: Optional[CustomLogger] = callback if isinstance(callback, CustomLogger) else None
if isinstance(callback, str):
resolved = _resolve_audit_log_callback(callback)
if resolved is None: