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

fix DBSpendUpdateWriter

Browse source
This commit is contained in:
Ishaan Jaff 2025-03-27 16:44:46 -07:00
parent a0fd508de4
commit 7995fd7c98
1 changed files with 13 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -1,3 +1,10 @@
"""
Module responsible for
1. Writing spend increments to either in memory list of transactions or to redis
2. Reading increments from redis or in memory list of transactions and committing them to db
"""
import asyncio import asyncio
import os import os
import traceback import traceback
@ -12,6 +19,12 @@ from litellm.proxy.utils import PrismaClient, ProxyUpdateSpend, hash_token
class DBSpendUpdateWriter: class DBSpendUpdateWriter:
"""
Module responsible for
1. Writing spend increments to either in memory list of transactions or to redis
2. Reading increments from redis or in memory list of transactions and committing them to db
"""
@staticmethod @staticmethod
async def update_database( # noqa: PLR0915 async def update_database( # noqa: PLR0915