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(store_model_in_db): assert the 400 contract in the unknown-model spend log test

Browse source
This commit is contained in:
mateo-berri 2026-09-04 18:23:36 -07:00
parent 59d42d36e6
commit f27699a1c9
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/store_model_in_db_tests/test_openai_error_handling.py
View file

@ -157,6 +157,9 @@ async def test_chat_completion_bad_model_with_spend_logs():
except json.JSONDecodeError:
print(f"Could not parse response body as JSON: {response.text}")
assert (
response.status_code == 400
), f"expected HTTP 400, got {response.status_code}: {response.text}"
assert (
litellm_call_id is not None
), "Failed to get LiteLLM Call ID from response headers"
@ -191,7 +194,6 @@ async def test_chat_completion_bad_model_with_spend_logs():
# Verify the structure of the log entry
assert log_entry["request_id"] == litellm_call_id
assert log_entry["model"] == "non-existent-model"
assert log_entry["model_group"] == "non-existent-model"
assert log_entry["spend"] == 0.0
assert log_entry["total_tokens"] == 0
assert log_entry["prompt_tokens"] == 0
@ -206,8 +208,6 @@ async def test_chat_completion_bad_model_with_spend_logs():
error_info = log_entry["metadata"]["error_information"]
assert "traceback" in error_info
assert error_info["error_code"] == "400"
assert error_info["error_class"] == "BadRequestError"
assert "litellm.BadRequestError" in error_info["error_message"]
assert "non-existent-model" in error_info["error_message"]
# Verify request details