Skip to content

Commit fcbfcff

Browse files
committed
Revert "chore: cleanup telemetry"
This reverts commit c5ec47e.
1 parent 3b9e595 commit fcbfcff

File tree

4 files changed

+10
-3
lines changed

4 files changed

+10
-3
lines changed

src/tools/atlas/atlasTool.ts

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,10 @@ For more information on Atlas API access roles, visit: https://www.mongodb.com/d
8282
* @param args - The arguments passed to the tool
8383
* @returns The tool metadata
8484
*/
85-
protected resolveTelemetryMetadata(...args: Parameters<ToolCallback<typeof this.argsShape>>): AtlasMetadata {
85+
protected resolveTelemetryMetadata(
86+
result: CallToolResult,
87+
...args: Parameters<ToolCallback<typeof this.argsShape>>
88+
): AtlasMetadata {
8689
const toolMetadata: AtlasMetadata = {};
8790
if (!args.length) {
8891
return toolMetadata;

src/tools/atlas/read/getPerformanceAdvisor.ts

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -134,11 +134,12 @@ export class GetPerformanceAdvisorTool extends AtlasToolBase {
134134
}
135135

136136
protected override resolveTelemetryMetadata(
137+
result: CallToolResult,
137138
args: ToolArgs<typeof this.argsShape>,
138139
extra: RequestHandlerExtra<ServerRequest, ServerNotification>
139140
): PerfAdvisorToolMetadata {
140141
return {
141-
...super.resolveTelemetryMetadata(args, extra),
142+
...super.resolveTelemetryMetadata(result, args, extra),
142143
operations: args.operations,
143144
};
144145
}

src/tools/mongodb/mongodbTool.ts

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -112,6 +112,8 @@ export abstract class MongoDBToolBase extends ToolBase {
112112
}
113113

114114
protected resolveTelemetryMetadata(
115+
// eslint-disable-next-line @typescript-eslint/no-unused-vars
116+
result: CallToolResult,
115117
// eslint-disable-next-line @typescript-eslint/no-unused-vars
116118
args: ToolArgs<typeof this.argsShape>
117119
): AtlasMetadata {

src/tools/tool.ts

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -264,6 +264,7 @@ export abstract class ToolBase {
264264
}
265265

266266
protected abstract resolveTelemetryMetadata(
267+
result: CallToolResult,
267268
...args: Parameters<ToolCallback<typeof this.argsShape>>
268269
): TelemetryToolMetadata;
269270

@@ -282,7 +283,7 @@ export abstract class ToolBase {
282283
return;
283284
}
284285
const duration = Date.now() - startTime;
285-
const metadata = this.resolveTelemetryMetadata(...args);
286+
const metadata = this.resolveTelemetryMetadata(result, ...args);
286287
const event: ToolEvent = {
287288
timestamp: new Date().toISOString(),
288289
source: "mdbmcp",

0 commit comments

Comments
 (0)