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

         */
        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;

        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 {

            return null;
        }
        session.touch();
        if (logger.isDebugEnabled()) {
            logger.debug("Session retrieved. sessionId={}, messageCount={}", sessionId, session.getMessages().size());
        }
        return session;
    }

    /**
     * Gets or creates a session.
     *
     * @param sessionId the session ID (can be null to create a new session)

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

            final List<LlmMessage> history = new ArrayList<>();
            for (final ChatMessage msg : session.getMessages()) {
                if (msg.isUser()) {
                    history.add(LlmMessage.user(msg.getContent()));
                } else if (msg.isAssistant()) {
                    final String content =

        assertEquals(originalException.getMessage(), deserializedException.getMessage());
        assertNotNull(deserializedException.getCause());
        assertEquals(originalException.getCause().getMessage(), deserializedException.getCause().getMessage());
    }

    @Test
    public void test_getMessage() {
        // Test getMessage() method
        String message = "Custom SSO error message";

        assertEquals("SSO process failed", topLevel.getMessage());
        assertEquals(level1, topLevel.getCause());
        assertEquals("Level 1: Authentication service error", topLevel.getCause().getMessage());
        assertEquals(level2, topLevel.getCause().getCause());
        assertEquals("Level 2: User lookup failed", topLevel.getCause().getCause().getMessage());

            // If exception thrown, validation is working
            assertTrue(e.getMessage().contains("Invalid negotiate context data length") || e.getMessage().contains("Buffer too small")
                    || e.getMessage().contains("negotiate context"), "Validation detected issue: " + e.getMessage());
        }
    }

    /**
     * Test negotiate context buffer overflow protection.
     */

        assertNotSame(exception2, exception3);
        assertNotSame(exception1, exception3);

        assertEquals("First instance", exception1.getMessage());
        assertEquals("Second instance", exception2.getMessage());
        assertEquals("Third instance", exception3.getMessage());
    }

    @Test
    public void test_specialCharactersInMessage() {
        // Test with special characters in message