send customization arn for chat telemetry

Author: Will-ShaoHua

packages/core/src/codewhisperer/tracker/codewhispererTracker.ts

@@ -15,6 +15,7 @@ import { InsertedCode } from '../../codewhispererChat/controllers/chat/model'
 import { codeWhispererClient } from '../client/codewhisperer'
 import { logSendTelemetryEventFailure } from '../../codewhispererChat/controllers/chat/telemetryHelper'
 import { Timeout } from '../../shared/utilities/timeoutUtils'
+import { activeCustomizationOrNull } from '../util/customizationUtil'
 
 /**
  * This singleton class is mainly used for calculating the percentage of user modification.
@@ -88,6 +89,8 @@ export class CodeWhispererTracker {
 
     public async emitTelemetryOnSuggestion(suggestion: AcceptedSuggestionEntry | InsertedCode) {
         let percentage = 1.0
+        const customization = activeCustomizationOrNull()
+
         try {
             if (suggestion.fileUrl?.scheme !== '') {
                 const document = await vscode.workspace.openTextDocument(suggestion.fileUrl)
@@ -118,6 +121,7 @@ export class CodeWhispererTracker {
                                 conversationId: event.cwsprChatConversationId,
                                 messageId: event.cwsprChatMessageId,
                                 modificationPercentage: event.cwsprChatModificationPercentage,
+                                customizationArn: customization?.arn,
                             },
                         },
                     })

packages/core/src/codewhisperer/util/customizationUtil.ts

@@ -122,6 +122,16 @@ export const getSelectedCustomization = (): Customization => {
     }
 }
 
+// codewhisperer service will throw validation exception if arn is empty string, thus we need to specifically pass undefined if it's base customization
+export function activeCustomizationOrNull(): Customization | undefined {
+    const customization = getSelectedCustomization()
+    if (customization === baseCustomization) {
+        return undefined
+    }
+
+    return customization
+}
+
 export const setSelectedCustomization = async (customization: Customization) => {
     if (!AuthUtil.instance.isValidEnterpriseSsoInUse() || !AuthUtil.instance.conn) {
         return

packages/core/src/codewhispererChat/controllers/chat/telemetryHelper.ts

@@ -34,6 +34,7 @@ import { isAwsError } from '../../../shared/errors'
 import { ChatMessageInteractionType } from '../../../codewhisperer/client/codewhispereruserclient'
 import { supportedLanguagesList } from '../chat/chatRequest/converter'
 import { AuthUtil } from '../../../codewhisperer/util/authUtil'
+import { activeCustomizationOrNull } from '../../../codewhisperer/util/customizationUtil'
 
 export function logSendTelemetryEventFailure(error: any) {
     let requestId: string | undefined
@@ -258,6 +259,7 @@ export class CWCTelemetryHelper {
                         acceptedLineCount: event.cwsprChatAcceptedNumberOfLines,
                         acceptedSnippetHasReference: false,
                         hasProjectLevelContext: this.responseWithProjectContext.get(event.cwsprChatMessageId),
+                        customizationArn: activeCustomizationOrNull()?.arn,
                     },
                 },
             })
@@ -381,6 +383,7 @@ export class CWCTelemetryHelper {
                         hasProjectLevelContext: triggerPayload.relevantTextDocuments
                             ? triggerPayload.relevantTextDocuments.length > 0
                             : false,
+                        customizationArn: activeCustomizationOrNull()?.arn,
                     },
                 },
             })