mirror/QodeAssist
1
0
Fork 0
mirror of https://github.com/Palm1r/QodeAssist.git synced 2026-07-01 02:29:13 -04:00
Code Issues Packages Projects Releases Wiki Activity

feat: Add possibility to overwrite agent config tools enabling

Browse Source
This commit is contained in:
Petr Mironychev
2026-06-29 18:32:45 +02:00
parent 9cc57c602b
commit 080947c0dc
6 changed files with 188 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

70
sources/agents/AgentFactory.cpp
View File

@@ -89,6 +89,37 @@ void writeProviderOverride(const QString &name, const QString &providerInstance)
s->setValue(providerOverrideKey(name), providerInstance);
s->sync();
}
constexpr auto kToolsOverrideGroup = "QodeAssist/AgentToolsOverrides";
Utils::Key toolsOverrideKey(const QString &name)
{
return Utils::Key(
QStringLiteral("%1/%2").arg(QLatin1StringView(kToolsOverrideGroup), name).toUtf8());
}
std::optional<bool> readToolsOverride(const QString &name)
{
auto *s = Core::ICore::settings();
if (!s)
return std::nullopt;
const Utils::Key key = toolsOverrideKey(name);
if (!s->contains(key))
return std::nullopt;
return s->value(key).toBool();
}
void writeToolsOverride(const QString &name, std::optional<bool> enabled)
{
auto *s = Core::ICore::settings();
if (!s)
return;
if (!enabled.has_value())
s->remove(toolsOverrideKey(name));
else
s->setValue(toolsOverrideKey(name), *enabled);
s->sync();
}
} // namespace
namespace QodeAssist {
@@ -146,6 +177,11 @@ void AgentFactory::reload()
const QString overrideProvider = readProviderOverride(cfg.name);
if (!overrideProvider.isEmpty())
cfg.providerInstance = overrideProvider;
m_baseToolsByName.insert(cfg.name, cfg.enableTools);
const std::optional<bool> overrideTools = readToolsOverride(cfg.name);
if (overrideTools.has_value())
cfg.enableTools = *overrideTools;
}
emit agentsChanged();
}
@@ -289,6 +325,7 @@ void AgentFactory::clear()
m_indexByName.clear();
m_baseModelByName.clear();
m_baseProviderByName.clear();
m_baseToolsByName.clear();
}
Providers::ProviderInstanceFactory *AgentFactory::instanceFactory() const noexcept
@@ -361,4 +398,37 @@ void AgentFactory::clearAgentProviderOverride(const QString &name)
writeProviderOverride(name, QString());
}
bool AgentFactory::setAgentToolsOverride(
const QString &name, std::optional<bool> enabled, QString *error)
{
Q_ASSERT(thread() == QThread::currentThread());
const auto it = m_indexByName.constFind(name);
if (it == m_indexByName.constEnd()) {
if (error)
*error = QStringLiteral("Agent '%1' is not registered").arg(name);
return false;
}
AgentConfig &cfg = m_configs[it.value()];
const bool effective = enabled.has_value() ? *enabled
: m_baseToolsByName.value(name, cfg.enableTools);
if (cfg.enableTools == effective && readToolsOverride(name) == enabled)
return true;
writeToolsOverride(name, enabled);
cfg.enableTools = effective;
emit agentToolsChanged(name);
return true;
}
std::optional<bool> AgentFactory::agentToolsOverride(const QString &name) const
{
return readToolsOverride(name);
}
void AgentFactory::clearAgentToolsOverride(const QString &name)
{
writeToolsOverride(name, std::nullopt);
}
} // namespace QodeAssist