Merge pull request #1 from kelvin-klchu/stop-code-gen-telemetry-updated

Add stop code gen telemetry

Author: tverney

packages/core/src/amazonqFeatureDev/controllers/chat/controller.ts

@@ -749,21 +749,24 @@ export class FeatureDevController {
     }
 
     private async stopResponse(message: any) {
-        this.messenger.sendAnswer({
-            message: i18n('AWS.amazonq.featureDev.pillText.stoppingCodeGeneration'),
-            type: 'answer-part',
-            tabID: message.tabID,
-        })
-        this.messenger.sendUpdatePlaceholder(
-            message.tabID,
-            i18n('AWS.amazonq.featureDev.pillText.stoppingCodeGeneration')
-        )
-        this.messenger.sendChatInputEnabled(message.tabID, false)
+        await telemetry.amazonq_stopCodeGeneration.run(async (span) => {
+            span.record({ tabID: message.tabID })
+            this.messenger.sendAnswer({
+                message: i18n('AWS.amazonq.featureDev.pillText.stoppingCodeGeneration'),
+                type: 'answer-part',
+                tabID: message.tabID,
+            })
+            this.messenger.sendUpdatePlaceholder(
+                message.tabID,
+                i18n('AWS.amazonq.featureDev.pillText.stoppingCodeGeneration')
+            )
+            this.messenger.sendChatInputEnabled(message.tabID, false)
 
-        const session = await this.sessionStorage.getSession(message.tabID)
-        if (session.state?.tokenSource) {
-            session.state?.tokenSource?.cancel()
-        }
+            const session = await this.sessionStorage.getSession(message.tabID)
+            if (session.state?.tokenSource) {
+                session.state?.tokenSource?.cancel()
+            }
+        })
     }
 
     private async tabOpened(message: any) {

packages/core/src/shared/telemetry/vscodeTelemetry.json

@@ -358,6 +358,11 @@
             "name": "amazonqMessageDisplayedMs",
             "type": "int",
             "description": "Duration between the partner teams code receiving the message and when the message was finally displayed in ms"
+        },
+        {
+            "name": "tabID",
+            "type": "string",
+            "description": "The unique identifier of a tab"
         }
     ],
     "metrics": [
@@ -1260,6 +1265,16 @@
                     "required": false
                 }
             ]
+        },
+        {
+            "name": "amazonq_stopCodeGeneration",
+            "description": "User stopped the code generation",
+            "metadata": [
+                {
+                    "type": "tabID",
+                    "required": true
+                }
+            ]
         }
     ]
 }

packages/core/src/test/amazonqFeatureDev/controllers/chat/controller.test.ts

@@ -404,4 +404,18 @@ describe('Controller', () => {
             })
         })
     })
+
+    describe('stopResponse', () => {
+        it('should emit amazonq_stopCodeGeneration telemetry', async () => {
+            const getSessionStub = sinon.stub(controllerSetup.sessionStorage, 'getSession').resolves(session)
+            controllerSetup.emitters.stopResponse.fire({ tabID, conversationID })
+            await waitUntil(() => {
+                return Promise.resolve(getSessionStub.callCount > 0)
+            }, {})
+            assertTelemetry('amazonq_stopCodeGeneration', {
+                tabID: tabID,
+                result: 'Succeeded',
+            })
+        })
+    })
 })