phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 19:24:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix - locking in memory leads to failing tests

Browse source
This commit is contained in:
Ishaan Jaff 2025-03-28 20:10:58 -07:00
parent fb98bd3a4a
commit 93c20b8d2d
4 changed files with 25 additions and 38 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/proxy/db/db_spend_update_writer.py
View file

@ -172,10 +172,9 @@ class DBSpendUpdateWriter:
f"track_cost_callback: {entity_type.value}_id is None. Not tracking spend for {entity_type.value}"
)
return False
async with prisma_client.in_memory_transaction_lock:
transaction_list[entity_id] = response_cost + transaction_list.get(
entity_id, 0
)
transaction_list[entity_id] = response_cost + transaction_list.get(
entity_id, 0
)
return True
except Exception as e: