mirror of
https://github.com/google-gemini/gemini-cli.git
synced 2026-05-12 21:03:05 -07:00
feat(core): unify agent enabled and disabled flags (#17127)
This commit is contained in:
@@ -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));
|
||||||
|
|||||||
@@ -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',
|
||||||
|
|||||||
@@ -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 },
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
});
|
});
|
||||||
|
|||||||
@@ -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;
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -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;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -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."
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
|||||||
Reference in New Issue
Block a user