session.addMessage(message);

        assertNotNull(session.getMessages());
        assertEquals(1, session.getMessages().size());
    }

    @Test
    public void test_addUserMessage() {
        final ChatSession session = new ChatSession();
        session.addUserMessage("User message content");

        assertEquals(1, session.getMessages().size());

    }

    /**
     * Returns a copy of the message list in this session.
     *
     * @return a new list containing all messages
     */
    public List<ChatMessage> getMessages() {
        synchronized (messagesLock) {
            if (messages == null) {
                return new ArrayList<>();
            }
            return new ArrayList<>(messages);
        }
    }

     * Gets the messages in this request.
     *
     * @return the list of messages
     */
    public List<LlmMessage> getMessages() {
        return messages;
    }

    /**
     * Sets the messages in this request.
     *
     * @param messages the list of messages
     */
    public void setMessages(final List<LlmMessage> messages) {
        this.messages = messages;
    }

    /**
     * Gets the model name.

         */
        public List<WitnessNotificationMessage> getMessages() {
            return messages;
        }

        /**
         * Sets the list of notification messages.
         *
         * @param messages the list of individual notification messages contained in this response
         */
        public void setMessages(List<WitnessNotificationMessage> messages) {
            this.messages = messages;

        assertEquals(4, request.getMessages().size());
        assertEquals("Q1", request.getMessages().get(0).getContent());
        assertEquals("A1", request.getMessages().get(1).getContent());
        assertEquals("Q2", request.getMessages().get(2).getContent());
        assertEquals("A2", request.getMessages().get(3).getContent());
    }

    @Test

        final String llmType = getLlmType();
        if (logger.isDebugEnabled()) {
            logger.debug("[LLM] Starting LLM chat request. llmType={}, messageCount={}", llmType, request.getMessages().size());
            for (final LlmMessage msg : request.getMessages()) {
                logger.debug("[LLM] message: role={}, content={}", msg.getRole(), msg.getContent());
            }
        }
        try {

                    if (response != null) {
                        // Process each notification message
                        for (WitnessAsyncNotifyMessage.WitnessNotificationMessage message : response.getMessages()) {
                            processAsyncNotificationMessage(registration, message);
                        }
                    }

            if (e instanceof LlmException) {
                logger.warn("[RAG] LLM error during chat. sessionId={}, error={}", session.getSessionId(), e.getMessage());
            } else {
                logger.warn("[RAG] Unexpected error during chat. sessionId={}, error={}", session.getSessionId(), e.getMessage(), e);
            }
            throw e;
        } finally {
            session.trimHistory(getMaxHistoryMessages());
        }

        String fullMessage = message + "\n" + "Exception: "
                + e.getClass().getName() + "\n" + "Message: "
                + e.getMessage() + "\n" + "Stack trace:\n"
                + stackTrace;

        processingEnv.getMessager().printMessage(Diagnostic.Kind.ERROR, fullMessage);
    }

        // Then
        assertTrue(exception.getMessage().toLowerCase().contains("sharing") || exception.getMessage().toLowerCase().contains("violation")
                || exception.getMessage().toLowerCase().contains("used") || exception.getMessage().toLowerCase().contains("being used")
                || exception.getMessage().toLowerCase().contains("process"));
    }

    @Test