From 3309f8f81744dd376de84b195c2450870555f434 Mon Sep 17 00:00:00 2001
From: pashpashpash <nik@nugbase.com>
Date: Mon, 5 May 2025 00:23:47 -0700
Subject: [PATCH] tracking models in diff edit failures

---
 src/core/task/index.ts                     | 2 +-
 src/services/telemetry/TelemetryService.ts | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/core/task/index.ts b/src/core/task/index.ts
index f6b30a06e0..c706fa95cc 100644
--- a/src/core/task/index.ts
+++ b/src/core/task/index.ts
@@ -1915,7 +1915,7 @@ export class Task {
 											: "other_diff_error"
 
 									// Add telemetry for diff edit failure
-									telemetryService.captureDiffEditFailure(this.taskId, errorType)
+									telemetryService.captureDiffEditFailure(this.taskId, this.api.getModel().id, errorType)
 
 									pushToolResult(
 										formatResponse.toolError(
diff --git a/src/services/telemetry/TelemetryService.ts b/src/services/telemetry/TelemetryService.ts
index df1ea3ff4b..c149b22ab9 100644
--- a/src/services/telemetry/TelemetryService.ts
+++ b/src/services/telemetry/TelemetryService.ts
@@ -418,12 +418,13 @@ class PostHogClient {
 	 * @param taskId Unique identifier for the task
 	 * @param errorType Type of error that occurred (e.g., "search_not_found", "invalid_format")
 	 */
-	public captureDiffEditFailure(taskId: string, errorType?: string) {
+	public captureDiffEditFailure(taskId: string, modelId: string, errorType?: string) {
 		this.capture({
 			event: PostHogClient.EVENTS.TASK.DIFF_EDIT_FAILED,
 			properties: {
 				taskId,
 				errorType,
+				modelId,
 			},
 		})
 	}