phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix create_audit_log_for_update

Browse source
This commit is contained in:
Ishaan Jaff 2024-10-25 16:48:25 +04:00
parent 61e28ebb3e
commit eb24ce25ea
3 changed files with 2 additions and 34 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -28,7 +28,7 @@ async def create_audit_log_for_update(request_data: LiteLLM_AuditLogs):
if isinstance(request_data.before_value, dict):
request_data.before_value = json.dumps(request_data.before_value)
_request_data = request_data.dict(exclude_none=True)
_request_data = request_data.model_dump(exclude_none=True)
try:
await prisma_client.db.litellm_auditlog.create(

2
litellm/proxy/proxy_config.yaml
View file

@ -13,7 +13,7 @@ general_settings:
proxy_batch_write_at: 60 # Batch write spend updates every 60s
litellm_settings:
success_callback: ["langfuse"]
store_audit_logs: true
# https://docs.litellm.ai/docs/proxy/reliability#default-fallbacks
default_fallbacks: ["gpt-4o-2024-08-06", "claude-3-5-sonnet-20240620"]

32
litellm/proxy/proxy_server.py
View file

@ -6434,38 +6434,6 @@ async def list_end_user(
return returned_response
async def create_audit_log_for_update(request_data: LiteLLM_AuditLogs):
if premium_user is not True:
return
if litellm.store_audit_logs is not True:
return
if prisma_client is None:
raise Exception("prisma_client is None, no DB connected")
verbose_proxy_logger.debug("creating audit log for %s", request_data)
if isinstance(request_data.updated_values, dict):
request_data.updated_values = json.dumps(request_data.updated_values)
if isinstance(request_data.before_value, dict):
request_data.before_value = json.dumps(request_data.before_value)
_request_data = request_data.dict(exclude_none=True)
try:
await prisma_client.db.litellm_auditlog.create(
data={
**_request_data, # type: ignore
}
)
except Exception as e:
# [Non-Blocking Exception. Do not allow blocking LLM API call]
verbose_proxy_logger.error(f"Failed Creating audit log {e}")
return
#### BUDGET TABLE MANAGEMENT ####