mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-07 02:59:05 +00:00
fix: correct patch target for spend log queue size in tests
Co-authored-by: Ishaan Jaff <ishaan-jaff@users.noreply.github.com>
This commit is contained in:
parent
09ca2315b1
commit
282633d3d2
1 changed files with 2 additions and 2 deletions
|
|
@ -148,7 +148,7 @@ class TestSpendLogQueueBounding:
|
|||
|
||||
max_size = 100 # Use a small value for testing
|
||||
with patch(
|
||||
"litellm.proxy.db.db_spend_update_writer.MAX_SPEND_LOG_TRANSACTIONS_QUEUE_SIZE",
|
||||
"litellm.constants.MAX_SPEND_LOG_TRANSACTIONS_QUEUE_SIZE",
|
||||
max_size,
|
||||
):
|
||||
# Fill the queue beyond the limit
|
||||
|
|
@ -178,7 +178,7 @@ class TestSpendLogQueueBounding:
|
|||
|
||||
max_size = 20
|
||||
with patch(
|
||||
"litellm.proxy.db.db_spend_update_writer.MAX_SPEND_LOG_TRANSACTIONS_QUEUE_SIZE",
|
||||
"litellm.constants.MAX_SPEND_LOG_TRANSACTIONS_QUEUE_SIZE",
|
||||
max_size,
|
||||
):
|
||||
# Fill exactly to the limit
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue