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

fix: success_callback logic for cost_tracking

Browse source 
Signed-off-by: Sihyeon Jang <uneedsihyeon@gmail.com>
This commit is contained in:
Sihyeon Jang 2023-12-21 16:07:15 +09:00
parent 87fca1808a
commit 4f41c3c513
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

@ -358,8 +358,6 @@ def cost_tracking():
print("setting litellm success callback to track cost") print("setting litellm success callback to track cost")
if (track_cost_callback) not in litellm.success_callback: # type: ignore if (track_cost_callback) not in litellm.success_callback: # type: ignore
litellm.success_callback.append(track_cost_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( async def track_cost_callback(
kwargs, # kwargs to completion kwargs, # kwargs to completion