Function _mock_completion

tests/test_compiler.py:1115–1129

Create a mock for litellm.completion that returns responses in order.

(responses: list[str])

def _mock_completion(responses: list[str]):
    """Create a mock for litellm.completion that returns responses in order."""
    call_count = {"n": 0}

    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


def _mock_acompletion(responses: list[str]):