mirror of
https://github.com/Palm1r/QodeAssist.git
synced 2026-06-30 01:59:11 -04:00
108 lines
2.5 KiB
C++
108 lines
2.5 KiB
C++
// Copyright (C) 2024-2026 Petr Mironychev
|
|
// SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
#include "ChatAgentController.hpp"
|
|
|
|
#include <QSettings>
|
|
|
|
#include <coreplugin/icore.h>
|
|
|
|
#include <AgentConfig.hpp>
|
|
#include <AgentFactory.hpp>
|
|
#include <sources/settings/PipelinesConfig.hpp>
|
|
|
|
namespace QodeAssist::Chat {
|
|
|
|
namespace {
|
|
const char kChatAgentKey[] = "QodeAssist.chatActiveAgent";
|
|
}
|
|
|
|
ChatAgentController::ChatAgentController(QObject *parent)
|
|
: QObject(parent)
|
|
{
|
|
if (auto *settings = Core::ICore::settings())
|
|
m_currentAgent = settings->value(kChatAgentKey).toString();
|
|
}
|
|
|
|
void ChatAgentController::setAgentFactory(AgentFactory *factory)
|
|
{
|
|
m_agentFactory = factory;
|
|
if (factory)
|
|
connect(
|
|
factory,
|
|
&AgentFactory::agentsChanged,
|
|
this,
|
|
&ChatAgentController::reload,
|
|
Qt::UniqueConnection);
|
|
reload();
|
|
}
|
|
|
|
QStringList ChatAgentController::availableAgents() const
|
|
{
|
|
return m_availableAgents;
|
|
}
|
|
|
|
QString ChatAgentController::currentAgent() const
|
|
{
|
|
return m_currentAgent;
|
|
}
|
|
|
|
void ChatAgentController::setCurrentAgent(const QString &name)
|
|
{
|
|
if (name == m_currentAgent || !m_availableAgents.contains(name))
|
|
return;
|
|
|
|
applyCurrentAgent(name);
|
|
}
|
|
|
|
void ChatAgentController::reload()
|
|
{
|
|
if (!m_agentFactory)
|
|
return;
|
|
|
|
const QStringList all = m_agentFactory->configNames();
|
|
const QStringList roster = Settings::PipelinesConfig::load().rosters.chatAssistant;
|
|
|
|
QStringList filtered;
|
|
for (const QString &name : roster) {
|
|
if (all.contains(name))
|
|
filtered.append(name);
|
|
}
|
|
|
|
if (filtered != m_availableAgents) {
|
|
m_availableAgents = filtered;
|
|
emit availableAgentsChanged();
|
|
}
|
|
ensureValidCurrent();
|
|
}
|
|
|
|
void ChatAgentController::ensureValidCurrent()
|
|
{
|
|
if (m_availableAgents.contains(m_currentAgent))
|
|
return;
|
|
|
|
const QString next = m_availableAgents.isEmpty() ? QString() : m_availableAgents.first();
|
|
if (next == m_currentAgent)
|
|
return;
|
|
|
|
applyCurrentAgent(next);
|
|
}
|
|
|
|
void ChatAgentController::applyCurrentAgent(const QString &name)
|
|
{
|
|
m_currentAgent = name;
|
|
if (auto *settings = Core::ICore::settings())
|
|
settings->setValue(kChatAgentKey, m_currentAgent);
|
|
emit currentAgentChanged();
|
|
}
|
|
|
|
bool ChatAgentController::currentSupportsTools() const
|
|
{
|
|
if (!m_agentFactory || m_currentAgent.isEmpty())
|
|
return false;
|
|
const AgentConfig *config = m_agentFactory->configByName(m_currentAgent);
|
|
return config && config->enableTools;
|
|
}
|
|
|
|
} // namespace QodeAssist::Chat
|