From 8505d7baed46a2bb3754018ee8c69eba28e35f9f Mon Sep 17 00:00:00 2001
From: Aidan Ton <atona@amazon.com>
Date: Wed, 19 Mar 2025 17:23:05 -0700
Subject: [PATCH] /review: passing referenceTrackerConfiguration to
 StartCodeFixJob

---
 .../Feature-25bf4b40-331e-4b28-8a89-0a19075b6b79.json       | 4 ++++
 packages/core/src/codewhisperer/client/user-service-2.json  | 3 ++-
 .../src/codewhisperer/commands/startCodeFixGeneration.ts    | 6 ++++++
 packages/core/src/codewhisperer/service/codeFixHandler.ts   | 2 ++
 4 files changed, 14 insertions(+), 1 deletion(-)
 create mode 100644 packages/amazonq/.changes/next-release/Feature-25bf4b40-331e-4b28-8a89-0a19075b6b79.json

diff --git a/packages/amazonq/.changes/next-release/Feature-25bf4b40-331e-4b28-8a89-0a19075b6b79.json b/packages/amazonq/.changes/next-release/Feature-25bf4b40-331e-4b28-8a89-0a19075b6b79.json
new file mode 100644
index 00000000000..911f0babb24
--- /dev/null
+++ b/packages/amazonq/.changes/next-release/Feature-25bf4b40-331e-4b28-8a89-0a19075b6b79.json
@@ -0,0 +1,4 @@
+{
+	"type": "Feature",
+	"description": "/review: passing referenceTrackerConfiguration to StartCodeFixJob"
+}
diff --git a/packages/core/src/codewhisperer/client/user-service-2.json b/packages/core/src/codewhisperer/client/user-service-2.json
index ed7661bec4c..833245ef183 100644
--- a/packages/core/src/codewhisperer/client/user-service-2.json
+++ b/packages/core/src/codewhisperer/client/user-service-2.json
@@ -2206,7 +2206,8 @@
                 "uploadId": { "shape": "UploadId" },
                 "description": { "shape": "StartCodeFixJobRequestDescriptionString" },
                 "ruleId": { "shape": "StartCodeFixJobRequestRuleIdString" },
-                "codeFixName": { "shape": "CodeFixName" }
+                "codeFixName": { "shape": "CodeFixName" },
+                "referenceTrackerConfiguration": { "shape": "ReferenceTrackerConfiguration" }
             }
         },
         "StartCodeFixJobRequestDescriptionString": {
diff --git a/packages/core/src/codewhisperer/commands/startCodeFixGeneration.ts b/packages/core/src/codewhisperer/commands/startCodeFixGeneration.ts
index a255864c7eb..8e9a5240812 100644
--- a/packages/core/src/codewhisperer/commands/startCodeFixGeneration.ts
+++ b/packages/core/src/codewhisperer/commands/startCodeFixGeneration.ts
@@ -18,6 +18,7 @@ import AdmZip from 'adm-zip'
 import path from 'path'
 import { TelemetryHelper } from '../util/telemetryHelper'
 import { tempDirPath } from '../../shared/filesystemUtilities'
+import { CodeWhispererSettings } from '../util/codewhispererSettings'
 
 export async function startCodeFixGeneration(
     client: DefaultCodeWhispererClient,
@@ -69,6 +70,11 @@ export async function startCodeFixGeneration(
                 end: { line: issue.endLine, character: 0 },
             },
             issue.recommendation.text,
+            {
+                recommendationsWithReferences: CodeWhispererSettings.instance.isSuggestionsWithCodeReferencesEnabled()
+                    ? 'ALLOW'
+                    : 'BLOCK',
+            },
             codeFixName,
             issue.ruleId
         )
diff --git a/packages/core/src/codewhisperer/service/codeFixHandler.ts b/packages/core/src/codewhisperer/service/codeFixHandler.ts
index 854cb848a52..4aa74a91ac7 100644
--- a/packages/core/src/codewhisperer/service/codeFixHandler.ts
+++ b/packages/core/src/codewhisperer/service/codeFixHandler.ts
@@ -50,6 +50,7 @@ export async function createCodeFixJob(
     uploadId: string,
     snippetRange: CodeWhispererUserClient.Range,
     description: string,
+    referenceTrackerConfiguration: CodeWhispererUserClient.ReferenceTrackerConfiguration,
     codeFixName?: string,
     ruleId?: string
 ) {
@@ -60,6 +61,7 @@ export async function createCodeFixJob(
         codeFixName,
         ruleId,
         description,
+        referenceTrackerConfiguration,
     }
 
     const resp = await client.startCodeFixJob(req).catch((err) => {