scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-06 08:16:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 78

Merge pull request #39842 from BerriAI/litellm_lit_6949_unknown_model_spend_log_test

Browse source 
test(store_model_in_db): assert the 400 contract in the unknown-model spend log test
This commit is contained in:
Mateo Wang 2026-09-04 20:29:56 -07:00 • committed by GitHub
commit b77b7f158f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
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,7 @@ 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["model_group"] in ("", "non-existent-model")
assert log_entry["spend"] == 0.0
assert log_entry["total_tokens"] == 0
assert log_entry["prompt_tokens"] == 0
@ -206,8 +209,7 @@ 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 error_info["error_class"] in ("ProxyModelNotFoundError", "BadRequestError")
assert "non-existent-model" in error_info["error_message"]
# Verify request details