thread.h

Go to the documentation of this file.

#pragma once
 
#include <QAbstractListModel>
#include <QDateTime>
#include <QSortFilterProxyModel>
#include <QVector>
 
namespace llm_chat {
 
class Message;
 
class Thread : public QAbstractListModel {
  Q_OBJECT
  Q_PROPERTY(QList<Message *> messages READ messages CONSTANT FINAL)
  Q_PROPERTY(QDateTime createdAt READ createdAt CONSTANT FINAL)
 
 public:
  enum MessageRoles {
    TextRole = Qt::UserRole + 1,
    IsUserRole,
    ContextRole,
    FinishedRole
  };
 
  explicit Thread(QObject *parent = nullptr);
 
  [[nodiscard]] int rowCount(
      const QModelIndex &parent = QModelIndex()) const override;
  [[nodiscard]] QVariant data(const QModelIndex &index,
                              int role = Qt::DisplayRole) const override;
  [[nodiscard]] QHash<int, QByteArray> roleNames() const override;
  Q_INVOKABLE void addMessage(const QString &text, bool is_user,
                              const QVector<QVariant> &context = {},
                              const bool in_progress = false);
 
  [[nodiscard]] inline const QList<Message *> &messages() const {
    return m_Messages;
  }
  void clearMessages();
 
  [[nodiscard]] inline const QDateTime &createdAt() const {
    return m_CreatedAt;
  }
 
  void removeLatestMessage();
 
 private:
  QList<Message *> m_Messages;
  QDateTime m_CreatedAt;
};
 
}  // namespace llm_chat