Skip to content

Commit 41b0efa

Browse files
committed
fix(telemetry): address PR 1071 review feedback
1 parent d65a372 commit 41b0efa

7 files changed

Lines changed: 140 additions & 277 deletions

File tree

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

Lines changed: 0 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -23,16 +23,6 @@ vi.mock("@roo-code/core", () => ({
2323
},
2424
}))
2525

26-
vi.mock("@roo-code/telemetry", () => ({
27-
TelemetryService: {
28-
instance: {
29-
captureToolUsage: vi.fn(),
30-
captureConsecutiveMistakeError: vi.fn(),
31-
},
32-
},
33-
}))
34-
35-
import { TelemetryService } from "@roo-code/telemetry"
3626
import { customToolRegistry } from "@roo-code/core"
3727

3828
describe("presentAssistantMessage - Custom Tool Recording", () => {
@@ -118,7 +108,6 @@ describe("presentAssistantMessage - Custom Tool Recording", () => {
118108

119109
// Should record as "custom_tool", not "my_custom_tool"
120110
expect(mockTask.recordToolUsage).toHaveBeenCalledWith("custom_tool")
121-
expect(TelemetryService.instance.captureToolUsage).toHaveBeenCalledWith(mockTask.taskId, "custom_tool")
122111
})
123112
})
124113

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

172161
// Should record as "read_file", not "custom_tool"
173162
expect(mockTask.recordToolUsage).toHaveBeenCalledWith("read_file")
174-
expect(TelemetryService.instance.captureToolUsage).toHaveBeenCalledWith(mockTask.taskId, "read_file")
175163
})
176164

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

214202
// Should record as "use_mcp_tool", not "custom_tool"
215203
expect(mockTask.recordToolUsage).toHaveBeenCalledWith("use_mcp_tool")
216-
expect(TelemetryService.instance.captureToolUsage).toHaveBeenCalledWith(mockTask.taskId, "use_mcp_tool")
217204
})
218205
})
219206

@@ -355,7 +342,6 @@ describe("presentAssistantMessage - Custom Tool Recording", () => {
355342

356343
// Should not record usage for partial blocks
357344
expect(mockTask.recordToolUsage).not.toHaveBeenCalled()
358-
expect(TelemetryService.instance.captureToolUsage).not.toHaveBeenCalled()
359345
})
360346
})
361347
})

src/core/assistant-message/presentAssistantMessage.ts

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -235,7 +235,6 @@ 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")
239238
}
240239

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

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

0 commit comments

Comments
 (0)