fixup

Author: jpinkney-aws

packages/core/src/amazonq/messages/chatMessageDuration.ts

@@ -43,10 +43,11 @@ export class AmazonQChatMessageDuration {
     /**
      * Stop listening to all incoming events and emit what we've found
      */
-    static stopChatMessageTelemetry(msg: { tabID: string; time: number }) {
-        const { tabID, time } = msg
+    static stopChatMessageTelemetry(msg: { tabID: string; time: number; tabType: TabType }) {
+        const { tabID, time, tabType } = msg
+
         // We can't figure out what trace this event was associated with
-        if (!tabID) {
+        if (!tabID || tabType !== 'cwc') {
             return
         }
 

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

@@ -491,7 +491,7 @@ export class CWCTelemetryHelper {
     }
 
     public setResponseStreamStartTime(tabID: string) {
-        this.responseStreamStartTime.set(tabID, globals.clock.Date.now())
+        this.responseStreamStartTime.set(tabID, performance.now())
         this.responseStreamTimeForChunks.set(tabID, [performance.now()])
         this.displayTimeForChunks.set(tabID, [])
     }
@@ -545,7 +545,7 @@ export class CWCTelemetryHelper {
     }
 
     public setResponseStreamTotalTime(tabID: string) {
-        const totalTime = globals.clock.Date.now() - (this.responseStreamStartTime.get(tabID) ?? 0)
+        const totalTime = performance.now() - (this.responseStreamStartTime.get(tabID) ?? 0)
         this.responseStreamTotalTime.set(tabID, Math.round(totalTime))
     }