Skip to content

Commit face817

Browse files
committed
fix failing test
1 parent 7bdf52c commit face817

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/test_litellm/proxy/hooks/test_dynamic_rate_limiter_v3.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1403,13 +1403,13 @@ async def mock_increment(pipeline_operations, parent_otel_span=None):
14031403
end_time=None,
14041404
)
14051405

1406-
# Verify increments happened with actual token count (50 completion tokens)
1406+
# Verify increments happened with actual token count (60 total tokens)
14071407
assert len(increment_calls) == 2, f"Expected 2 increment calls, got {len(increment_calls)}"
14081408

1409-
# Both should increment by 50 (completion_tokens, since rate_limit_type defaults to 'output')
1409+
# Both should increment by 50 (total_tokens, since rate_limit_type defaults to 'total')
14101410
for call in increment_calls:
1411-
assert call["increment_value"] == 50, (
1412-
f"Expected increment of 50 tokens, got {call['increment_value']} for key {call['key']}"
1411+
assert call["increment_value"] == 60, (
1412+
f"Expected increment of 60 tokens, got {call['increment_value']} for key {call['key']}"
14131413
)
14141414

14151415
# Verify correct keys were used

0 commit comments

Comments
 (0)