(fix) cost tracking

2024-02-07 17:06:05 -08:00 · 2024-02-07 17:06:05 -08:00 · bb469278c6
commit bb469278c6
parent bbbd37f0cb
1 changed files with 1 additions and 1 deletions
--- 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']}"