60 lines
1.2 KiB
C++
60 lines
1.2 KiB
C++
#ifndef CHATBLOCKMODEL_H
|
|
#define CHATBLOCKMODEL_H
|
|
|
|
#include <QAbstractListModel>
|
|
#include <QList>
|
|
#include <QString>
|
|
#include <QRegularExpression>
|
|
|
|
struct ChatBlock {
|
|
enum Type {
|
|
User,
|
|
Assistant,
|
|
Reasoning,
|
|
Call,
|
|
Tool,
|
|
Error,
|
|
Info,
|
|
Default
|
|
};
|
|
|
|
Type type = Default;
|
|
QString name;
|
|
QString content;
|
|
};
|
|
|
|
class ChatBlockModel : public QAbstractListModel
|
|
{
|
|
Q_OBJECT
|
|
Q_PROPERTY(int count READ rowCount NOTIFY countChanged)
|
|
|
|
public:
|
|
enum Roles {
|
|
BlockTypeRole = Qt::UserRole + 1,
|
|
BlockNameRole,
|
|
ContentRole,
|
|
};
|
|
|
|
explicit ChatBlockModel(QObject *parent = nullptr);
|
|
|
|
int rowCount(const QModelIndex &parent = QModelIndex()) const override;
|
|
QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override;
|
|
QHash<int, QByteArray> roleNames() const override;
|
|
|
|
Q_INVOKABLE void appendData(const QString &text);
|
|
Q_INVOKABLE void reset();
|
|
|
|
signals:
|
|
void countChanged();
|
|
|
|
private:
|
|
void processLine(const QString &line);
|
|
|
|
QList<ChatBlock> m_blocks;
|
|
QString m_lineBuffer;
|
|
|
|
static const QRegularExpression s_headerRe;
|
|
};
|
|
|
|
#endif // CHATBLOCKMODEL_H
|