feat(telemetry): Add extension name to ToolCallEvent telemetry (#12343)

Co-authored-by: Shnatu <snatu@google.com>
This commit is contained in:
Shardul Natu
2025-10-31 06:50:22 -07:00
committed by GitHub
parent 12472ce9b2
commit 236334d015
6 changed files with 13 additions and 1 deletions

View File

@@ -628,6 +628,7 @@ export async function discoverTools(
mcpServerConfig.trust,
undefined,
cliConfig,
mcpServerConfig.extension?.name,
mcpServerConfig.extension?.id,
messageBus,
);

View File

@@ -218,6 +218,7 @@ export class DiscoveredMCPTool extends BaseDeclarativeTool<
readonly trust?: boolean,
nameOverride?: string,
private readonly cliConfig?: Config,
override readonly extensionName?: string,
override readonly extensionId?: string,
messageBus?: MessageBus,
) {
@@ -230,6 +231,7 @@ export class DiscoveredMCPTool extends BaseDeclarativeTool<
true, // isOutputMarkdown
false, // canUpdateOutput,
messageBus,
extensionName,
extensionId,
);
}
@@ -244,6 +246,7 @@ export class DiscoveredMCPTool extends BaseDeclarativeTool<
this.trust,
`${this.serverName}__${this.serverToolName}`,
this.cliConfig,
this.extensionName,
this.extensionId,
this.messageBus,
);

View File

@@ -306,6 +306,7 @@ export abstract class DeclarativeTool<
readonly isOutputMarkdown: boolean = true,
readonly canUpdateOutput: boolean = false,
readonly messageBus?: MessageBus,
readonly extensionName?: string,
readonly extensionId?: string,
) {}