From 3c463f6715af996e30905a8482fce718461ad234 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 19 Apr 2025 10:02:25 -0700 Subject: [PATCH] test fix - output_cost_per_reasoning_token was added to model cost map --- litellm/model_prices_and_context_window_backup.json | 1 + model_prices_and_context_window.json | 1 + tests/local_testing/test_get_model_info.py | 1 + 3 files changed, 3 insertions(+) diff --git a/litellm/model_prices_and_context_window_backup.json b/litellm/model_prices_and_context_window_backup.json index c2456ac3db..40d399aa68 100644 --- a/litellm/model_prices_and_context_window_backup.json +++ b/litellm/model_prices_and_context_window_backup.json @@ -5,6 +5,7 @@ "max_output_tokens": "max output tokens, if the provider specifies it. if not default to max_tokens", "input_cost_per_token": 0.0000, "output_cost_per_token": 0.000, + "output_cost_per_reasoning_token": 0.000, "litellm_provider": "one of https://docs.litellm.ai/docs/providers", "mode": "one of: chat, embedding, completion, image_generation, audio_transcription, audio_speech, image_generation, moderation, rerank", "supports_function_calling": true, diff --git a/model_prices_and_context_window.json b/model_prices_and_context_window.json index c2456ac3db..40d399aa68 100644 --- a/model_prices_and_context_window.json +++ b/model_prices_and_context_window.json @@ -5,6 +5,7 @@ "max_output_tokens": "max output tokens, if the provider specifies it. if not default to max_tokens", "input_cost_per_token": 0.0000, "output_cost_per_token": 0.000, + "output_cost_per_reasoning_token": 0.000, "litellm_provider": "one of https://docs.litellm.ai/docs/providers", "mode": "one of: chat, embedding, completion, image_generation, audio_transcription, audio_speech, image_generation, moderation, rerank", "supports_function_calling": true, diff --git a/tests/local_testing/test_get_model_info.py b/tests/local_testing/test_get_model_info.py index eccd34f7f9..41af51dee7 100644 --- a/tests/local_testing/test_get_model_info.py +++ b/tests/local_testing/test_get_model_info.py @@ -489,6 +489,7 @@ def test_aaamodel_prices_and_context_window_json_is_valid(): "output_cost_per_token_above_128k_tokens": {"type": "number"}, "output_cost_per_token_above_200k_tokens": {"type": "number"}, "output_cost_per_token_batches": {"type": "number"}, + "output_cost_per_reasoning_token": {"type": "number"}, "output_db_cost_per_token": {"type": "number"}, "output_dbu_cost_per_token": {"type": "number"}, "output_vector_size": {"type": "number"},