mirror of
https://github.com/Palm1r/QodeAssist.git
synced 2026-07-23 11:41:03 -04:00
81 lines
2.4 KiB
C++
81 lines
2.4 KiB
C++
// Copyright (C) 2024-2026 Petr Mironychev
|
|
// SPDX-License-Identifier: GPL-3.0-or-later
|
|
// Additional attribution terms under GPLv3 §7(b) apply — see LICENSE
|
|
|
|
#pragma once
|
|
|
|
#include "MessagePart.hpp"
|
|
|
|
#include <QAbstractListModel>
|
|
#include <QtQmlIntegration>
|
|
|
|
#include "session/HistoryProjection.hpp"
|
|
|
|
namespace QodeAssist::Chat {
|
|
|
|
class ChatModel : public QAbstractListModel
|
|
{
|
|
Q_OBJECT
|
|
Q_PROPERTY(int sessionPromptTokens READ sessionPromptTokens NOTIFY sessionUsageChanged FINAL)
|
|
Q_PROPERTY(int sessionCompletionTokens READ sessionCompletionTokens NOTIFY sessionUsageChanged FINAL)
|
|
Q_PROPERTY(int sessionCachedPromptTokens READ sessionCachedPromptTokens NOTIFY sessionUsageChanged FINAL)
|
|
Q_PROPERTY(int sessionTotalTokens READ sessionTotalTokens NOTIFY sessionUsageChanged FINAL)
|
|
QML_ELEMENT
|
|
|
|
public:
|
|
enum ChatRole { System, User, Assistant, Tool, FileEdit, Thinking, Permission, Plan };
|
|
Q_ENUM(ChatRole)
|
|
|
|
enum Roles {
|
|
RoleType = Qt::UserRole,
|
|
Content,
|
|
Attachments,
|
|
IsRedacted,
|
|
Images,
|
|
PromptTokens,
|
|
CompletionTokens,
|
|
CachedPromptTokens,
|
|
ReasoningTokens,
|
|
TotalTokens,
|
|
ToolKind,
|
|
ToolStatus,
|
|
ToolDetails,
|
|
ToolName,
|
|
ToolResult
|
|
};
|
|
Q_ENUM(Roles)
|
|
|
|
explicit ChatModel(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;
|
|
|
|
void resetMessages(const QList<Session::MessageRow> &rows);
|
|
void appendMessages(const QList<Session::MessageRow> &rows);
|
|
void updateMessage(int index, const Session::MessageRow &row);
|
|
void removeMessages(int first, int count);
|
|
|
|
Q_INVOKABLE QList<MessagePart> processMessageContent(const QString &content) const;
|
|
Q_INVOKABLE QVariantList userMessagePreviews(int maxLength = 80) const;
|
|
|
|
int sessionPromptTokens() const;
|
|
int sessionCompletionTokens() const;
|
|
int sessionCachedPromptTokens() const;
|
|
int sessionTotalTokens() const;
|
|
|
|
void setChatFilePath(const QString &filePath);
|
|
QString chatFilePath() const;
|
|
|
|
signals:
|
|
void modelReseted();
|
|
void sessionUsageChanged();
|
|
|
|
private:
|
|
QList<Session::MessageRow> m_messages;
|
|
QString m_chatFilePath;
|
|
};
|
|
|
|
} // namespace QodeAssist::Chat
|
|
Q_DECLARE_METATYPE(QodeAssist::Chat::MessagePart)
|