index.tsx

return (
  <div key={`chatMessageContainer-${index}`} className={wrapperClassName}>
    <div key={`chatMessage-${index}`} className={className}>
      {icon}
      <div className="message-content">
        <p className="message-content-text">{message.message}</p>
        {message.sourceDocs && (
          <ModalDemo message={message}></ModalDemo>
        )}{' '}
      </div>
    </div>
  </div>
);