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

test: update tests

Browse source
This commit is contained in:
Krrish Dholakia 2025-01-22 18:34:14 -08:00
parent ca9f7ceed6
commit 4131659bd1
1 changed files with 4 additions and 3 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

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