From 32f8bc28cedbe21cab198f9f2a9ad1d7f59f2fa5 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 22 Jan 2025 19:57:10 -0800 Subject: [PATCH] test: fix test --- tests/logging_callback_tests/test_spend_logs.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/tests/logging_callback_tests/test_spend_logs.py b/tests/logging_callback_tests/test_spend_logs.py index 32cb87a90f7..faad534cecb 100644 --- a/tests/logging_callback_tests/test_spend_logs.py +++ b/tests/logging_callback_tests/test_spend_logs.py @@ -170,6 +170,12 @@ def test_spend_logs_payload(model_id: Optional[str]): "end_time": datetime.datetime(2024, 6, 7, 12, 43, 30, 954146), "cache_hit": None, "response_cost": 2.4999999999999998e-05, + "standard_logging_object": { + "request_tags": ["model-anthropic-claude-v2.1", "app-ishaan-prod"], + "metadata": { + "user_api_key_end_user_id": "test-user", + }, + }, }, "response_obj": litellm.ModelResponse( id=model_id, @@ -343,6 +349,7 @@ def test_spend_logs_payload_with_prompts_enabled(monkeypatch): "metadata": { "user_api_key_end_user_id": "test-user", }, + "request_tags": ["model-anthropic-claude-v2.1", "app-ishaan-prod"], } input_args["kwargs"]["standard_logging_object"] = standard_logging_payload