From 8385caf3812736ad3f2d379edea3d978d44ea8f9 Mon Sep 17 00:00:00 2001
From: Harshdeep Singh <6162866+harsh62@users.noreply.github.com>
Date: Mon, 17 Mar 2025 12:33:42 -0400
Subject: [PATCH] fix(storage): clean up cached files upon completion

---
 .../Support/Internal/StorageMultipartUploadClient.swift       | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/AmplifyPlugins/Storage/Sources/AWSS3StoragePlugin/Support/Internal/StorageMultipartUploadClient.swift b/AmplifyPlugins/Storage/Sources/AWSS3StoragePlugin/Support/Internal/StorageMultipartUploadClient.swift
index ac1baacaa6..b6d547de0a 100644
--- a/AmplifyPlugins/Storage/Sources/AWSS3StoragePlugin/Support/Internal/StorageMultipartUploadClient.swift
+++ b/AmplifyPlugins/Storage/Sources/AWSS3StoragePlugin/Support/Internal/StorageMultipartUploadClient.swift
@@ -146,6 +146,10 @@ class DefaultStorageMultipartUploadClient: StorageMultipartUploadClient {
             Task {
                 do {
                     let partialFileURL = try result.get()
+                    defer {
+                        // Clean up the temporary file after we're done with it
+                        self.fileSystem.removeFileIfExists(fileURL: partialFileURL)
+                    }
 
                     let operation = AWSS3SigningOperation.uploadPart(partNumber: partNumber, uploadId: uploadId)
                     let preSignedURL = try await serviceProxy.preSignedURLBuilder.getPreSignedURL(