From 4131659bd114ab6f0da305d9b6062ec44468d79c Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 22 Jan 2025 18:34:14 -0800 Subject: [PATCH] test: update tests --- tests/logging_callback_tests/test_spend_logs.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/tests/logging_callback_tests/test_spend_logs.py b/tests/logging_callback_tests/test_spend_logs.py index 9ecfe470461..32cb87a90f7 100644 --- a/tests/logging_callback_tests/test_spend_logs.py +++ b/tests/logging_callback_tests/test_spend_logs.py @@ -192,7 +192,6 @@ def test_spend_logs_payload(model_id: Optional[str]): ), "start_time": datetime.datetime(2024, 6, 7, 12, 43, 30, 308604), "end_time": datetime.datetime(2024, 6, 7, 12, 43, 30, 954146), - "end_user_id": None, } payload: SpendLogsPayload = get_logging_payload(**input_args) @@ -229,6 +228,7 @@ def test_spend_logs_payload_whisper(): "metadata": { "user_api_key": "88dc28d0f030c55ed4ab77ed8faf098196cb1c05df778539800c9f1243fe6b4b", "user_api_key_alias": None, + "user_api_key_end_user_id": "test-user", "user_api_end_user_max_budget": None, "litellm_api_version": "1.40.19", "global_max_parallel_requests": None, @@ -293,7 +293,6 @@ def test_spend_logs_payload_whisper(): response_obj=response, start_time=datetime.datetime.now(), end_time=datetime.datetime.now(), - end_user_id="test-user", ) print("payload: ", payload) @@ -335,13 +334,15 @@ def test_spend_logs_payload_with_prompts_enabled(monkeypatch): ), "start_time": datetime.datetime.now(), "end_time": datetime.datetime.now(), - "end_user_id": "user123", } # Create a standard logging payload standard_logging_payload = { "messages": [{"role": "user", "content": "Hello!"}], "response": {"role": "assistant", "content": "Hi there!"}, + "metadata": { + "user_api_key_end_user_id": "test-user", + }, } input_args["kwargs"]["standard_logging_object"] = standard_logging_payload