Skip to content

Commit 5b14a02

Browse files
authored
Adding credentialUrl to differentiate the Internal Users vs External users (#5159)
1 parent 2eb3c14 commit 5b14a02

File tree

2 files changed

+6
-0
lines changed

2 files changed

+6
-0
lines changed

plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqCodeTest/CodeWhispererUTGChatManager.kt

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,7 @@ import software.aws.toolkits.jetbrains.services.codemodernizer.utils.calculateTo
4141
import software.aws.toolkits.jetbrains.services.codewhisperer.codetest.sessionconfig.CodeTestSessionConfig
4242
import software.aws.toolkits.jetbrains.services.codewhisperer.credentials.CodeWhispererClientAdaptor
4343
import software.aws.toolkits.jetbrains.services.codewhisperer.util.CodeWhispererUtil.promptReAuth
44+
import software.aws.toolkits.jetbrains.services.cwc.controller.chat.telemetry.getStartUrl
4445
import software.aws.toolkits.jetbrains.services.cwc.messages.ChatMessageType
4546
import software.aws.toolkits.jetbrains.services.cwc.messages.CodeReference
4647
import software.aws.toolkits.jetbrains.settings.CodeWhispererSettings
@@ -511,6 +512,7 @@ class CodeWhispererUTGChatManager(val project: Project, private val cs: Coroutin
511512
cwsprChatProgrammingLanguage = session.programmingLanguage.languageId,
512513
hasUserPromptSupplied = session.hasUserPromptSupplied,
513514
isSupportedLanguage = true,
515+
credentialStartUrl = getStartUrl(project),
514516
jobGroup = session.testGenerationJobGroupName,
515517
jobId = session.testGenerationJob,
516518
result = if (e.message == message("testgen.message.cancelled")) MetricResult.Cancelled else MetricResult.Failed,

plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqCodeTest/controller/CodeTestChatController.kt

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -77,6 +77,7 @@ import software.aws.toolkits.jetbrains.services.cwc.clients.chat.model.TriggerTy
7777
import software.aws.toolkits.jetbrains.services.cwc.clients.chat.v1.ChatSessionV1.Companion.validLanguages
7878
import software.aws.toolkits.jetbrains.services.cwc.controller.chat.StaticPrompt
7979
import software.aws.toolkits.jetbrains.services.cwc.controller.chat.StaticTextResponse
80+
import software.aws.toolkits.jetbrains.services.cwc.controller.chat.telemetry.getStartUrl
8081
import software.aws.toolkits.jetbrains.services.cwc.editor.context.ActiveFileContext
8182
import software.aws.toolkits.jetbrains.services.cwc.editor.context.ActiveFileContextExtractor
8283
import software.aws.toolkits.jetbrains.services.cwc.editor.context.ExtractionTriggerType
@@ -285,6 +286,7 @@ class CodeTestChatController(
285286
cwsprChatProgrammingLanguage = session.programmingLanguage.languageId,
286287
hasUserPromptSupplied = session.hasUserPromptSupplied,
287288
isSupportedLanguage = false,
289+
credentialStartUrl = getStartUrl(project),
288290
result = MetricResult.Succeeded,
289291
perfClientLatency = (Instant.now().toEpochMilli() - session.startTimeOfTestGeneration)
290292
)
@@ -583,6 +585,7 @@ class CodeTestChatController(
583585
cwsprChatProgrammingLanguage = session.programmingLanguage.languageId,
584586
hasUserPromptSupplied = session.hasUserPromptSupplied,
585587
isSupportedLanguage = true,
588+
credentialStartUrl = getStartUrl(project = context.project),
586589
jobGroup = session.testGenerationJobGroupName,
587590
jobId = session.testGenerationJob,
588591
acceptedCount = session.numberOfUnitTestCasesGenerated?.toLong(),
@@ -775,6 +778,7 @@ class CodeTestChatController(
775778
cwsprChatProgrammingLanguage = session.programmingLanguage.languageId,
776779
hasUserPromptSupplied = session.hasUserPromptSupplied,
777780
isSupportedLanguage = true,
781+
credentialStartUrl = getStartUrl(project = context.project),
778782
jobGroup = session.testGenerationJobGroupName,
779783
jobId = session.testGenerationJob,
780784
acceptedCount = 0,

0 commit comments

Comments
 (0)