summaryrefslogtreecommitdiff
path: root/packages/core/src
diff options
context:
space:
mode:
authorJingboWang1997-1 <[email protected]>2025-06-11 14:32:23 -0700
committerGitHub <[email protected]>2025-06-11 14:32:23 -0700
commit6ecdecbdcca59d4daff7ec5921afc746e7d725ab (patch)
tree5d51fbb8b19eed975614d5109e5c1ca590cc5d9d /packages/core/src
parent122678cc09131ee57dca2714bc94b039c371729b (diff)
add excludeTools flag to settings.json config (#957)
Diffstat (limited to 'packages/core/src')
-rw-r--r--packages/core/src/config/config.ts19
1 files changed, 18 insertions, 1 deletions
diff --git a/packages/core/src/config/config.ts b/packages/core/src/config/config.ts
index 297178fd..b94585a5 100644
--- a/packages/core/src/config/config.ts
+++ b/packages/core/src/config/config.ts
@@ -65,6 +65,7 @@ export interface ConfigParameters {
question?: string;
fullContext?: boolean;
coreTools?: string[];
+ excludeTools?: string[];
toolDiscoveryCommand?: string;
toolCallCommand?: string;
mcpServerCommand?: string;
@@ -95,6 +96,7 @@ export class Config {
private readonly question: string | undefined;
private readonly fullContext: boolean;
private readonly coreTools: string[] | undefined;
+ private readonly excludeTools: string[] | undefined;
private readonly toolDiscoveryCommand: string | undefined;
private readonly toolCallCommand: string | undefined;
private readonly mcpServerCommand: string | undefined;
@@ -126,6 +128,7 @@ export class Config {
this.question = params.question;
this.fullContext = params.fullContext ?? false;
this.coreTools = params.coreTools;
+ this.excludeTools = params.excludeTools;
this.toolDiscoveryCommand = params.toolDiscoveryCommand;
this.toolCallCommand = params.toolCallCommand;
this.mcpServerCommand = params.mcpServerCommand;
@@ -210,6 +213,10 @@ export class Config {
return this.coreTools;
}
+ getExcludeTools(): string[] | undefined {
+ return this.excludeTools;
+ }
+
getToolDiscoveryCommand(): string | undefined {
return this.toolDiscoveryCommand;
}
@@ -360,12 +367,22 @@ export function createToolRegistry(config: Config): Promise<ToolRegistry> {
const tools = config.getCoreTools()
? new Set(config.getCoreTools())
: undefined;
+ const excludeTools = config.getExcludeTools()
+ ? new Set(config.getExcludeTools())
+ : undefined;
// helper to create & register core tools that are enabled
// eslint-disable-next-line @typescript-eslint/no-explicit-any
const registerCoreTool = (ToolClass: any, ...args: unknown[]) => {
// check both the tool name (.Name) and the class name (.name)
- if (!tools || tools.has(ToolClass.Name) || tools.has(ToolClass.name)) {
+ if (
+ // coreTools contain tool name
+ (!tools || tools.has(ToolClass.Name) || tools.has(ToolClass.name)) &&
+ // excludeTools don't contain tool name
+ (!excludeTools ||
+ (!excludeTools.has(ToolClass.Name) &&
+ !excludeTools.has(ToolClass.name)))
+ ) {
registry.registerTool(new ToolClass(...args));
}
};