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

test(proxy): use the module-level HTTPException import in the /nvidia_nim route tests

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
yassin 2026-09-15 23:57:16 +00:00
parent 771b2509d1
commit 65f9d9bbbd
1 changed files with 0 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/test_litellm/proxy/pass_through_endpoints/test_llm_pass_through_endpoints.py
View file

@ -5484,8 +5484,6 @@ class TestNvidiaNimProxyRoute:
["v1/infer", "unknown-group/v1/infer", "nim-page-elements-v2/v1/infer", "gpt-4o/v1/infer"],
)
async def test_path_without_a_nim_model_group_is_rejected_before_any_upstream_call(self, endpoint):
from fastapi import HTTPException
captured: list[dict] = []
router = self._recording_router(
captured,
@ -5500,8 +5498,6 @@ class TestNvidiaNimProxyRoute:
@pytest.mark.asyncio
async def test_a_group_mixing_nim_and_other_deployments_is_rejected_before_any_upstream_call(self):
from fastapi import HTTPException
captured: list[dict] = []
class MixedRouter:
@ -5526,8 +5522,6 @@ class TestNvidiaNimProxyRoute:
@pytest.mark.asyncio
async def test_no_router_is_rejected_before_any_upstream_call(self):
from fastapi import HTTPException
with pytest.raises(HTTPException) as exc_info:
await self._relay(None, "nim-page-elements/v1/infer", NIM_INFER_BODY)