scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-08-28 05:25:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 36

test(router): type the acreate_file fallback test helpers

Browse source
This commit is contained in:
mateo-berri 2026-08-18 19:57:24 -07:00
parent 2a1c21b72d
commit b7f4f531b3
2 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
tests/test_litellm/router_utils/test_fallback_event_handlers.py
View file

@ -1,4 +1,5 @@
import json
from typing import NoReturn
from unittest.mock import MagicMock, patch
import httpx
@ -167,7 +168,7 @@ async def _acreate_batch(*args, **kwargs):
raise AssertionError("only used for its __name__")
async def _acreate_file(*args, **kwargs):
async def _acreate_file(*args: object, **kwargs: object) -> NoReturn:
raise AssertionError("only used for its __name__")

2
tests/test_litellm/test_router.py
View file

@ -518,7 +518,7 @@ async def test_async_router_acreate_file_does_not_fall_back_across_model_groups(
fallbacks=[{"azure-gpt": ["openai-gpt"]}],
)
def fail_azure(*args, **kwargs):
def fail_azure(*args: object, **kwargs: object) -> MagicMock:
if kwargs.get("model") == "azure/my-azure-deployment":
raise litellm.APIConnectionError(
message="Connection error.",