From bb469278c6dcd26ff5bccc1f48aa851cefd64232 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Wed, 7 Feb 2024 17:06:05 -0800 Subject: [PATCH] (fix) cost tracking --- litellm/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/utils.py b/litellm/utils.py index 592be4aa9d..7aad34b2cd 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -1508,7 +1508,7 @@ class Logging: # base_model defaults to None if not set on model_info self.model_call_details["response_cost"] = litellm.completion_cost( completion_response=complete_streaming_response, - base_model=base_model, + model=base_model, ) verbose_logger.debug( f"Model={self.model}; cost={self.model_call_details['response_cost']}"