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

litellm_fix(test): fix router silent experiment tests to properly mock async functions (#20140)

Browse source
This commit is contained in:
shin-bot-litellm 2026-01-31 07:39:05 -08:00 • committed by GitHub
parent 10194d96cf
commit e35e6504fc
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 7 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
tests/test_litellm/test_router_silent_experiment.py
View file

@ -93,15 +93,12 @@ async def test_router_silent_experiment_acompletion():
router = Router(model_list=model_list)
# Mock litellm.acompletion
mock_acompletion = MagicMock()
# Create a future that resolves to a ModelResponse
# Use AsyncMock for async function mocking
mock_response = litellm.ModelResponse(choices=[{"message": {"content": "hello"}}])
future = asyncio.Future()
future.set_result(mock_response)
mock_acompletion.return_value = future
mock_acompletion = AsyncMock(return_value=mock_response)
with patch("litellm.acompletion", mock_acompletion):
# Patch at the litellm.router module level where it's imported and used
with patch.object(litellm, "acompletion", mock_acompletion):
response = await router.acompletion(
model="primary-model",
messages=[{"role": "user", "content": "hi"}],
@ -177,11 +174,11 @@ def test_router_silent_experiment_completion():
router = Router(model_list=model_list)
# Mock litellm.completion
mock_completion = MagicMock()
mock_response = litellm.ModelResponse(choices=[{"message": {"content": "hello"}}])
mock_completion.return_value = mock_response
mock_completion = MagicMock(return_value=mock_response)
with patch("litellm.completion", mock_completion):
# Patch at the litellm module level
with patch.object(litellm, "completion", mock_completion):
response = router.completion(
model="primary-model",
messages=[{"role": "user", "content": "hi"}],