zach
1fe8a3a959
C++ / fmt (pull_request) Failing after 6s
C++ / build (pull_request) Failing after 10s
C++ / clang-tidy (pull_request) Failing after 9s
JS/TS / fmt (pull_request) Successful in 10s
JS/TS / lint (pull_request) Successful in 8s
Python / static (pull_request) Successful in 30s
Rust / build (pull_request) Successful in 49s
Python / verify (pull_request) Successful in 1m35s
Rust / fmt (pull_request) Successful in 36s
Rust / clippy (pull_request) Successful in 54s
101 lines
2.7 KiB
C++
101 lines
2.7 KiB
C++
#pragma once
|
|
|
|
#include "attachment.hpp"
|
|
#include "generation.hpp"
|
|
|
|
#include <QList>
|
|
#include <QObject>
|
|
#include <QString>
|
|
#include <QtQml>
|
|
|
|
namespace ZShell::llm {
|
|
|
|
class ChatSession;
|
|
|
|
class ChatMessage : public QObject {
|
|
Q_OBJECT
|
|
QML_ELEMENT
|
|
QML_UNCREATABLE("Chat messages are created by ChatMessageModel")
|
|
|
|
Q_PROPERTY(Role role READ role CONSTANT)
|
|
Q_PROPERTY(qint64 timestamp READ timestamp CONSTANT)
|
|
Q_PROPERTY(QString content READ content NOTIFY contentChanged)
|
|
Q_PROPERTY(
|
|
int generationCount READ generationCount NOTIFY generationsChanged)
|
|
Q_PROPERTY(
|
|
QList<ZShell::llm::ChatGeneration*> generations READ generations NOTIFY
|
|
generationsChanged)
|
|
Q_PROPERTY(
|
|
ZShell::llm::ChatGeneration* activeGeneration READ activeGeneration
|
|
NOTIFY activeGenerationChanged)
|
|
Q_PROPERTY(
|
|
int activeGenerationIndex READ activeGenerationIndex NOTIFY
|
|
activeGenerationChanged)
|
|
Q_PROPERTY(
|
|
QList<ZShell::llm::Attachment*> attachments READ attachments NOTIFY
|
|
attachmentsChanged)
|
|
|
|
public:
|
|
enum class Role : int { User = 0, Assistant };
|
|
Q_ENUM(Role)
|
|
|
|
ChatMessage(
|
|
ChatSession* session,
|
|
Role role,
|
|
qint64 timestamp,
|
|
QObject* parent = nullptr);
|
|
|
|
[[nodiscard]] ChatSession* session() const { return m_session; }
|
|
[[nodiscard]] Role role() const { return m_role; }
|
|
[[nodiscard]] qint64 timestamp() const { return m_timestamp; }
|
|
[[nodiscard]] QString content() const;
|
|
[[nodiscard]] int generationCount() const {
|
|
return static_cast<int>(m_generations.size());
|
|
}
|
|
[[nodiscard]] QList<ChatGeneration*> generations() const {
|
|
return m_generations;
|
|
}
|
|
[[nodiscard]] ChatGeneration* activeGeneration() const {
|
|
return generation(m_active);
|
|
}
|
|
[[nodiscard]] int activeGenerationIndex() const { return m_active; }
|
|
[[nodiscard]] QList<Attachment*> attachments() const {
|
|
return m_attachments;
|
|
}
|
|
[[nodiscard]] ChatGeneration* generation(int index) const {
|
|
if (index < 0 || index >= m_generations.size()) return nullptr;
|
|
return m_generations.at(index);
|
|
}
|
|
|
|
Q_INVOKABLE void setActiveGeneration(int index);
|
|
void setAttachments(QList<Attachment*> attachments);
|
|
Q_INVOKABLE void edit(const QString& newContent);
|
|
Q_INVOKABLE void retry();
|
|
Q_INVOKABLE void generate();
|
|
|
|
ChatGeneration* addGeneration(qint64 timestamp);
|
|
ChatGeneration* appendGeneration(qint64 timestamp);
|
|
void removeGeneration(ChatGeneration* generation);
|
|
|
|
void setContent(const QString& value);
|
|
|
|
Q_SIGNALS:
|
|
void contentChanged();
|
|
void generationsChanged();
|
|
void activeGenerationChanged();
|
|
void attachmentsChanged();
|
|
|
|
private:
|
|
void setActiveInternal(int index);
|
|
|
|
ChatSession* m_session;
|
|
Role m_role;
|
|
qint64 m_timestamp;
|
|
QString m_content;
|
|
QList<ChatGeneration*> m_generations;
|
|
QList<Attachment*> m_attachments;
|
|
int m_active = -1;
|
|
};
|
|
|
|
} // namespace ZShell::llm
|