From aa9d97b23da97c86f2cdb2b86535d8a540279c6f Mon Sep 17 00:00:00 2001 From: Julio Quinteros Pro Date: Wed, 18 Feb 2026 11:24:22 -0300 Subject: [PATCH] fix(tests): use record.getMessage() instead of record.message for LogRecord MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit LogRecord objects do not have a .message attribute by default — it is only populated after the record has been formatted by a Formatter. The correct way to retrieve the formatted log message is getMessage(). This fixes AttributeError: 'LogRecord' object has no attribute 'message' in test_cost_calculation_uses_debug_level and test_batch_cost_calculation_uses_debug_level. Co-Authored-By: Claude Sonnet 4.6 --- tests/test_litellm/test_cost_calculation_log_level.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_litellm/test_cost_calculation_log_level.py b/tests/test_litellm/test_cost_calculation_log_level.py index 8ee9ad95cd..61f1e716e4 100644 --- a/tests/test_litellm/test_cost_calculation_log_level.py +++ b/tests/test_litellm/test_cost_calculation_log_level.py @@ -67,7 +67,7 @@ def test_cost_calculation_uses_debug_level(): # Find the cost calculation log records cost_calc_records = [ record for record in handler.records - if "selected model name for cost calculation" in record.message + if "selected model name for cost calculation" in record.getMessage() ] # Verify that cost calculation logs are at DEBUG level @@ -126,7 +126,7 @@ def test_batch_cost_calculation_uses_debug_level(): # Find batch cost calculation log records batch_cost_records = [ record for record in handler.records - if "Calculating batch cost per token" in record.message + if "Calculating batch cost per token" in record.getMessage() ] # Verify logs exist and are at DEBUG level