@@ -14,6 +14,7 @@ import software.amazon.awssdk.services.codewhispererruntime.model.GetCodeFixJobR
14
14
import software.amazon.awssdk.services.codewhispererruntime.model.GetCodeFixJobResponse
15
15
import software.amazon.awssdk.services.codewhispererruntime.model.Position
16
16
import software.amazon.awssdk.services.codewhispererruntime.model.Range
17
+ import software.amazon.awssdk.services.codewhispererruntime.model.RecommendationsWithReferencesPreference
17
18
import software.amazon.awssdk.services.codewhispererruntime.model.StartCodeFixJobRequest
18
19
import software.amazon.awssdk.services.codewhispererruntime.model.StartCodeFixJobResponse
19
20
import software.amazon.awssdk.services.codewhispererruntime.model.UploadContext
@@ -28,6 +29,7 @@ import software.aws.toolkits.jetbrains.services.codewhisperer.credentials.CodeWh
28
29
import software.aws.toolkits.jetbrains.services.codewhisperer.util.CodeWhispererConstants
29
30
import software.aws.toolkits.jetbrains.services.codewhisperer.util.CodeWhispererZipUploadManager
30
31
import software.aws.toolkits.jetbrains.services.codewhisperer.util.getTelemetryErrorMessage
32
+ import software.aws.toolkits.jetbrains.settings.CodeWhispererSettings
31
33
import software.aws.toolkits.resources.message
32
34
import java.io.File
33
35
import java.nio.file.Path
@@ -161,12 +163,19 @@ class AmazonQCodeFixSession(val project: Project) {
161
163
codeFixName : String? = null,
162
164
ruleId : String? = null,
163
165
): StartCodeFixJobResponse {
166
+ val includeCodeWithReference = if (CodeWhispererSettings .getInstance().isIncludeCodeWithReference()) {
167
+ RecommendationsWithReferencesPreference .ALLOW
168
+ } else {
169
+ RecommendationsWithReferencesPreference .BLOCK
170
+ }
171
+
164
172
val request = StartCodeFixJobRequest .builder()
165
173
.uploadId(uploadId)
166
174
.snippetRange(snippetRange)
167
175
.codeFixName(codeFixName)
168
176
.ruleId(ruleId)
169
177
.description(description)
178
+ .referenceTrackerConfiguration { it.recommendationsWithReferences(includeCodeWithReference) }
170
179
.build()
171
180
172
181
return try {
0 commit comments