Class _FakeClassifierClient

tests/test_llm_classifier_request_processor.py:42–67

class _FakeClassifierClient:
    def __init__(
        self,
        response: str | Exception,
        *,
        usage: _FakeUsage | None = None,
    ) -> None:
        self.response = response
        self.usage = usage
        self.calls: list[dict[str, str]] = []

    async def classify(
        self,
        *,
        model: str,
        system_prompt: str,
        request_summary: str,
    ) -> ClassifierCompletion:
        self.calls.append({
            "model": model,
            "system_prompt": system_prompt,
            "request_summary": request_summary,
        })
        if isinstance(self.response, Exception):
            raise self.response
        return ClassifierCompletion(content=self.response, usage=self.usage)


def _messages_with_prior_assistant_turns(n: int) -> list[dict[str, str]]: