scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-09 03:18:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 108

test: fix test

Browse source
This commit is contained in:
Krrish Dholakia 2025-01-22 19:57:10 -08:00
parent fd4ac110be
commit 32f8bc28ce
1 changed files with 7 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
tests/logging_callback_tests/test_spend_logs.py
View file

@ -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