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());

        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("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());

        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));

        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

        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() {

        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

        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");

                .param("size", ComponentUtil.getFessConfig().getPageDictionaryMaxFetchSize())
                .execute()) {
            final Map<String, Object> contentMap = response.getContent(OpenSearchCurl.jsonParser());
            @SuppressWarnings("unchecked")
            final List<Map<String, Object>> fileList = (List<Map<String, Object>>) contentMap.get("file");

            final String truncated = newest.getContent().substring(0, Math.min(budgetChars, newest.getContent().length()));
            request.addMessage(new LlmMessage(newest.getRole(), truncated));
            if (logger.isDebugEnabled()) {
                logger.debug("[RAG:ANSWER] Newest history message truncated to fit budget. originalLength={}, truncatedLength={}",
                        newest.getContent().length(), truncated.length());