feat(core): unify agent enabled and disabled flags (#17127)

This commit is contained in:
Sandy Tao
2026-01-20 13:00:43 -08:00
committed by GitHub
parent f42b4c80ac
commit f0f705d3ca
8 changed files with 25 additions and 39 deletions
@@ -2117,10 +2117,6 @@ export const SETTINGS_SCHEMA_DEFINITIONS: Record<
type: 'boolean', type: 'boolean',
description: 'Whether to enable the agent.', description: 'Whether to enable the agent.',
}, },
disabled: {
type: 'boolean',
description: 'Whether to disable the agent.',
},
}, },
}, },
CustomTheme: { CustomTheme: {
@@ -149,7 +149,7 @@ describe('agentsCommand', () => {
}); });
// Add agent to disabled overrides so validation passes // Add agent to disabled overrides so validation passes
mockContext.services.settings.merged.agents.overrides['test-agent'] = { mockContext.services.settings.merged.agents.overrides['test-agent'] = {
disabled: true, enabled: false,
}; };
vi.mocked(enableAgent).mockReturnValue({ vi.mocked(enableAgent).mockReturnValue({
@@ -264,7 +264,7 @@ describe('agentsCommand', () => {
it('should show info message if agent is already disabled', async () => { it('should show info message if agent is already disabled', async () => {
mockConfig.getAgentRegistry().getAllAgentNames.mockReturnValue([]); mockConfig.getAgentRegistry().getAllAgentNames.mockReturnValue([]);
mockContext.services.settings.merged.agents.overrides['test-agent'] = { mockContext.services.settings.merged.agents.overrides['test-agent'] = {
disabled: true, enabled: false,
}; };
const disableCommand = agentsCommand.subCommands?.find( const disableCommand = agentsCommand.subCommands?.find(
@@ -85,10 +85,10 @@ async function enableAction(
const allAgents = agentRegistry.getAllAgentNames(); const allAgents = agentRegistry.getAllAgentNames();
const overrides = settings.merged.agents.overrides; const overrides = settings.merged.agents.overrides;
const disabledAgents = Object.keys(overrides).filter( const disabledAgents = Object.keys(overrides).filter(
(name) => overrides[name]?.disabled === true, (name) => overrides[name]?.enabled === false,
); );
if (allAgents.includes(agentName)) { if (allAgents.includes(agentName) && !disabledAgents.includes(agentName)) {
return { return {
type: 'message', type: 'message',
messageType: 'info', messageType: 'info',
@@ -96,7 +96,7 @@ async function enableAction(
}; };
} }
if (!disabledAgents.includes(agentName)) { if (!disabledAgents.includes(agentName) && !allAgents.includes(agentName)) {
return { return {
type: 'message', type: 'message',
messageType: 'error', messageType: 'error',
@@ -155,7 +155,7 @@ async function disableAction(
const allAgents = agentRegistry.getAllAgentNames(); const allAgents = agentRegistry.getAllAgentNames();
const overrides = settings.merged.agents.overrides; const overrides = settings.merged.agents.overrides;
const disabledAgents = Object.keys(overrides).filter( const disabledAgents = Object.keys(overrides).filter(
(name) => overrides[name]?.disabled === true, (name) => overrides[name]?.enabled === false,
); );
if (disabledAgents.includes(agentName)) { if (disabledAgents.includes(agentName)) {
@@ -206,7 +206,7 @@ function completeAgentsToEnable(context: CommandContext, partialArg: string) {
const overrides = settings.merged.agents.overrides; const overrides = settings.merged.agents.overrides;
const disabledAgents = Object.entries(overrides) const disabledAgents = Object.entries(overrides)
.filter(([_, override]) => override?.disabled === true) .filter(([_, override]) => override?.enabled === false)
.map(([name]) => name); .map(([name]) => name);
return disabledAgents.filter((name) => name.startsWith(partialArg)); return disabledAgents.filter((name) => name.startsWith(partialArg));
+11 -12
View File
@@ -29,8 +29,8 @@ export interface AgentActionResult {
} }
/** /**
* Enables an agent by ensuring it is not disabled in any writable scope (User and Workspace). * Enables an agent by ensuring it is enabled in any writable scope (User and Workspace).
* It sets `agents.overrides.<agentName>.disabled` to `false` if it was found to be `true`. * It sets `agents.overrides.<agentName>.enabled` to `true`.
*/ */
export function enableAgent( export function enableAgent(
settings: LoadedSettings, settings: LoadedSettings,
@@ -45,9 +45,9 @@ export function enableAgent(
const scopePath = settings.forScope(scope).path; const scopePath = settings.forScope(scope).path;
const agentOverrides = const agentOverrides =
settings.forScope(scope).settings.agents?.overrides; settings.forScope(scope).settings.agents?.overrides;
const isDisabled = agentOverrides?.[agentName]?.disabled === true; const isEnabled = agentOverrides?.[agentName]?.enabled === true;
if (isDisabled) { if (!isEnabled) {
foundInDisabledScopes.push({ scope, path: scopePath }); foundInDisabledScopes.push({ scope, path: scopePath });
} else { } else {
alreadyEnabledScopes.push({ scope, path: scopePath }); alreadyEnabledScopes.push({ scope, path: scopePath });
@@ -68,9 +68,8 @@ export function enableAgent(
const modifiedScopes: ModifiedScope[] = []; const modifiedScopes: ModifiedScope[] = [];
for (const { scope, path } of foundInDisabledScopes) { for (const { scope, path } of foundInDisabledScopes) {
if (isLoadableSettingScope(scope)) { if (isLoadableSettingScope(scope)) {
// Explicitly enable it to override any lower-precedence disables, or just clear the disable. // Explicitly enable it.
// Setting to false ensures it is enabled. settings.setValue(scope, `agents.overrides.${agentName}.enabled`, true);
settings.setValue(scope, `agents.overrides.${agentName}.disabled`, false);
modifiedScopes.push({ scope, path }); modifiedScopes.push({ scope, path });
} }
} }
@@ -85,7 +84,7 @@ export function enableAgent(
} }
/** /**
* Disables an agent by setting `agents.overrides.<agentName>.disabled` to `true` in the specified scope. * Disables an agent by setting `agents.overrides.<agentName>.enabled` to `false` in the specified scope.
*/ */
export function disableAgent( export function disableAgent(
settings: LoadedSettings, settings: LoadedSettings,
@@ -105,9 +104,9 @@ export function disableAgent(
const scopePath = settings.forScope(scope).path; const scopePath = settings.forScope(scope).path;
const agentOverrides = settings.forScope(scope).settings.agents?.overrides; const agentOverrides = settings.forScope(scope).settings.agents?.overrides;
const isDisabled = agentOverrides?.[agentName]?.disabled === true; const isEnabled = agentOverrides?.[agentName]?.enabled !== false;
if (isDisabled) { if (!isEnabled) {
return { return {
status: 'no-op', status: 'no-op',
agentName, agentName,
@@ -127,7 +126,7 @@ export function disableAgent(
if (isLoadableSettingScope(otherScope)) { if (isLoadableSettingScope(otherScope)) {
const otherOverrides = const otherOverrides =
settings.forScope(otherScope).settings.agents?.overrides; settings.forScope(otherScope).settings.agents?.overrides;
if (otherOverrides?.[agentName]?.disabled === true) { if (otherOverrides?.[agentName]?.enabled === false) {
alreadyDisabledInOther.push({ alreadyDisabledInOther.push({
scope: otherScope, scope: otherScope,
path: settings.forScope(otherScope).path, path: settings.forScope(otherScope).path,
@@ -135,7 +134,7 @@ export function disableAgent(
} }
} }
settings.setValue(scope, `agents.overrides.${agentName}.disabled`, true); settings.setValue(scope, `agents.overrides.${agentName}.enabled`, false);
return { return {
status: 'success', status: 'success',
+3 -3
View File
@@ -309,7 +309,7 @@ describe('AgentRegistry', () => {
const config = makeMockedConfig({ const config = makeMockedConfig({
agents: { agents: {
overrides: { overrides: {
generalist: { enabled: true, disabled: true }, generalist: { enabled: false },
}, },
}, },
}); });
@@ -704,7 +704,7 @@ describe('AgentRegistry', () => {
const config = makeMockedConfig({ const config = makeMockedConfig({
agents: { agents: {
overrides: { overrides: {
MockAgent: { disabled: true }, MockAgent: { enabled: false },
}, },
}, },
}); });
@@ -719,7 +719,7 @@ describe('AgentRegistry', () => {
const config = makeMockedConfig({ const config = makeMockedConfig({
agents: { agents: {
overrides: { overrides: {
RemoteAgent: { disabled: true }, RemoteAgent: { enabled: false },
}, },
}, },
}); });
+3 -7
View File
@@ -152,7 +152,7 @@ export class AgentRegistry {
// Only register the agent if it's enabled in the settings and not explicitly disabled via overrides. // Only register the agent if it's enabled in the settings and not explicitly disabled via overrides.
if ( if (
investigatorSettings?.enabled && investigatorSettings?.enabled &&
!agentsOverrides[CodebaseInvestigatorAgent.name]?.disabled agentsOverrides[CodebaseInvestigatorAgent.name]?.enabled !== false
) { ) {
let model; let model;
const settingsModel = investigatorSettings.model; const settingsModel = investigatorSettings.model;
@@ -200,7 +200,7 @@ export class AgentRegistry {
// Register the CLI help agent if it's explicitly enabled and not explicitly disabled via overrides. // Register the CLI help agent if it's explicitly enabled and not explicitly disabled via overrides.
if ( if (
cliHelpSettings.enabled && cliHelpSettings.enabled &&
!agentsOverrides[CliHelpAgent.name]?.disabled agentsOverrides[CliHelpAgent.name]?.enabled !== false
) { ) {
this.registerLocalAgent(CliHelpAgent(this.config)); this.registerLocalAgent(CliHelpAgent(this.config));
} }
@@ -280,13 +280,9 @@ export class AgentRegistry {
const isExperimental = definition.experimental === true; const isExperimental = definition.experimental === true;
let isEnabled = !isExperimental; let isEnabled = !isExperimental;
if (overrides) { if (overrides && overrides.enabled !== undefined) {
if (overrides.disabled !== undefined) {
isEnabled = !overrides.disabled;
} else if (overrides.enabled !== undefined) {
isEnabled = overrides.enabled; isEnabled = overrides.enabled;
} }
}
return isEnabled; return isEnabled;
} }
-1
View File
@@ -177,7 +177,6 @@ export interface AgentRunConfig {
export interface AgentOverride { export interface AgentOverride {
modelConfig?: ModelConfig; modelConfig?: ModelConfig;
runConfig?: AgentRunConfig; runConfig?: AgentRunConfig;
disabled?: boolean;
enabled?: boolean; enabled?: boolean;
} }
-4
View File
@@ -1946,10 +1946,6 @@
"enabled": { "enabled": {
"type": "boolean", "type": "boolean",
"description": "Whether to enable the agent." "description": "Whether to enable the agent."
},
"disabled": {
"type": "boolean",
"description": "Whether to disable the agent."
} }
} }
}, },