Adding credentialUrl to differentiate the Internal Users vs External …

…users
laileni-aws · Dec 3, 2024 · c5ea1f0 · c5ea1f0
1 parent 337959a
commit c5ea1f0
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/packages/core/src/amazonqTest/chat/controller/controller.ts b/packages/core/src/amazonqTest/chat/controller/controller.ts
@@ -255,6 +255,7 @@ export class TestController {
             result: isCancel ? 'Cancelled' : 'Failed',
             reasonDesc: getTelemetryReasonDesc(data.error),
             isSupportedLanguage: true,
+            credentialStartUrl: AuthUtil.instance.startUrl,
         })
         if (session.stopIteration) {
             // Error from Science
@@ -733,6 +734,7 @@ export class TestController {
             isCodeBlockSelected: session.isCodeBlockSelected,
             perfClientLatency: session.latencyOfTestGeneration,
             isSupportedLanguage: true,
+            credentialStartUrl: AuthUtil.instance.startUrl,
             result: 'Succeeded',
         })
 
@@ -855,6 +857,7 @@ export class TestController {
                 isCodeBlockSelected: session.isCodeBlockSelected,
                 perfClientLatency: session.latencyOfTestGeneration,
                 isSupportedLanguage: true,
+                credentialStartUrl: AuthUtil.instance.startUrl,
                 result: 'Succeeded',
             })
             telemetry.ui_click.emit({ elementId: 'unitTestGeneration_rejectDiff' })

diff --git a/packages/core/src/amazonqTest/chat/controller/messenger/messenger.ts b/packages/core/src/amazonqTest/chat/controller/messenger/messenger.ts
@@ -36,7 +36,7 @@ import { CodeReference } from '../../../../amazonq/webview/ui/apps/amazonqCommon
 import { getHttpStatusCode, getRequestId, getTelemetryReasonDesc, ToolkitError } from '../../../../shared/errors'
 import { sleep, waitUntil } from '../../../../shared/utilities/timeoutUtils'
 import { keys } from '../../../../shared/utilities/tsUtils'
-import { testGenState } from '../../../../codewhisperer'
+import { AuthUtil, testGenState } from '../../../../codewhisperer'
 import { cancellingProgressField, testGenCompletedField } from '../../../models/constants'
 import { telemetry } from '../../../../shared/telemetry/telemetry'
 
@@ -281,6 +281,7 @@ export class Messenger {
                         result: 'Cancelled',
                         reasonDesc: getTelemetryReasonDesc(CodeWhispererConstants.unitTestGenerationCancelMessage),
                         isSupportedLanguage: false,
+                        credentialStartUrl: AuthUtil.instance.startUrl,
                     })
 
                     this.dispatcher.sendUpdatePromptProgress(
@@ -294,6 +295,7 @@ export class Messenger {
                         perfClientLatency: performance.now() - session.testGenerationStartTime,
                         result: 'Succeeded',
                         isSupportedLanguage: false,
+                        credentialStartUrl: AuthUtil.instance.startUrl,
                     })
                     this.dispatcher.sendUpdatePromptProgress(
                         new UpdatePromptProgressMessage(tabID, testGenCompletedField)