Skip to content
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
{
"type" : "feature",
"description" : "/review: passing referenceTrackerConfiguration to StartCodeFixJob"
}
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@ import software.amazon.awssdk.services.codewhispererruntime.model.GetCodeFixJobR
import software.amazon.awssdk.services.codewhispererruntime.model.GetCodeFixJobResponse
import software.amazon.awssdk.services.codewhispererruntime.model.Position
import software.amazon.awssdk.services.codewhispererruntime.model.Range
import software.amazon.awssdk.services.codewhispererruntime.model.RecommendationsWithReferencesPreference
import software.amazon.awssdk.services.codewhispererruntime.model.StartCodeFixJobRequest
import software.amazon.awssdk.services.codewhispererruntime.model.StartCodeFixJobResponse
import software.amazon.awssdk.services.codewhispererruntime.model.UploadContext
Expand All @@ -28,6 +29,7 @@ import software.aws.toolkits.jetbrains.services.codewhisperer.credentials.CodeWh
import software.aws.toolkits.jetbrains.services.codewhisperer.util.CodeWhispererConstants
import software.aws.toolkits.jetbrains.services.codewhisperer.util.CodeWhispererZipUploadManager
import software.aws.toolkits.jetbrains.services.codewhisperer.util.getTelemetryErrorMessage
import software.aws.toolkits.jetbrains.settings.CodeWhispererSettings
import software.aws.toolkits.resources.message
import java.io.File
import java.nio.file.Path
Expand Down Expand Up @@ -161,12 +163,19 @@ class AmazonQCodeFixSession(val project: Project) {
codeFixName: String? = null,
ruleId: String? = null,
): StartCodeFixJobResponse {
val includeCodeWithReference = if (CodeWhispererSettings.getInstance().isIncludeCodeWithReference()) {
RecommendationsWithReferencesPreference.ALLOW
} else {
RecommendationsWithReferencesPreference.BLOCK
}

val request = StartCodeFixJobRequest.builder()
.uploadId(uploadId)
.snippetRange(snippetRange)
.codeFixName(codeFixName)
.ruleId(ruleId)
.description(description)
.referenceTrackerConfiguration { it.recommendationsWithReferences(includeCodeWithReference) }
.build()

return try {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -2082,7 +2082,8 @@
"uploadId": { "shape": "UploadId" },
"description": { "shape": "StartCodeFixJobRequestDescriptionString" },
"ruleId": { "shape": "StartCodeFixJobRequestRuleIdString" },
"codeFixName": { "shape": "CodeFixName" }
"codeFixName": { "shape": "CodeFixName" },
"referenceTrackerConfiguration": { "shape": "ReferenceTrackerConfiguration" }
}
},
"StartCodeFixJobRequestDescriptionString": {
Expand Down
Loading