feat(proxy/utils.py): enable end_user + team id tracking in spend logs

allows separating proxy owner vs. end user in the spen
d logs
This commit is contained in:
Krrish Dholakia 2024-02-29 16:04:52 -08:00
parent d9e272942b
commit c60fa9c84d
5 changed files with 18 additions and 6 deletions

View file

@ -1152,9 +1152,9 @@ async def update_database(
payload["spend"] = response_cost
if prisma_client is not None:
await prisma_client.insert_data(data=payload, table_name="spend")
elif custom_db_client is not None:
await custom_db_client.insert_data(payload, table_name="spend")
except Exception as e:
verbose_proxy_logger.info(f"Update Spend Logs DB failed to execute")