forked from phoenix/litellm-mirror
Merge pull request #1618 from BerriAI/litellm_sagemaker_cost_tracking_fixes
fix(utils.py): fix sagemaker cost tracking for streaming
This commit is contained in:
commit
612f74a426
11 changed files with 282 additions and 79 deletions
|
@ -718,6 +718,9 @@ async def update_database(
|
|||
existing_spend_obj = await custom_db_client.get_data(
|
||||
key=id, table_name="user"
|
||||
)
|
||||
verbose_proxy_logger.debug(
|
||||
f"Updating existing_spend_obj: {existing_spend_obj}"
|
||||
)
|
||||
if existing_spend_obj is None:
|
||||
existing_spend = 0
|
||||
existing_spend_obj = LiteLLM_UserTable(
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue