feat: Add agent roles (#287)

* feat: Add agent roles * doc: Add agent roles to docs
2026-04-01 02:12:45 -04:00 · 2025-12-04 19:41:30 +01:00
parent 498eb4d932
commit fc33bb60d0
22 changed files with 1713 additions and 293 deletions
--- a/settings/AgentRoleDialog.cpp
+++ b/settings/AgentRoleDialog.cpp
@ -0,0 +1,122 @@
+/* 
+ * Copyright (C) 2024-2025 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 "AgentRoleDialog.hpp"
+
+#include <QDialogButtonBox>
+#include <QFormLayout>
+#include <QLabel>
+#include <QLineEdit>
+#include <QPushButton>
+#include <QTextEdit>
+#include <QVBoxLayout>
+
+namespace QodeAssist::Settings {
+
+AgentRoleDialog::AgentRoleDialog(QWidget *parent)
+    : QDialog(parent)
+    , m_editMode(false)
+{
+    setWindowTitle(tr("Add Agent Role"));
+    setupUI();
+}
+
+AgentRoleDialog::AgentRoleDialog(const AgentRole &role, bool editMode, QWidget *parent)
+    : QDialog(parent)
+    , m_editMode(editMode)
+{
+    setWindowTitle(editMode ? tr("Edit Agent Role") : tr("Duplicate Agent Role"));
+    setupUI();
+    setRole(role);
+}
+
+void AgentRoleDialog::setupUI()
+{
+    auto *mainLayout = new QVBoxLayout(this);
+    auto *formLayout = new QFormLayout();
+
+    m_nameEdit = new QLineEdit(this);
+    m_nameEdit->setPlaceholderText(tr("e.g., Developer, Code Reviewer"));
+    formLayout->addRow(tr("Name:"), m_nameEdit);
+
+    m_idEdit = new QLineEdit(this);
+    m_idEdit->setPlaceholderText(tr("e.g., developer, code_reviewer"));
+    formLayout->addRow(tr("ID:"), m_idEdit);
+
+    m_descriptionEdit = new QTextEdit(this);
+    m_descriptionEdit->setPlaceholderText(tr("Brief description of this role..."));
+    m_descriptionEdit->setMaximumHeight(80);
+    formLayout->addRow(tr("Description:"), m_descriptionEdit);
+
+    mainLayout->addLayout(formLayout);
+
+    auto *promptLabel = new QLabel(tr("System Prompt:"), this);
+    mainLayout->addWidget(promptLabel);
+
+    m_systemPromptEdit = new QTextEdit(this);
+    m_systemPromptEdit->setPlaceholderText(
+        tr("You are an expert in...\n\nYour role is to:\n- Task 1\n- Task 2\n- Task 3"));
+    mainLayout->addWidget(m_systemPromptEdit);
+
+    m_buttonBox = new QDialogButtonBox(QDialogButtonBox::Ok | QDialogButtonBox::Cancel, this);
+    mainLayout->addWidget(m_buttonBox);
+
+    connect(m_buttonBox, &QDialogButtonBox::accepted, this, &AgentRoleDialog::accept);
+    connect(m_buttonBox, &QDialogButtonBox::rejected, this, &AgentRoleDialog::reject);
+    connect(m_nameEdit, &QLineEdit::textChanged, this, &AgentRoleDialog::validateInput);
+    connect(m_idEdit, &QLineEdit::textChanged, this, &AgentRoleDialog::validateInput);
+    connect(m_systemPromptEdit, &QTextEdit::textChanged, this, &AgentRoleDialog::validateInput);
+
+    if (m_editMode) {
+        m_idEdit->setEnabled(false);
+        m_idEdit->setToolTip(tr("ID cannot be changed for existing roles"));
+    }
+
+    setMinimumSize(600, 500);
+    validateInput();
+}
+
+void AgentRoleDialog::validateInput()
+{
+    bool valid = !m_nameEdit->text().trimmed().isEmpty()
+                 && !m_idEdit->text().trimmed().isEmpty()
+                 && !m_systemPromptEdit->toPlainText().trimmed().isEmpty();
+
+    m_buttonBox->button(QDialogButtonBox::Ok)->setEnabled(valid);
+}
+
+AgentRole AgentRoleDialog::getRole() const
+{
+    return AgentRole{
+        m_idEdit->text().trimmed(),
+        m_nameEdit->text().trimmed(),
+        m_descriptionEdit->toPlainText().trimmed(),
+        m_systemPromptEdit->toPlainText().trimmed(),
+        false};
+}
+
+void AgentRoleDialog::setRole(const AgentRole &role)
+{
+    m_idEdit->setText(role.id);
+    m_nameEdit->setText(role.name);
+    m_descriptionEdit->setPlainText(role.description);
+    m_systemPromptEdit->setPlainText(role.systemPrompt);
+}
+
+} // namespace QodeAssist::Settings