Skip to content

Commit 814ad4d

Browse files
committed
Telemetry: UTG send plaintext instead of python in amazonq_utgGenerateTests event
1 parent 05a5492 commit 814ad4d

File tree

3 files changed

+6
-6
lines changed

3 files changed

+6
-6
lines changed

packages/core/src/amazonqTest/chat/controller/controller.ts

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -246,7 +246,7 @@ export class TestController {
246246
cwsprChatProgrammingLanguage: session.fileLanguage ?? 'plaintext',
247247
jobId: session.listOfTestGenerationJobId[0], // For RIV, UTG does only one StartTestGeneration API call
248248
jobGroup: session.testGenerationJobGroupName,
249-
requestId: session.requestId,
249+
requestId: session.startTestGenerationRequestId,
250250
hasUserPromptSupplied: session.hasUserPromptSupplied,
251251
isCodeBlockSelected: session.isCodeBlockSelected,
252252
buildPayloadBytes: session.srcPayloadSize,
@@ -726,7 +726,7 @@ export class TestController {
726726
cwsprChatProgrammingLanguage: session.fileLanguage ?? 'plaintext',
727727
jobId: session.listOfTestGenerationJobId[0], // For RIV, UTG does only one StartTestGeneration API call so jobId = session.listOfTestGenerationJobId[0]
728728
jobGroup: session.testGenerationJobGroupName,
729-
requestId: session.requestId,
729+
requestId: session.startTestGenerationRequestId,
730730
buildPayloadBytes: session.srcPayloadSize,
731731
buildZipFileBytes: session.srcZipFileSize,
732732
artifactsUploadDuration: session.artifactsUploadDuration,
@@ -850,7 +850,7 @@ export class TestController {
850850
cwsprChatProgrammingLanguage: session.fileLanguage ?? 'plaintext',
851851
jobId: session.listOfTestGenerationJobId[0], // For RIV, UTG does only one StartTestGeneration API call so jobId = session.listOfTestGenerationJobId[0]
852852
jobGroup: session.testGenerationJobGroupName,
853-
requestId: session.requestId,
853+
requestId: session.startTestGenerationRequestId,
854854
buildPayloadBytes: session.srcPayloadSize,
855855
buildZipFileBytes: session.srcZipFileSize,
856856
artifactsUploadDuration: session.artifactsUploadDuration,

packages/core/src/amazonqTest/chat/session/session.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ export class Session {
3131
//This is unique per each test generation cycle
3232
public testGenerationJobGroupName: string | undefined = undefined
3333
public listOfTestGenerationJobId: string[] = []
34-
public requestId: string | undefined = undefined
34+
public startTestGenerationRequestId: string | undefined = undefined
3535
public testGenerationJob: TestGenerationJob | undefined
3636

3737
// Start Test generation

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -94,13 +94,13 @@ export async function createTestJob(
9494
logger.debug('target line range end: %O', firstTargetLineRangeList?.end)
9595

9696
const resp = await codewhispererClient.codeWhispererClient.startTestGeneration(req).catch((err) => {
97-
ChatSessionManager.Instance.getSession().requestId = err.requestId
97+
ChatSessionManager.Instance.getSession().startTestGenerationRequestId = err.requestId
9898
logger.error(`Failed creating test job. Request id: ${err.requestId}`)
9999
throw err
100100
})
101101
logger.info('Unit test generation request id: %s', resp.$response.requestId)
102102
logger.debug('Unit test generation data: %O', resp.$response.data)
103-
ChatSessionManager.Instance.getSession().requestId = resp.$response.requestId
103+
ChatSessionManager.Instance.getSession().startTestGenerationRequestId = resp.$response.requestId
104104
if (resp.$response.error) {
105105
logger.error('Unit test generation error: %O', resp.$response.error)
106106
}

0 commit comments

Comments
 (0)