Returns responses in order across successive generate() calls.
| 16 | |
| 17 | |
| 18 | class SequentialMockLLM(LLMBackend): |
| 19 | """Returns responses in order across successive generate() calls.""" |
| 20 | |
| 21 | def __init__(self, responses: list[str]) -> None: |
| 22 | self._responses = iter(responses) |
| 23 | |
| 24 | def generate(self, prompt: str, temperature: float = 0.0) -> str: |
| 25 | return next(self._responses, "") |
| 26 | |
| 27 | |
| 28 | @pytest.fixture |