telemetry(amazonq): Adding cwsprChatProgrammingLanguage parameter to insertAtCursor and CopyAtClipboard events

plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/cwc/controller/chat/messenger/ChatPromptHandler.kt

@@ -33,6 +33,7 @@
     private val codeReferences = mutableListOf<CodeReference>()
     private var requestId: String = ""
     private var statusCode: Int = 0
+    private var codeBlockLanguage: String = "plaintext"
 
     companion object {
         private val CODE_BLOCK_PATTERN = Regex("<pre>\\s*<code")
@@ -143,6 +144,24 @@
             }
     }
 
+    private fun extractCodeBlockLanguage(message: String): String {
+        // This fulfills both the cases of unit test generation(java, python) and general use case(Non java and Non python) languages.
+        val codeBlockStart = message.indexOf("```")
+        if (codeBlockStart == -1) {
+            return "plaintext"
+        }
+
+        val languageStart = codeBlockStart + 3
+        val languageEnd = message.indexOf('\n', languageStart)
+
+        if (languageEnd == -1) {
+            return "plaintext"
+        }
+
+        val language = message.substring(languageStart, languageEnd).trim()
+        return if (language.isNotEmpty()) language else "plaintext"
+    }
+
     private fun processChatEvent(
         tabId: String,
         triggerId: String,
@@ -209,6 +228,10 @@
             } else {
                 responseText.toString()
             }
+            if (codeBlockLanguage == "plaintext") {
+                // To get the language of generated code in Q chat.
+                codeBlockLanguage = extractCodeBlockLanguage(message)
+            }
             ChatMessage(
                 tabId = tabId,
                 triggerId = triggerId,
@@ -217,6 +240,7 @@
                 message = message,
                 codeReference = codeReferences,
                 userIntent = data.userIntent,
+                codeBlockLanguage = codeBlockLanguage,
             )
         } else {
             null

plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/cwc/controller/chat/telemetry/TelemetryHelper.kt

@@ -219,7 +219,8 @@ class TelemetryHelper(private val context: AmazonQAppInitContext, private val se
                     credentialStartUrl = getStartUrl(context.project),
                     cwsprChatCodeBlockIndex = message.codeBlockIndex?.toLong(),
                     cwsprChatTotalCodeBlocks = message.totalCodeBlocks?.toLong(),
-                    cwsprChatHasProjectContext = getMessageHasProjectContext(message.messageId)
+                    cwsprChatHasProjectContext = getMessageHasProjectContext(message.messageId),
+                    cwsprChatProgrammingLanguage = message.codeBlockLanguage,
                 )
                 ChatInteractWithMessageEvent.builder().apply {
                     conversationId(getConversationId(message.tabId).orEmpty())
@@ -244,7 +245,8 @@ class TelemetryHelper(private val context: AmazonQAppInitContext, private val se
                     credentialStartUrl = getStartUrl(context.project),
                     cwsprChatCodeBlockIndex = message.codeBlockIndex?.toLong(),
                     cwsprChatTotalCodeBlocks = message.totalCodeBlocks?.toLong(),
-                    cwsprChatHasProjectContext = getMessageHasProjectContext(message.messageId)
+                    cwsprChatHasProjectContext = getMessageHasProjectContext(message.messageId),
+                    cwsprChatProgrammingLanguage = message.codeBlockLanguage,
                 )
                 ChatInteractWithMessageEvent.builder().apply {
                     conversationId(getConversationId(message.tabId).orEmpty())

plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/cwc/messages/CwcMessage.kt

@@ -72,6 +72,7 @@ sealed interface IncomingCwcMessage : CwcMessage {
         val eventId: String?,
         val codeBlockIndex: Int?,
         val totalCodeBlocks: Int?,
+        val codeBlockLanguage: String?,
     ) : IncomingCwcMessage
 
     data class InsertCodeAtCursorPosition(
@@ -84,6 +85,7 @@ sealed interface IncomingCwcMessage : CwcMessage {
         val eventId: String?,
         val codeBlockIndex: Int?,
         val totalCodeBlocks: Int?,
+        val codeBlockLanguage: String?,
     ) : IncomingCwcMessage
 
     data class TriggerTabIdReceived(
@@ -214,6 +216,7 @@ data class ChatMessage(
     val relatedSuggestions: List<Suggestion>? = null,
     val codeReference: List<CodeReference>? = null,
     val userIntent: UserIntent? = null,
+    val codeBlockLanguage: String? = "plaintext",
 ) : UiMessage(
     tabId = tabId,
     type = "chatMessage",

plugins/amazonq/chat/jetbrains-community/tst/software/aws/toolkits/jetbrains/services/amazonq/TelemetryHelperTest.kt

@@ -420,7 +420,8 @@ class TelemetryHelperTest {
                 "insertionTargetType",
                 "eventId",
                 codeBlockIndex,
-                totalCodeBlocks
+                totalCodeBlocks,
+                lang
             )
         )
 
@@ -484,7 +485,8 @@ class TelemetryHelperTest {
                 emptyList(),
                 eventId,
                 codeBlockIndex,
-                totalCodeBlocks
+                totalCodeBlocks,
+                lang
             )
         )
 

plugins/amazonq/mynah-ui/src/mynah-ui/ui/apps/cwChatConnector.ts

@@ -101,6 +101,7 @@ export class Connector {
         codeBlockIndex?: number,
         totalCodeBlocks?: number,
         userIntent?: string,
+        codeBlockLanguage?: string,
     ): void => {
         this.sendMessageToExtension({
             tabID: tabID,
@@ -113,7 +114,8 @@ export class Connector {
             eventId,
             codeBlockIndex,
             totalCodeBlocks,
-            userIntent
+            userIntent,
+            codeBlockLanguage
         })
     }
 
@@ -127,6 +129,7 @@ export class Connector {
         codeBlockIndex?: number,
         totalCodeBlocks?: number,
         userIntent?: string,
+        codeBlockLanguage?: string,
     ): void => {
         this.sendMessageToExtension({
             tabID: tabID,
@@ -139,7 +142,8 @@ export class Connector {
             eventId,
             codeBlockIndex,
             totalCodeBlocks,
-            userIntent
+            userIntent,
+            codeBlockLanguage
         })
     }
 
@@ -271,6 +275,7 @@ export class Connector {
                 canBeVoted: true,
                 codeReference: messageData.codeReference,
                 userIntent: messageData.userIntent,
+                codeBlockLanguage: messageData.codeBlockLanguage,
             }
 
             // If it is not there we will not set it
@@ -304,6 +309,7 @@ export class Connector {
                 messageId: messageData.messageId,
                 codeReference: messageData.codeReference,
                 userIntent: messageData.userIntent,
+                codeBlockLanguage: messageData.codeBlockLanguage,
                 followUp:
                     messageData.followUps !== undefined && messageData.followUps.length > 0
                         ? {

plugins/amazonq/mynah-ui/src/mynah-ui/ui/connector.ts

@@ -31,7 +31,8 @@ export interface ChatPayload {
 }
 
 export interface CWCChatItem extends ChatItem {
-    userIntent?: string
+    userIntent?: string,
+    codeBlockLanguage?: string,
 }
 
 export interface ConnectorProps {
@@ -235,7 +236,8 @@ export class Connector {
         eventId?: string,
         codeBlockIndex?: number,
         totalCodeBlocks?: number,
-        userIntent?: string
+        userIntent?: string,
+        codeBlockLanguage?: string
     ): void => {
         switch (this.tabsStorage.getTab(tabID)?.type) {
             case 'cwc':
@@ -248,7 +250,8 @@ export class Connector {
                     eventId,
                     codeBlockIndex,
                     totalCodeBlocks,
-                    userIntent
+                    userIntent,
+                    codeBlockLanguage
                 )
                 break
             case 'featuredev':
@@ -266,7 +269,8 @@ export class Connector {
         eventId?: string,
         codeBlockIndex?: number,
         totalCodeBlocks?: number,
-        userIntent?: string
+        userIntent?: string,
+        codeBlockLanguage?: string
     ): void => {
         switch (this.tabsStorage.getTab(tabID)?.type) {
             case 'cwc':
@@ -279,7 +283,8 @@ export class Connector {
                     eventId,
                     codeBlockIndex,
                     totalCodeBlocks,
-                    userIntent
+                    userIntent,
+                    codeBlockLanguage
                 )
                 break
             case 'featuredev':

plugins/amazonq/mynah-ui/src/mynah-ui/ui/main.ts

@@ -24,7 +24,7 @@ export const createMynahUI = (ideApi: any, featureDevInitEnabled: boolean, codeT
     let mynahUI: MynahUI
     // eslint-disable-next-line prefer-const
     let connector: Connector
-    const messageUserIntentMap = new Map<string, string>()
+    const responseMetadata = new Map<string, string[]>()
 
     const tabsStorage = new TabsStorage({
         onTabTimeout: tabID => {
@@ -254,8 +254,8 @@ export const createMynahUI = (ideApi: any, featureDevInitEnabled: boolean, codeT
                         ? { type: ChatItemType.CODE_RESULT, fileList: item.fileList }
                         : {}),
                 })
-                if (item.messageId !== undefined && item.userIntent !== undefined) {
-                    messageUserIntentMap.set(item.messageId, item.userIntent)
+                if (item.messageId !== undefined && item.userIntent !== undefined && item.codeBlockLanguage !== undefined) {
+                    responseMetadata.set(item.messageId, [item.userIntent, item.codeBlockLanguage])
                 }
                 return
             }
@@ -466,7 +466,8 @@ export const createMynahUI = (ideApi: any, featureDevInitEnabled: boolean, codeT
                         eventId,
                         codeBlockIndex,
                         totalCodeBlocks,
-                        messageUserIntentMap.get(messageId) ?? undefined
+                        responseMetadata.get(messageId)?.[0] ?? undefined,
+                        responseMetadata.get(messageId)?.[1] ?? undefined
                     )
                     break
                 case 'copy':
@@ -479,7 +480,8 @@ export const createMynahUI = (ideApi: any, featureDevInitEnabled: boolean, codeT
                         eventId,
                         codeBlockIndex,
                         totalCodeBlocks,
-                        messageUserIntentMap.get(messageId) ?? undefined
+                        responseMetadata.get(messageId)?.[0] ?? undefined,
+                        responseMetadata.get(messageId)?.[1] ?? undefined
                     )
                     mynahUI.notify({
                         type: NotificationType.SUCCESS,

plugins/core/jetbrains-community/resources/telemetryOverride.json

@@ -692,6 +692,10 @@
                 {
                     "type": "cwsprChatHasProjectContext",
                     "required": false
+                },
+                {
+                    "type": "cwsprChatProgrammingLanguage",
+                    "required": false
                 }
             ]
         },