assertEquals("User 2", messages.get(0).getContent());
        assertEquals(ChatMessage.ROLE_ASSISTANT, messages.get(1).getRole());
        assertEquals("Assistant 2", messages.get(1).getContent());
        assertEquals(ChatMessage.ROLE_USER, messages.get(2).getRole());
        assertEquals("User 3", messages.get(2).getContent());
        assertEquals(ChatMessage.ROLE_ASSISTANT, messages.get(3).getRole());

        assertEquals("", documentHelper.getContent(null, responseData, "", dataMap));
        assertEquals("", documentHelper.getContent(null, responseData, " ", dataMap));
        assertEquals("", documentHelper.getContent(null, responseData, "  ", dataMap));
        assertEquals("", documentHelper.getContent(null, responseData, "\t", dataMap));

        final LlmChatResponse response = new LlmChatResponse();
        response.setContent("Test content");
        assertEquals("Test content", response.getContent());

        response.setContent("Updated content");
        assertEquals("Updated content", response.getContent());
    }

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

        assertEquals("Q2", messages.get(1).getContent());
        assertEquals("A2", messages.get(2).getContent());
        assertEquals("Q3", messages.get(3).getContent());
        assertEquals("A3", messages.get(4).getContent());
        assertEquals("Q4", messages.get(5).getContent());
        assertEquals("A4", messages.get(6).getContent());
    }

    @Test

        assertEquals("Hello", message.getContent());
        assertNotNull(message.getTimestamp());
        assertNotNull(message.getSources());
    }

    @Test
    public void test_userMessage() {
        final ChatMessage message = ChatMessage.userMessage("Hello, World!");
        assertEquals(ChatMessage.ROLE_USER, message.getRole());
        assertEquals("Hello, World!", message.getContent());
        assertTrue(message.isUser());

        assertNull(message.getRole());
        assertNull(message.getContent());
    }

    @Test
    public void test_constructorWithRoleAndContent() {
        final LlmMessage message = new LlmMessage("system", "You are a helpful assistant.");
        assertEquals("system", message.getRole());
        assertEquals("You are a helpful assistant.", message.getContent());
    }

    @Test
    public void test_systemFactory() {

        return connection.getHeaderField(index);
    }

    @Override
    public Object getContent() throws IOException {
        try {
            handshake();
        } catch (final IOException ex) {}
        return connection.getContent();
    }

    @Override
    public Object getContent(final Class[] classes) throws IOException {
        try {
            handshake();

        final LlmChatRequest result = request.addMessage(message);

        assertSame(request, result);
        assertEquals(1, request.getMessages().size());
        assertEquals("Test message", request.getMessages().get(0).getContent());
    }

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

        final LlmChatRequest result = request.addSystemMessage("System prompt");

        assertEquals("Q1", history.get(0).getContent());
        assertEquals("assistant", history.get(1).getRole());
        assertEquals("A1", history.get(1).getContent());
        assertEquals("user", history.get(2).getRole());
        assertEquals("Q2", history.get(2).getContent());
        assertEquals("assistant", history.get(3).getRole());
        assertEquals("A2", history.get(3).getContent());
    }

    @Test

                } else {
                    String content;
                    try {
                        content = new String(InputStreamUtil.getBytes(entity.getContent()), Constants.UTF_8_CHARSET);
                    } catch (final IOException e) {
                        content = e.getMessage();
                    }