From 51b583cf1329179c25d8889f2b493b226ecac626 Mon Sep 17 00:00:00 2001 From: Matt Rubens Date: Tue, 15 Apr 2025 10:43:45 -0400 Subject: [PATCH] Add telemetry for prompt enhancement --- src/core/webview/webviewMessageHandler.ts | 4 ++++ src/services/telemetry/TelemetryService.ts | 7 +++++++ 2 files changed, 11 insertions(+) diff --git a/src/core/webview/webviewMessageHandler.ts b/src/core/webview/webviewMessageHandler.ts index 3f264d2a876..cdbe81c8cef 100644 --- a/src/core/webview/webviewMessageHandler.ts +++ b/src/core/webview/webviewMessageHandler.ts @@ -1020,6 +1020,10 @@ export const webviewMessageHandler = async (provider: ClineProvider, message: We ), ) + // Capture telemetry for prompt enhancement + const currentCline = provider.getCurrentCline() + telemetryService.capturePromptEnhanced(currentCline?.taskId) + await provider.postMessageToWebview({ type: "enhancedPrompt", text: enhancedPrompt, diff --git a/src/services/telemetry/TelemetryService.ts b/src/services/telemetry/TelemetryService.ts index 4a61bebfff8..492d3e0aded 100644 --- a/src/services/telemetry/TelemetryService.ts +++ b/src/services/telemetry/TelemetryService.ts @@ -28,6 +28,7 @@ class PostHogClient { CHECKPOINT_RESTORED: "Checkpoint Restored", CHECKPOINT_DIFFED: "Checkpoint Diffed", CODE_ACTION_USED: "Code Action Used", + PROMPT_ENHANCED: "Prompt Enhanced", }, ERRORS: { SCHEMA_VALIDATION_ERROR: "Schema Validation Error", @@ -275,6 +276,12 @@ class TelemetryService { }) } + public capturePromptEnhanced(taskId?: string): void { + this.captureEvent(PostHogClient.EVENTS.TASK.PROMPT_ENHANCED, { + ...(taskId && { taskId }), + }) + } + public captureSchemaValidationError({ schemaName, error }: { schemaName: string; error: ZodError }): void { this.captureEvent(PostHogClient.EVENTS.ERRORS.SCHEMA_VALIDATION_ERROR, { schemaName,