fix: bug fix for exportResultsArchive to call with profileArn as parameter (#1300)

* fix: for exportResultsArchive to call with profileArn
---------

Co-authored-by: Pranav Firake <[email protected]>

Author: pranav-firake

server/aws-lsp-codewhisperer/src/language-server/netTransform/transformHandler.ts

@@ -379,7 +379,7 @@ export class TransformHandler {
     async downloadExportResultArchive(exportId: string, saveToDir: string) {
         let result
         try {
-            result = await this.serviceManager.getStreamingClient().client.exportResultArchive({
+            result = await this.serviceManager.getStreamingClient().exportResultArchive({
                 exportId,
                 exportIntent: ExportIntent.TRANSFORMATION,
             })

server/aws-lsp-codewhisperer/src/shared/streamingClientService.ts

@@ -4,6 +4,8 @@ import {
     GenerateAssistantResponseCommandOutput as GenerateAssistantResponseCommandOutputCodeWhispererStreaming,
     SendMessageCommandInput as SendMessageCommandInputCodeWhispererStreaming,
     SendMessageCommandOutput as SendMessageCommandOutputCodeWhispererStreaming,
+    ExportResultArchiveCommandInput as ExportResultArchiveCommandInputCodeWhispererStreaming,
+    ExportResultArchiveCommandOutput as ExportResultArchiveCommandOutputCodeWhispererStreaming,
 } from '@amzn/codewhisperer-streaming'
 import {
     QDeveloperStreaming,
@@ -122,6 +124,19 @@ export class StreamingClientServiceToken extends StreamingClientServiceBase {
 
         return response
     }
+
+    public async exportResultArchive(
+        request: ExportResultArchiveCommandInputCodeWhispererStreaming,
+        abortController?: AbortController
+    ): Promise<ExportResultArchiveCommandOutputCodeWhispererStreaming> {
+        const controller: AbortController = abortController ?? new AbortController()
+        this.inflightRequests.add(controller)
+        const response = await this.client.exportResultArchive(
+            this.profileArn ? { ...request, profileArn: this.profileArn } : request
+        )
+        this.inflightRequests.delete(controller)
+        return response
+    }
 }
 
 export class StreamingClientServiceIAM extends StreamingClientServiceBase {