Adding Q Chat metric: totalNumberOfCodeBlocks to telemetry

Author: laileni-aws

package-lock.json

@@ -88,6 +88,16 @@ export class Messenger {
             )
         )
     }
+
+    public countTotalNumberOfCodeBlocks(message: string): number {
+        if (message === undefined) {
+            return 0
+        }
+        const countOfCodeBlocks = message.match(/^```/gm)
+        const numberOfTripleBackTicksInMarkdown = countOfCodeBlocks ? countOfCodeBlocks.length : 0
+        return Math.floor(numberOfTripleBackTicksInMarkdown / 2)
+    }
+
     public async sendAIResponse(
         response: GenerateAssistantResponseCommandOutput,
         session: ChatSession,
@@ -264,6 +274,7 @@ export class Messenger {
                     messageID,
                     responseCode,
                     codeReferenceCount: codeReference.length,
+                    totalNumberOfCodeBlocksInResponse: this.countTotalNumberOfCodeBlocks(message),
                 })
             })
     }

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

@@ -76,6 +76,7 @@ export interface PromptAnswer {
     messageID: string
     responseCode: number
     codeReferenceCount: number
+    totalNumberOfCodeBlocksInResponse: number
 }
 
 export interface StopResponseMessage {

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

@@ -321,7 +321,7 @@ export class CWCTelemetryHelper {
             cwsprChatProgrammingLanguage: triggerPayload.fileLanguage,
             cwsprChatActiveEditorTotalCharacters: triggerPayload.fileText?.length,
             cwsprChatActiveEditorImportCount: triggerPayload.codeQuery?.fullyQualifiedNames?.used?.length,
-            cwsprChatResponseCodeSnippetCount: 0, // TODO
+            cwsprChatResponseCodeSnippetCount: message.totalNumberOfCodeBlocksInResponse,
             cwsprChatResponseCode: message.responseCode,
             cwsprChatSourceLinkCount: message.suggestionCount,
             cwsprChatReferencesCount: message.codeReferenceCount,