phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix(proxy_server.py): use consistent 400-status code error code for exceeded budget errors

Browse source 
standardizes error code for budget exceeded errors to status code 400
This commit is contained in:
Krrish Dholakia 2024-06-11 16:10:58 -07:00
parent a389ec95a2
commit c27ae34a39
4 changed files with 19 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tests/test_keys.py
View file

@ -664,7 +664,7 @@ async def test_key_crossing_budget():
response = await chat_completion(session=session, key=key)
pytest.fail("Should have failed - Key crossed it's budget")
except Exception as e:
assert "ExceededTokenBudget: Current spend for token:" in str(e)
assert "Budget has been exceeded!" in str(e)
@pytest.mark.skip(reason="AWS Suspended Account")