scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-07 02:59:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

test_stream_transformation_error_sync

Browse source
This commit is contained in:
Ishaan Jaffer 2026-01-24 17:19:21 -08:00
parent 06f7a7ec69
commit 73dd1bd97a
1 changed files with 9 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
tests/test_litellm/google_genai/test_google_genai_handler.py
View file

@ -183,8 +183,10 @@ def test_stream_transformation_error_sync():
"translate_completion_output_params_streaming",
return_value=None
):
# Mock at the module level where it's imported
with patch("litellm.google_genai.adapters.handler.litellm.completion", return_value=mock_stream):
# Mock litellm.completion at the module level where it's imported
# We need to patch it in the handler module, not in litellm itself
with patch("litellm.google_genai.adapters.handler.litellm") as mock_litellm:
mock_litellm.completion.return_value = mock_stream
# Call the handler with stream=True and expect a ValueError
with pytest.raises(ValueError, match="Failed to transform streaming response"):
GenerateContentToCompletionHandler.generate_content_handler(
@ -210,8 +212,11 @@ async def test_stream_transformation_error_async():
"translate_completion_output_params_streaming",
return_value=None
):
# Mock at the module level where it's imported
with patch("litellm.google_genai.adapters.handler.litellm.acompletion", return_value=mock_stream):
# Mock litellm.acompletion at the module level where it's imported
# We need to patch it in the handler module, not in litellm itself
with patch("litellm.google_genai.adapters.handler.litellm") as mock_litellm:
# Use AsyncMock for async function
mock_litellm.acompletion = AsyncMock(return_value=mock_stream)
# Call the handler with stream=True and expect a ValueError
with pytest.raises(ValueError, match="Failed to transform streaming response"):
await GenerateContentToCompletionHandler.async_generate_content_handler(