Class FakeLLM

backend/agent/tests/test_code_agent.py:28–43

Fake LLM client that returns configurable responses.

class FakeLLM:
    """Fake LLM client that returns configurable responses."""

    def __init__(self, responses: list[str] | None = None):
        self.calls: list[dict[str, Any]] = []
        self._responses = list(responses or [])
        self._call_idx = 0

    def chat(self, messages: list[dict], **kwargs: Any) -> LLMResponse:
        self.calls.append({"messages": messages, **kwargs})
        if self._responses:
            text = self._responses[min(self._call_idx, len(self._responses) - 1)]
        else:
            text = '```filename:main.py\nprint("hello")\n```'
        self._call_idx += 1
        return LLMResponse(content=text, model="fake-model")


@dataclass