@@ -14,6 +14,7 @@ import software.amazon.awssdk.services.codewhispererruntime.model.GetCodeFixJobR
1414import software.amazon.awssdk.services.codewhispererruntime.model.GetCodeFixJobResponse
1515import software.amazon.awssdk.services.codewhispererruntime.model.Position
1616import software.amazon.awssdk.services.codewhispererruntime.model.Range
17+ import software.amazon.awssdk.services.codewhispererruntime.model.RecommendationsWithReferencesPreference
1718import software.amazon.awssdk.services.codewhispererruntime.model.StartCodeFixJobRequest
1819import software.amazon.awssdk.services.codewhispererruntime.model.StartCodeFixJobResponse
1920import software.amazon.awssdk.services.codewhispererruntime.model.UploadContext
@@ -28,6 +29,7 @@ import software.aws.toolkits.jetbrains.services.codewhisperer.credentials.CodeWh
2829import software.aws.toolkits.jetbrains.services.codewhisperer.util.CodeWhispererConstants
2930import software.aws.toolkits.jetbrains.services.codewhisperer.util.CodeWhispererZipUploadManager
3031import software.aws.toolkits.jetbrains.services.codewhisperer.util.getTelemetryErrorMessage
32+ import software.aws.toolkits.jetbrains.settings.CodeWhispererSettings
3133import software.aws.toolkits.resources.message
3234import java.io.File
3335import java.nio.file.Path
@@ -161,12 +163,19 @@ class AmazonQCodeFixSession(val project: Project) {
161163 codeFixName : String? = null,
162164 ruleId : String? = null,
163165 ): StartCodeFixJobResponse {
166+ val includeCodeWithReference = if (CodeWhispererSettings .getInstance().isIncludeCodeWithReference()) {
167+ RecommendationsWithReferencesPreference .ALLOW
168+ } else {
169+ RecommendationsWithReferencesPreference .BLOCK
170+ }
171+
164172 val request = StartCodeFixJobRequest .builder()
165173 .uploadId(uploadId)
166174 .snippetRange(snippetRange)
167175 .codeFixName(codeFixName)
168176 .ruleId(ruleId)
169177 .description(description)
178+ .referenceTrackerConfiguration { it.recommendationsWithReferences(includeCodeWithReference) }
170179 .build()
171180
172181 return try {
0 commit comments