diff --git a/.changes/next-release/feature-21a9e26b-edbf-4ad3-a5f4-2662420d4f0d.json b/.changes/next-release/feature-21a9e26b-edbf-4ad3-a5f4-2662420d4f0d.json new file mode 100644 index 00000000000..b74af2af422 --- /dev/null +++ b/.changes/next-release/feature-21a9e26b-edbf-4ad3-a5f4-2662420d4f0d.json @@ -0,0 +1,4 @@ +{ + "type" : "bugfix", + "description" : "/review: Respect user option to allow code suggestions with references" +} diff --git a/plugins/amazonq/codewhisperer/jetbrains-community/src/software/aws/toolkits/jetbrains/services/codewhisperer/codescan/AmazonQCodeFixSession.kt b/plugins/amazonq/codewhisperer/jetbrains-community/src/software/aws/toolkits/jetbrains/services/codewhisperer/codescan/AmazonQCodeFixSession.kt index a1452f74cfa..899fbd030a7 100644 --- a/plugins/amazonq/codewhisperer/jetbrains-community/src/software/aws/toolkits/jetbrains/services/codewhisperer/codescan/AmazonQCodeFixSession.kt +++ b/plugins/amazonq/codewhisperer/jetbrains-community/src/software/aws/toolkits/jetbrains/services/codewhisperer/codescan/AmazonQCodeFixSession.kt @@ -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 @@ -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 @@ -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 {