scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-30 01:52:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 70

test(rust): enforce shared upstream error contract in wheel checks (#43520)

Browse source 
Co-authored-by: Yujong Lee <yujong@berri.ai>
This commit is contained in:
devin-ai-integration[bot] 2026-09-28 08:15:41 -07:00 • committed by GitHub
parent 2c9b0e00ac
commit c60c714278
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 6 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
tests/unit/rust_bridge/native_route_wheel_test.py
View file

@ -153,13 +153,8 @@ def success_value(route: str, response: dict[object, object]) -> object:
return response["choices"][0]["message"]["content"]
def assert_rate_limit(native: object, route: str, error: BaseException) -> None:
if route == "chat_completions":
upstream_error: Final = native.RustUpstreamError
if not isinstance(error, upstream_error) or error.args[0] != 429:
raise AssertionError(f"{route} returned the wrong 429 error: {error!r}")
return
if not isinstance(error, RuntimeError) or "429" not in str(error):
def assert_rate_limit(route: str, error: BaseException) -> None:
if error.args != (429, native_response(429, route).decode()):
raise AssertionError(f"{route} returned the wrong 429 error: {error!r}")
@ -169,8 +164,8 @@ def exercise_sync(native: object, api_base: str) -> None:
assert_success(route, function(**route_kwargs(route, api_base, "success")))
try:
function(**route_kwargs(route, api_base, "429"))
except (RuntimeError, native.RustUpstreamError) as error:
assert_rate_limit(native, route, error)
except native.RustUpstreamError as error:
assert_rate_limit(route, error)
else:
raise AssertionError(f"{route} accepted a 429 response")
@ -181,8 +176,8 @@ async def exercise_async(native: object, api_base: str) -> None:
assert_success(route, await function(**route_kwargs(route, api_base, "success")))
try:
await function(**route_kwargs(route, api_base, "429"))
except (RuntimeError, native.RustUpstreamError) as error:
assert_rate_limit(native, route, error)
except native.RustUpstreamError as error:
assert_rate_limit(route, error)
else:
raise AssertionError(f"a{route} accepted a 429 response")