Skip to content

Commit f39946f

Browse files
committed
Revert "fix(telemetry): stop emitting per-tool events"
This reverts commit 5006049.
1 parent 5006049 commit f39946f

2 files changed

Lines changed: 5 additions & 3 deletions

File tree

src/core/assistant-message/__tests__/presentAssistantMessage-custom-tool.spec.ts

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -118,7 +118,7 @@ describe("presentAssistantMessage - Custom Tool Recording", () => {
118118

119119
// Should record as "custom_tool", not "my_custom_tool"
120120
expect(mockTask.recordToolUsage).toHaveBeenCalledWith("custom_tool")
121-
expect(TelemetryService.instance.captureToolUsage).not.toHaveBeenCalled()
121+
expect(TelemetryService.instance.captureToolUsage).toHaveBeenCalledWith(mockTask.taskId, "custom_tool")
122122
})
123123
})
124124

@@ -171,7 +171,7 @@ describe("presentAssistantMessage - Custom Tool Recording", () => {
171171

172172
// Should record as "read_file", not "custom_tool"
173173
expect(mockTask.recordToolUsage).toHaveBeenCalledWith("read_file")
174-
expect(TelemetryService.instance.captureToolUsage).not.toHaveBeenCalled()
174+
expect(TelemetryService.instance.captureToolUsage).toHaveBeenCalledWith(mockTask.taskId, "read_file")
175175
})
176176

177177
it("should record MCP tool usage as 'use_mcp_tool' (not custom_tool)", async () => {
@@ -213,7 +213,7 @@ describe("presentAssistantMessage - Custom Tool Recording", () => {
213213

214214
// Should record as "use_mcp_tool", not "custom_tool"
215215
expect(mockTask.recordToolUsage).toHaveBeenCalledWith("use_mcp_tool")
216-
expect(TelemetryService.instance.captureToolUsage).not.toHaveBeenCalled()
216+
expect(TelemetryService.instance.captureToolUsage).toHaveBeenCalledWith(mockTask.taskId, "use_mcp_tool")
217217
})
218218
})
219219

src/core/assistant-message/presentAssistantMessage.ts

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -235,6 +235,7 @@ export async function presentAssistantMessage(cline: Task) {
235235

236236
if (!mcpBlock.partial) {
237237
cline.recordToolUsage("use_mcp_tool") // Record as use_mcp_tool for analytics
238+
TelemetryService.instance.captureToolUsage(cline.taskId, "use_mcp_tool")
238239
}
239240

240241
// Resolve sanitized server name back to original server name
@@ -557,6 +558,7 @@ export async function presentAssistantMessage(cline: Task) {
557558
const isCustomTool = stateExperiments?.customTools && customToolRegistry.has(block.name)
558559
const recordName = isCustomTool ? "custom_tool" : block.name
559560
cline.recordToolUsage(recordName)
561+
TelemetryService.instance.captureToolUsage(cline.taskId, recordName)
560562

561563
// Track legacy format usage for read_file tool (for migration monitoring)
562564
if (block.name === "read_file" && block.usedLegacyFormat) {

0 commit comments

Comments
 (0)