forked from phoenix/litellm-mirror
Merge pull request #1211 from sihyeonn/fix/sh-success-callback
fix: success_callback logic for cost_tracking
This commit is contained in:
commit
8afdc12918
1 changed files with 0 additions and 2 deletions
|
@ -362,8 +362,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
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue