assertEquals(Integer.valueOf(75), response.getCompletionTokens());
        assertEquals(Integer.valueOf(325), response.getTotalTokens());
    }

    @Test
    public void test_setModel() {
        final LlmChatResponse response = new LlmChatResponse();

        response.setModel("llama3");
        assertEquals("llama3", response.getModel());

        response.setModel("gpt-3.5-turbo");

        request.setMessages(messages);

        assertSame(messages, request.getMessages());
        assertEquals(2, request.getMessages().size());
    }

    @Test
    public void test_setModel() {
        final LlmChatRequest request = new LlmChatRequest();

        final LlmChatRequest result = request.setModel("gpt-4");

        assertSame(request, result);