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

Merge pull request #1211 from sihyeonn/fix/sh-success-callback

Browse source 
fix: success_callback logic for cost_tracking
This commit is contained in:
Krish Dholakia 2023-12-23 11:41:30 +05:30 • committed by GitHub
commit 8afdc12918
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/proxy/proxy_server.py
View file

@ -362,8 +362,6 @@ def cost_tracking():
print("setting litellm success callback to track cost")
if (track_cost_callback) not in litellm.success_callback: # type: ignore
litellm.success_callback.append(track_cost_callback) # type: ignore
else:
litellm.success_callback = track_cost_callback # type: ignore
async def track_cost_callback(
kwargs, # kwargs to completion