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.

        assertSame(request, result);
        assertEquals(1, request.getMessages().size());
        assertEquals(LlmMessage.ROLE_SYSTEM, request.getMessages().get(0).getRole());
        assertEquals("System prompt", request.getMessages().get(0).getContent());
    }

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

        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 {

            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)

            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 =

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

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