Skip to content

Commit 417370f

Browse files
authored
Merge pull request #6821 from atonaamz/master
feat(amazonq): passing referenceTrackerConfiguration to StartCodeFixJob
2 parents bd37885 + 3fb8a9e commit 417370f

File tree

4 files changed

+14
-1
lines changed

4 files changed

+14
-1
lines changed
Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
{
2+
"type": "Feature",
3+
"description": "/review: passing referenceTrackerConfiguration to StartCodeFixJob"
4+
}

packages/core/src/codewhisperer/client/user-service-2.json

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2206,7 +2206,8 @@
22062206
"uploadId": { "shape": "UploadId" },
22072207
"description": { "shape": "StartCodeFixJobRequestDescriptionString" },
22082208
"ruleId": { "shape": "StartCodeFixJobRequestRuleIdString" },
2209-
"codeFixName": { "shape": "CodeFixName" }
2209+
"codeFixName": { "shape": "CodeFixName" },
2210+
"referenceTrackerConfiguration": { "shape": "ReferenceTrackerConfiguration" }
22102211
}
22112212
},
22122213
"StartCodeFixJobRequestDescriptionString": {

packages/core/src/codewhisperer/commands/startCodeFixGeneration.ts

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@ import AdmZip from 'adm-zip'
1818
import path from 'path'
1919
import { TelemetryHelper } from '../util/telemetryHelper'
2020
import { tempDirPath } from '../../shared/filesystemUtilities'
21+
import { CodeWhispererSettings } from '../util/codewhispererSettings'
2122

2223
export async function startCodeFixGeneration(
2324
client: DefaultCodeWhispererClient,
@@ -69,6 +70,11 @@ export async function startCodeFixGeneration(
6970
end: { line: issue.endLine, character: 0 },
7071
},
7172
issue.recommendation.text,
73+
{
74+
recommendationsWithReferences: CodeWhispererSettings.instance.isSuggestionsWithCodeReferencesEnabled()
75+
? 'ALLOW'
76+
: 'BLOCK',
77+
},
7278
codeFixName,
7379
issue.ruleId
7480
)

packages/core/src/codewhisperer/service/codeFixHandler.ts

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,7 @@ export async function createCodeFixJob(
5050
uploadId: string,
5151
snippetRange: CodeWhispererUserClient.Range,
5252
description: string,
53+
referenceTrackerConfiguration: CodeWhispererUserClient.ReferenceTrackerConfiguration,
5354
codeFixName?: string,
5455
ruleId?: string
5556
) {
@@ -60,6 +61,7 @@ export async function createCodeFixJob(
6061
codeFixName,
6162
ruleId,
6263
description,
64+
referenceTrackerConfiguration,
6365
}
6466

6567
const resp = await client.startCodeFixJob(req).catch((err) => {

0 commit comments

Comments
 (0)