scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-12 23:01:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 88

Merge pull request #21476 from BerriAI/fix/test-log-record-message-attr

Browse source 
fix(tests): use record.getMessage() instead of record.message for LogRecord
This commit is contained in:
jquinter 2026-02-19 02:21:54 -03:00 • committed by GitHub
commit abc4603675
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/test_litellm/test_cost_calculation_log_level.py
View file

@ -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