Method set_sequential_responses

tests/mock_client.py:56–61

Set the mock to return different responses sequentially. :param responses: A list of ChatCompletion responses to return in order.

(self, responses: list[ChatCompletion])

        self.chat.completions.create.return_value = response

    def set_sequential_responses(self, responses: list[ChatCompletion]):
        """
        Set the mock to return different responses sequentially.
        :param responses: A list of ChatCompletion responses to return in order.
        """
        self.chat.completions.create.side_effect = responses

    def assert_create_called_with(self, **kwargs):
        self.chat.completions.create.assert_called_with(**kwargs)