Skip to content

Commit 5b9ee01

Browse files
Merge main into feature/q-lsp
2 parents 9e3760e + 1872cf7 commit 5b9ee01

File tree

2 files changed

+10
-13
lines changed

2 files changed

+10
-13
lines changed

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

Lines changed: 9 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@ import com.intellij.openapi.Disposable
77
import com.intellij.openapi.application.ApplicationManager
88
import com.intellij.openapi.components.service
99
import com.intellij.openapi.project.Project
10+
import com.intellij.util.text.nullize
1011
import software.amazon.awssdk.auth.credentials.AnonymousCredentialsProvider
1112
import software.amazon.awssdk.services.codewhisperer.CodeWhispererClient
1213
import software.amazon.awssdk.services.codewhisperer.model.CreateCodeScanRequest
@@ -146,7 +147,7 @@ interface CodeWhispererClientAdaptor : Disposable {
146147
sessionId: String,
147148
requestId: String,
148149
language: CodeWhispererProgrammingLanguage,
149-
customizationArn: String,
150+
customizationArn: String?,
150151
acceptedCharacterCount: Int,
151152
unmodifiedAcceptedTokenCount: Int,
152153
): SendTelemetryEventResponse
@@ -427,7 +428,7 @@ open class CodeWhispererClientAdaptorImpl(override val project: Project) : CodeW
427428
it.timestamp(Instant.now())
428429
it.suggestionReferenceCount(suggestionReferenceCount)
429430
it.generatedLine(lineCount)
430-
it.customizationArn(requestContext.customizationArn)
431+
it.customizationArn(requestContext.customizationArn.nullize(nullizeSpaces = true))
431432
it.numberOfRecommendations(numberOfRecommendations)
432433
it.acceptedCharacterCount(acceptedCharCount)
433434
}
@@ -473,7 +474,7 @@ open class CodeWhispererClientAdaptorImpl(override val project: Project) : CodeW
473474
it.timestamp(Instant.now())
474475
it.suggestionReferenceCount(suggestionReferenceCount)
475476
it.generatedLine(lineCount)
476-
it.customizationArn(requestContext.customizationArn)
477+
it.customizationArn(requestContext.customizationArn.nullize(nullizeSpaces = true))
477478
it.numberOfRecommendations(numberOfRecommendations)
478479
it.acceptedCharacterCount(acceptedCharCount)
479480
}
@@ -495,7 +496,7 @@ open class CodeWhispererClientAdaptorImpl(override val project: Project) : CodeW
495496
requestBuilder.telemetryEvent { telemetryEventBuilder ->
496497
telemetryEventBuilder.codeCoverageEvent {
497498
it.programmingLanguage { languageBuilder -> languageBuilder.languageName(language.toCodeWhispererRuntimeLanguage().languageId) }
498-
it.customizationArn(customizationArn)
499+
it.customizationArn(customizationArn.nullize(nullizeSpaces = true))
499500
it.acceptedCharacterCount(acceptedTokenCount.toInt())
500501
it.totalCharacterCount(totalTokenCount.toInt())
501502
it.timestamp(Instant.now())
@@ -512,7 +513,7 @@ open class CodeWhispererClientAdaptorImpl(override val project: Project) : CodeW
512513
sessionId: String,
513514
requestId: String,
514515
language: CodeWhispererProgrammingLanguage,
515-
customizationArn: String,
516+
customizationArn: String?,
516517
acceptedCharacterCount: Int,
517518
unmodifiedAcceptedTokenCount: Int,
518519
): SendTelemetryEventResponse = bearerClient().sendTelemetryEvent { requestBuilder ->
@@ -523,7 +524,7 @@ open class CodeWhispererClientAdaptorImpl(override val project: Project) : CodeW
523524
it.programmingLanguage { languageBuilder ->
524525
languageBuilder.languageName(language.toCodeWhispererRuntimeLanguage().languageId)
525526
}
526-
it.customizationArn(customizationArn)
527+
it.customizationArn(customizationArn.nullize(nullizeSpaces = true))
527528
// deprecated field, service side should not use this % anymore
528529
it.modificationPercentage(0.0)
529530
it.timestamp(Instant.now())
@@ -760,9 +761,7 @@ open class CodeWhispererClientAdaptorImpl(override val project: Project) : CodeW
760761
it.responseLength(responseLength)
761762
it.numberOfCodeBlocks(numberOfCodeBlocks)
762763
it.hasProjectLevelContext(hasProjectLevelContext)
763-
customization?.arn?.let { arn ->
764-
it.customizationArn(arn)
765-
}
764+
it.customizationArn(customization?.arn.nullize(nullizeSpaces = true))
766765
}
767766
}
768767
requestBuilder.optOutPreference(getTelemetryOptOutPreference())
@@ -815,9 +814,7 @@ open class CodeWhispererClientAdaptorImpl(override val project: Project) : CodeW
815814
}
816815
it.modificationPercentage(modificationPercentage)
817816
it.hasProjectLevelContext(hasProjectLevelContext)
818-
customization?.arn?.let { arn ->
819-
it.customizationArn(arn)
820-
}
817+
it.customizationArn(customization?.arn.nullize(nullizeSpaces = true))
821818
}
822819
}
823820
requestBuilder.optOutPreference(getTelemetryOptOutPreference())

plugins/amazonq/codewhisperer/jetbrains-community/src/software/aws/toolkits/jetbrains/services/codewhisperer/telemetry/CodeWhispererUserModificationTracker.kt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -265,7 +265,7 @@ class CodeWhispererUserModificationTracker(private val project: Project) : Dispo
265265
suggestion.sessionId,
266266
suggestion.requestId,
267267
CodeWhispererLanguageManager.getInstance().getLanguage(suggestion.vFile),
268-
CodeWhispererModelConfigurator.getInstance().activeCustomization(project)?.arn.orEmpty(),
268+
CodeWhispererModelConfigurator.getInstance().activeCustomization(project)?.arn,
269269
suggestion.suggestion.length,
270270
getUnmodifiedAcceptedCharsCount(suggestion.suggestion, modifiedSuggestion)
271271
)

0 commit comments

Comments
 (0)