Function _mock_acompletion

tests/test_compiler.py:1132–1146

Create an async mock for litellm.acompletion.

(responses: list[str])

def _mock_acompletion(responses: list[str]):
    """Create an async mock for litellm.acompletion."""
    call_count = {"n": 0}

    async def side_effect(*args, **kwargs):
        idx = min(call_count["n"], len(responses) - 1)
        call_count["n"] += 1
        mock_resp = MagicMock()
        mock_resp.choices = [MagicMock()]
        mock_resp.choices[0].message.content = responses[idx]
        mock_resp.usage = MagicMock(prompt_tokens=100, completion_tokens=50)
        mock_resp.usage.prompt_tokens_details = None
        return mock_resp

    return side_effect


class TestCompileShortDoc: