MediaMetz/gemini-cli
1
0
Fork 0
mirror of https://github.com/google-gemini/gemini-cli.git synced 2026-04-26 04:54:25 -07:00
Code Issues Packages Projects Releases Wiki Activity

feat(core): Standardize Tool and Agent Invocation constructors (Phase 2) (#15775)

Browse Source
This commit is contained in:
Abhi
2026-01-04 15:51:23 -05:00
committed by GitHub
parent eec5d5ebf8
commit 90be9c3587
23 changed files with 140 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/core/src/test-utils/mock-tool.ts
+14 -6
View File
@@ -47,7 +47,7 @@ class MockToolInvocation extends BaseToolInvocation<
constructor(
private readonly tool: MockTool,
params: { [key: string]: unknown },
messageBus: MessageBus,
messageBus?: MessageBus,
) {
super(params, messageBus, tool.name, tool.displayName);
}
@@ -122,9 +122,11 @@ export class MockTool extends BaseDeclarativeTool<
protected createInvocation(
params: { [key: string]: unknown },
messageBus: MessageBus,
messageBus?: MessageBus,
_toolName?: string,
_toolDisplayName?: string,
): ToolInvocation<{ [key: string]: unknown }, ToolResult> {
return new MockToolInvocation(this, params, messageBus);
return new MockToolInvocation(this, params, messageBus ?? this.messageBus);
}
}
@@ -144,7 +146,7 @@ export class MockModifiableToolInvocation extends BaseToolInvocation<
constructor(
private readonly tool: MockModifiableTool,
params: Record<string, unknown>,
messageBus: MessageBus,
messageBus?: MessageBus,
) {
super(params, messageBus, tool.name, tool.displayName);
}
@@ -228,8 +230,14 @@ export class MockModifiableTool
protected createInvocation(
params: Record<string, unknown>,
messageBus: MessageBus,
messageBus?: MessageBus,
_toolName?: string,
_toolDisplayName?: string,
): ToolInvocation<Record<string, unknown>, ToolResult> {
return new MockModifiableToolInvocation(this, params, messageBus);
return new MockModifiableToolInvocation(
this,
params,
messageBus ?? this.messageBus,
);
}
}