mirror of
https://github.com/Palm1r/QodeAssist.git
synced 2025-05-28 03:10:28 -04:00
400 lines
12 KiB
C++
400 lines
12 KiB
C++
/*
|
|
* Copyright (C) 2024 Petr Mironychev
|
|
*
|
|
* This file is part of QodeAssist.
|
|
*
|
|
* QodeAssist is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* QodeAssist is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with QodeAssist. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "ChatRootView.hpp"
|
|
|
|
#include <QClipboard>
|
|
#include <QFileDialog>
|
|
#include <QMessageBox>
|
|
|
|
#include <coreplugin/icore.h>
|
|
#include <projectexplorer/project.h>
|
|
#include <projectexplorer/projectexplorer.h>
|
|
#include <projectexplorer/projectmanager.h>
|
|
#include <utils/theme/theme.h>
|
|
#include <utils/utilsicons.h>
|
|
|
|
#include "ChatAssistantSettings.hpp"
|
|
#include "ChatSerializer.hpp"
|
|
#include "GeneralSettings.hpp"
|
|
#include "Logger.hpp"
|
|
#include "ProjectSettings.hpp"
|
|
#include "context/TokenUtils.hpp"
|
|
#include "context/ContextManager.hpp"
|
|
|
|
namespace QodeAssist::Chat {
|
|
|
|
ChatRootView::ChatRootView(QQuickItem *parent)
|
|
: QQuickItem(parent)
|
|
, m_chatModel(new ChatModel(this))
|
|
, m_clientInterface(new ClientInterface(m_chatModel, this))
|
|
{
|
|
auto &settings = Settings::generalSettings();
|
|
|
|
connect(&settings.caModel,
|
|
&Utils::BaseAspect::changed,
|
|
this,
|
|
&ChatRootView::currentTemplateChanged);
|
|
|
|
connect(&Settings::chatAssistantSettings().sharingCurrentFile,
|
|
&Utils::BaseAspect::changed,
|
|
this,
|
|
&ChatRootView::isSharingCurrentFileChanged);
|
|
|
|
connect(
|
|
m_clientInterface,
|
|
&ClientInterface::messageReceivedCompletely,
|
|
this,
|
|
&ChatRootView::autosave);
|
|
|
|
connect(
|
|
m_clientInterface,
|
|
&ClientInterface::messageReceivedCompletely,
|
|
this,
|
|
&ChatRootView::updateInputTokensCount);
|
|
|
|
connect(m_chatModel, &ChatModel::modelReseted, [this]() { m_recentFilePath = QString(); });
|
|
connect(this, &ChatRootView::attachmentFilesChanged, &ChatRootView::updateInputTokensCount);
|
|
connect(this, &ChatRootView::linkedFilesChanged, &ChatRootView::updateInputTokensCount);
|
|
connect(&Settings::chatAssistantSettings().useSystemPrompt, &Utils::BaseAspect::changed,
|
|
this, &ChatRootView::updateInputTokensCount);
|
|
connect(&Settings::chatAssistantSettings().systemPrompt, &Utils::BaseAspect::changed,
|
|
this, &ChatRootView::updateInputTokensCount);
|
|
updateInputTokensCount();
|
|
}
|
|
|
|
ChatModel *ChatRootView::chatModel() const
|
|
{
|
|
return m_chatModel;
|
|
}
|
|
|
|
void ChatRootView::sendMessage(const QString &message, bool sharingCurrentFile)
|
|
{
|
|
if (m_inputTokensCount > m_chatModel->tokensThreshold()) {
|
|
QMessageBox::StandardButton reply = QMessageBox::question(
|
|
Core::ICore::dialogParent(),
|
|
tr("Token Limit Exceeded"),
|
|
tr("The chat history has exceeded the token limit.\n"
|
|
"Would you like to create new chat?"),
|
|
QMessageBox::Yes | QMessageBox::No);
|
|
|
|
if (reply == QMessageBox::Yes) {
|
|
autosave();
|
|
m_chatModel->clear();
|
|
m_recentFilePath = QString{};
|
|
return;
|
|
}
|
|
}
|
|
|
|
m_clientInterface->sendMessage(message, m_attachmentFiles, m_linkedFiles, sharingCurrentFile);
|
|
clearAttachmentFiles();
|
|
}
|
|
|
|
void ChatRootView::copyToClipboard(const QString &text)
|
|
{
|
|
QGuiApplication::clipboard()->setText(text);
|
|
}
|
|
|
|
void ChatRootView::cancelRequest()
|
|
{
|
|
m_clientInterface->cancelRequest();
|
|
}
|
|
|
|
void ChatRootView::clearAttachmentFiles()
|
|
{
|
|
if (!m_attachmentFiles.isEmpty()) {
|
|
m_attachmentFiles.clear();
|
|
emit attachmentFilesChanged();
|
|
}
|
|
}
|
|
|
|
void ChatRootView::clearLinkedFiles()
|
|
{
|
|
if (!m_linkedFiles.isEmpty()) {
|
|
m_linkedFiles.clear();
|
|
emit linkedFilesChanged();
|
|
}
|
|
}
|
|
|
|
QString ChatRootView::getChatsHistoryDir() const
|
|
{
|
|
QString path;
|
|
|
|
if (auto project = ProjectExplorer::ProjectManager::startupProject()) {
|
|
Settings::ProjectSettings projectSettings(project);
|
|
path = projectSettings.chatHistoryPath().toString();
|
|
} else {
|
|
path = QString("%1/qodeassist/chat_history").arg(Core::ICore::userResourcePath().toString());
|
|
}
|
|
|
|
QDir dir(path);
|
|
if (!dir.exists() && !dir.mkpath(".")) {
|
|
LOG_MESSAGE(QString("Failed to create directory: %1").arg(path));
|
|
return QString();
|
|
}
|
|
|
|
return path;
|
|
}
|
|
|
|
QString ChatRootView::currentTemplate() const
|
|
{
|
|
auto &settings = Settings::generalSettings();
|
|
return settings.caModel();
|
|
}
|
|
|
|
bool ChatRootView::isSharingCurrentFile() const
|
|
{
|
|
return Settings::chatAssistantSettings().sharingCurrentFile();
|
|
}
|
|
|
|
void ChatRootView::saveHistory(const QString &filePath)
|
|
{
|
|
auto result = ChatSerializer::saveToFile(m_chatModel, filePath);
|
|
if (!result.success) {
|
|
LOG_MESSAGE(QString("Failed to save chat history: %1").arg(result.errorMessage));
|
|
}
|
|
}
|
|
|
|
void ChatRootView::loadHistory(const QString &filePath)
|
|
{
|
|
auto result = ChatSerializer::loadFromFile(m_chatModel, filePath);
|
|
if (!result.success) {
|
|
LOG_MESSAGE(QString("Failed to load chat history: %1").arg(result.errorMessage));
|
|
} else {
|
|
m_recentFilePath = filePath;
|
|
}
|
|
updateInputTokensCount();
|
|
}
|
|
|
|
void ChatRootView::showSaveDialog()
|
|
{
|
|
QString initialDir = getChatsHistoryDir();
|
|
|
|
QFileDialog *dialog = new QFileDialog(nullptr, tr("Save Chat History"));
|
|
dialog->setAcceptMode(QFileDialog::AcceptSave);
|
|
dialog->setFileMode(QFileDialog::AnyFile);
|
|
dialog->setNameFilter(tr("JSON files (*.json)"));
|
|
dialog->setDefaultSuffix("json");
|
|
if (!initialDir.isEmpty()) {
|
|
dialog->setDirectory(initialDir);
|
|
dialog->selectFile(getSuggestedFileName() + ".json");
|
|
}
|
|
|
|
connect(dialog, &QFileDialog::finished, this, [this, dialog](int result) {
|
|
if (result == QFileDialog::Accepted) {
|
|
QStringList files = dialog->selectedFiles();
|
|
if (!files.isEmpty()) {
|
|
saveHistory(files.first());
|
|
}
|
|
}
|
|
dialog->deleteLater();
|
|
});
|
|
|
|
dialog->open();
|
|
}
|
|
|
|
void ChatRootView::showLoadDialog()
|
|
{
|
|
QString initialDir = getChatsHistoryDir();
|
|
|
|
QFileDialog *dialog = new QFileDialog(nullptr, tr("Load Chat History"));
|
|
dialog->setAcceptMode(QFileDialog::AcceptOpen);
|
|
dialog->setFileMode(QFileDialog::ExistingFile);
|
|
dialog->setNameFilter(tr("JSON files (*.json)"));
|
|
if (!initialDir.isEmpty()) {
|
|
dialog->setDirectory(initialDir);
|
|
}
|
|
|
|
connect(dialog, &QFileDialog::finished, this, [this, dialog](int result) {
|
|
if (result == QFileDialog::Accepted) {
|
|
QStringList files = dialog->selectedFiles();
|
|
if (!files.isEmpty()) {
|
|
loadHistory(files.first());
|
|
}
|
|
}
|
|
dialog->deleteLater();
|
|
});
|
|
|
|
dialog->open();
|
|
}
|
|
|
|
QString ChatRootView::getSuggestedFileName() const
|
|
{
|
|
QStringList parts;
|
|
|
|
if (m_chatModel->rowCount() > 0) {
|
|
QString firstMessage
|
|
= m_chatModel->data(m_chatModel->index(0), ChatModel::Content).toString();
|
|
QString shortMessage = firstMessage.split('\n').first().simplified().left(30);
|
|
shortMessage.replace(QRegularExpression("[^a-zA-Z0-9_-]"), "_");
|
|
parts << shortMessage;
|
|
}
|
|
|
|
parts << QDateTime::currentDateTime().toString("yyyy-MM-dd_HH-mm");
|
|
|
|
return parts.join("_");
|
|
}
|
|
|
|
void ChatRootView::autosave()
|
|
{
|
|
if (m_chatModel->rowCount() == 0 || !Settings::chatAssistantSettings().autosave()) {
|
|
return;
|
|
}
|
|
|
|
QString filePath = getAutosaveFilePath();
|
|
if (!filePath.isEmpty()) {
|
|
ChatSerializer::saveToFile(m_chatModel, filePath);
|
|
m_recentFilePath = filePath;
|
|
}
|
|
}
|
|
|
|
QString ChatRootView::getAutosaveFilePath() const
|
|
{
|
|
if (!m_recentFilePath.isEmpty()) {
|
|
return m_recentFilePath;
|
|
}
|
|
|
|
QString dir = getChatsHistoryDir();
|
|
if (dir.isEmpty()) {
|
|
return QString();
|
|
}
|
|
|
|
return QDir(dir).filePath(getSuggestedFileName() + ".json");
|
|
}
|
|
|
|
QStringList ChatRootView::attachmentFiles() const
|
|
{
|
|
return m_attachmentFiles;
|
|
}
|
|
|
|
QStringList ChatRootView::linkedFiles() const
|
|
{
|
|
return m_linkedFiles;
|
|
}
|
|
|
|
void ChatRootView::showAttachFilesDialog()
|
|
{
|
|
QFileDialog dialog(nullptr, tr("Select Files to Attach"));
|
|
dialog.setFileMode(QFileDialog::ExistingFiles);
|
|
|
|
if (auto project = ProjectExplorer::ProjectManager::startupProject()) {
|
|
dialog.setDirectory(project->projectDirectory().toString());
|
|
}
|
|
|
|
if (dialog.exec() == QDialog::Accepted) {
|
|
QStringList newFilePaths = dialog.selectedFiles();
|
|
if (!newFilePaths.isEmpty()) {
|
|
bool filesAdded = false;
|
|
for (const QString &filePath : newFilePaths) {
|
|
if (!m_attachmentFiles.contains(filePath)) {
|
|
m_attachmentFiles.append(filePath);
|
|
filesAdded = true;
|
|
}
|
|
}
|
|
if (filesAdded) {
|
|
emit attachmentFilesChanged();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
void ChatRootView::removeFileFromAttachList(int index)
|
|
{
|
|
if (index >= 0 && index < m_attachmentFiles.size()) {
|
|
m_attachmentFiles.removeAt(index);
|
|
emit attachmentFilesChanged();
|
|
}
|
|
}
|
|
|
|
void ChatRootView::showLinkFilesDialog()
|
|
{
|
|
QFileDialog dialog(nullptr, tr("Select Files to Attach"));
|
|
dialog.setFileMode(QFileDialog::ExistingFiles);
|
|
|
|
if (auto project = ProjectExplorer::ProjectManager::startupProject()) {
|
|
dialog.setDirectory(project->projectDirectory().toString());
|
|
}
|
|
|
|
if (dialog.exec() == QDialog::Accepted) {
|
|
QStringList newFilePaths = dialog.selectedFiles();
|
|
if (!newFilePaths.isEmpty()) {
|
|
bool filesAdded = false;
|
|
for (const QString &filePath : newFilePaths) {
|
|
if (!m_linkedFiles.contains(filePath)) {
|
|
m_linkedFiles.append(filePath);
|
|
filesAdded = true;
|
|
}
|
|
}
|
|
if (filesAdded) {
|
|
emit linkedFilesChanged();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
void ChatRootView::removeFileFromLinkList(int index)
|
|
{
|
|
if (index >= 0 && index < m_linkedFiles.size()) {
|
|
m_linkedFiles.removeAt(index);
|
|
emit linkedFilesChanged();
|
|
}
|
|
}
|
|
|
|
void ChatRootView::calculateMessageTokensCount(const QString &message)
|
|
{
|
|
m_messageTokensCount = Context::TokenUtils::estimateTokens(message);
|
|
updateInputTokensCount();
|
|
}
|
|
|
|
void ChatRootView::updateInputTokensCount()
|
|
{
|
|
int inputTokens = m_messageTokensCount;
|
|
auto& settings = Settings::chatAssistantSettings();
|
|
|
|
if (settings.useSystemPrompt()) {
|
|
inputTokens += Context::TokenUtils::estimateTokens(settings.systemPrompt());
|
|
}
|
|
|
|
if (!m_attachmentFiles.isEmpty()) {
|
|
auto attachFiles = Context::ContextManager::instance().getContentFiles(m_attachmentFiles);
|
|
inputTokens += Context::TokenUtils::estimateFilesTokens(attachFiles);
|
|
}
|
|
|
|
if (!m_linkedFiles.isEmpty()) {
|
|
auto linkFiles = Context::ContextManager::instance().getContentFiles(m_linkedFiles);
|
|
inputTokens += Context::TokenUtils::estimateFilesTokens(linkFiles);
|
|
}
|
|
|
|
const auto& history = m_chatModel->getChatHistory();
|
|
for (const auto& message : history) {
|
|
inputTokens += Context::TokenUtils::estimateTokens(message.content);
|
|
inputTokens += 4; // + role
|
|
}
|
|
|
|
m_inputTokensCount = inputTokens;
|
|
emit inputTokensCountChanged();
|
|
}
|
|
|
|
int ChatRootView::inputTokensCount() const
|
|
{
|
|
return m_inputTokensCount;
|
|
}
|
|
|
|
} // namespace QodeAssist::Chat
|