# set callback
litellm.success_callback = [track_cost_callback] # set custom callback function