mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-07 08:26:10 +00:00
update
This commit is contained in:
parent
84fd1ac86c
commit
df6aacea18
1 changed files with 5 additions and 0 deletions
|
|
@ -1971,6 +1971,8 @@ class Logging(LiteLLMLoggingBaseClass):
|
|||
event_type="sync_success"
|
||||
): # prevent double logging
|
||||
return
|
||||
if kwargs.get("response_cost") is not None:
|
||||
self.model_call_details["response_cost"] = kwargs["response_cost"]
|
||||
start_time, end_time, result = self._success_handler_helper_fn(
|
||||
start_time=start_time,
|
||||
end_time=end_time,
|
||||
|
|
@ -2443,6 +2445,9 @@ class Logging(LiteLLMLoggingBaseClass):
|
|||
): # prevent double logging
|
||||
return
|
||||
|
||||
if kwargs.get("response_cost") is not None:
|
||||
self.model_call_details["response_cost"] = kwargs["response_cost"]
|
||||
|
||||
## CALCULATE COST FOR BATCH JOBS
|
||||
if self.call_type == CallTypes.aretrieve_batch.value and isinstance(
|
||||
result, LiteLLMBatch
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue