From 40a4e97d8e007ca361d91e5e37b5e35bab0c3450 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Tue, 27 May 2025 21:15:22 -0700 Subject: [PATCH] test: update test --- tests/proxy_unit_tests/test_e2e_pod_lock_manager.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/proxy_unit_tests/test_e2e_pod_lock_manager.py b/tests/proxy_unit_tests/test_e2e_pod_lock_manager.py index 8db51e5aad..a9bede3498 100644 --- a/tests/proxy_unit_tests/test_e2e_pod_lock_manager.py +++ b/tests/proxy_unit_tests/test_e2e_pod_lock_manager.py @@ -400,7 +400,7 @@ async def test_e2e_size_of_redis_buffer(): for queue in initialized_queues: key = f"test_key_{queue.__class__.__name__}_{uuid.uuid4()}" new_keys_added.append(key) - await queue.add_update({key: {"spend": 1.0}}) + await queue.add_update({key: {"spend": 1.0, "entity_id": "test_entity_id", "entity_type": "user", "api_key": "test_api_key", "model": "test_model", "custom_llm_provider": "test_custom_llm_provider", "date": "2025-01-01", "prompt_tokens": 100, "completion_tokens": 100, "total_tokens": 200, "response_cost": 1.0, "api_requests": 1, "successful_requests": 1, "failed_requests": 0}}) print("initialized_queues=", initialized_queues) print("new_keys_added=", new_keys_added)