@@ -7,6 +7,7 @@ import com.intellij.openapi.Disposable
77import com.intellij.openapi.application.ApplicationManager
88import com.intellij.openapi.components.service
99import com.intellij.openapi.project.Project
10+ import com.intellij.util.text.nullize
1011import software.amazon.awssdk.auth.credentials.AnonymousCredentialsProvider
1112import software.amazon.awssdk.services.codewhisperer.CodeWhispererClient
1213import 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())
0 commit comments