scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-03 02:22:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 81

test(infinity): colocate error regression coverage

Browse source
This commit is contained in:
Ricky-7-Yan 2026-08-31 22:45:24 +08:00
parent 4fd17756fd
commit d3c58cc991
1 changed files with 9 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
tests/llm_translation/test_infinity.py
View file

@ -5,6 +5,7 @@ from unittest.mock import AsyncMock
import litellm
from litellm.llms.infinity.common_utils import InfinityError
from unittest.mock import patch, MagicMock
@ -17,6 +18,14 @@ from litellm.llms.custom_httpx.http_handler import HTTPHandler, AsyncHTTPHandler
from litellm.types.utils import EmbeddingResponse, Usage
def test_infinity_error_default_headers_are_not_shared() -> None:
first_error = InfinityError(status_code=500, message="first")
second_error = InfinityError(status_code=500, message="second")
assert first_error.headers is None
assert second_error.headers is None
@pytest.mark.asyncio()
async def test_infinity_rerank():
mock_response = AsyncMock()