Skip to content

Commit e5a0bb1

Browse files
authored
add JetBrains IdeCategory for enterprise telemetry (#5238)
1 parent 1121066 commit e5a0bb1

File tree

2 files changed

+7
-0
lines changed

2 files changed

+7
-0
lines changed

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

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@ import kotlinx.coroutines.future.await
2222
import kotlinx.coroutines.launch
2323
import kotlinx.coroutines.withContext
2424
import migration.software.aws.toolkits.jetbrains.services.codewhisperer.customization.CodeWhispererModelConfigurator
25+
import software.amazon.awssdk.services.codewhispererruntime.model.IdeCategory
2526
import software.amazon.awssdk.services.codewhispererruntime.model.Position
2627
import software.amazon.awssdk.services.codewhispererruntime.model.Range
2728
import software.amazon.awssdk.services.codewhispererruntime.model.Reference
@@ -571,6 +572,7 @@ class CodeTestChatController(
571572
session.testGenerationJob,
572573
session.testGenerationJobGroupName,
573574
session.programmingLanguage,
575+
IdeCategory.JETBRAINS,
574576
session.numberOfUnitTestCasesGenerated,
575577
session.numberOfUnitTestCasesGenerated,
576578
session.linesOfCodeGenerated,
@@ -766,6 +768,7 @@ class CodeTestChatController(
766768
session.testGenerationJob,
767769
session.testGenerationJobGroupName,
768770
session.programmingLanguage,
771+
IdeCategory.JETBRAINS,
769772
session.numberOfUnitTestCasesGenerated,
770773
0,
771774
session.linesOfCodeGenerated,

plugins/amazonq/codewhisperer/jetbrains-community/src/software/aws/toolkits/jetbrains/services/codewhisperer/credentials/CodeWhispererClientAdaptor.kt

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@ import software.amazon.awssdk.services.codewhispererruntime.model.GenerateComple
2727
import software.amazon.awssdk.services.codewhispererruntime.model.GetCodeFixJobRequest
2828
import software.amazon.awssdk.services.codewhispererruntime.model.GetCodeFixJobResponse
2929
import software.amazon.awssdk.services.codewhispererruntime.model.GetTestGenerationResponse
30+
import software.amazon.awssdk.services.codewhispererruntime.model.IdeCategory
3031
import software.amazon.awssdk.services.codewhispererruntime.model.InlineChatUserDecision
3132
import software.amazon.awssdk.services.codewhispererruntime.model.ListAvailableCustomizationsRequest
3233
import software.amazon.awssdk.services.codewhispererruntime.model.ListFeatureEvaluationsResponse
@@ -199,6 +200,7 @@ interface CodeWhispererClientAdaptor : Disposable {
199200
jobId: String,
200201
groupName: String,
201202
language: CodeWhispererProgrammingLanguage?,
203+
ideCategory: IdeCategory?,
202204
numberOfUnitTestCasesGenerated: Int?,
203205
numberOfUnitTestCasesAccepted: Int?,
204206
linesOfCodeGenerated: Int?,
@@ -668,6 +670,7 @@ open class CodeWhispererClientAdaptorImpl(override val project: Project) : CodeW
668670
jobId: String,
669671
groupName: String,
670672
language: CodeWhispererProgrammingLanguage?,
673+
ideCategory: IdeCategory?,
671674
numberOfUnitTestCasesGenerated: Int?,
672675
numberOfUnitTestCasesAccepted: Int?,
673676
linesOfCodeGenerated: Int?,
@@ -682,6 +685,7 @@ open class CodeWhispererClientAdaptorImpl(override val project: Project) : CodeW
682685
}
683686
it.jobId(jobId)
684687
it.groupName(groupName)
688+
it.ideCategory(ideCategory)
685689
it.numberOfUnitTestCasesGenerated(numberOfUnitTestCasesGenerated)
686690
it.numberOfUnitTestCasesAccepted(numberOfUnitTestCasesAccepted)
687691
it.linesOfCodeGenerated(linesOfCodeGenerated)

0 commit comments

Comments
 (0)