Browse Source

Add telemetry for prompt enhancement (#2651)

Matt Rubens 8 months ago
parent
commit
5f19ea4a06

+ 4 - 0
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,

+ 7 - 0
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,