}

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

        final LlmChatRequest result = request.addAssistantMessage("Assistant response");

        assertSame(request, result);
        assertEquals(1, request.getMessages().size());
        assertEquals(LlmMessage.ROLE_ASSISTANT, request.getMessages().get(0).getRole());

        addMessage(ChatMessage.userMessage(content));
    }

    /**
     * Adds an assistant message to this session.
     *
     * @param content the message content
     */
    public void addAssistantMessage(final String content) {
        addMessage(ChatMessage.assistantMessage(content));
    }

    /**
     * Updates the last accessed timestamp to the current time.
     */
    public void touch() {

    }

    /**
     * Adds an assistant message to the request.
     *
     * @param content the message content
     * @return this request for method chaining
     */
    public LlmChatRequest addAssistantMessage(final String content) {
        messages.add(LlmMessage.assistant(content));
        return this;
    }

    /**
     * Gets the messages in this request.
     *
     * @return the list of messages

        session.addUserMessage("User 1");
        session.addAssistantMessage("Assistant 1");
        session.addUserMessage("User 2");
        session.addAssistantMessage("Assistant 2");
        session.addUserMessage("User 3");
        session.addAssistantMessage("Assistant 3");
        assertEquals(6, session.getMessageCount());

            }
        });

        final ChatSession session = new ChatSession();
        session.addUserMessage("Q1");
        session.addAssistantMessage("A1");
        session.addUserMessage("Q2");
        session.addAssistantMessage("A2");

        final List<LlmMessage> history = chatClient.testExtractHistory(session);
        assertEquals(4, history.size());