diff --git a/apps/sim/lib/copilot/tools/server/files/edit-content.ts b/apps/sim/lib/copilot/tools/server/files/edit-content.ts
index 22d9539ba88..2bcaaa2be7a 100644
--- a/apps/sim/lib/copilot/tools/server/files/edit-content.ts
+++ b/apps/sim/lib/copilot/tools/server/files/edit-content.ts
@@ -201,9 +201,10 @@ export const editContentServerTool: BaseServerTool<EditContentArgs, EditContentR
           return { success: false, message: `Unsupported operation in intent: ${operation}` }
       }
 
+      let fileBuffer: Buffer
       if (docInfo.isDoc) {
         try {
-          await runSandboxTask(
+          fileBuffer = await runSandboxTask(
             docInfo.taskId!,
             { code: finalContent, workspaceId },
             { ownerKey: `user:${context.userId}`, signal: context.abortSignal }
@@ -215,9 +216,9 @@ export const editContentServerTool: BaseServerTool<EditContentArgs, EditContentR
             message: `${docInfo.formatName} generation failed: ${msg}. Fix the content and retry.`,
           }
         }
+      } else {
+        fileBuffer = Buffer.from(finalContent, 'utf-8')
       }
-
-      const fileBuffer = Buffer.from(finalContent, 'utf-8')
       assertServerToolNotAborted(context)
       const mime = docInfo.sourceMime || inferContentType(fileRecord.name, intent.contentType)
       await updateWorkspaceFileContent(workspaceId, intent.fileId, context.userId, fileBuffer, mime)
diff --git a/apps/sim/lib/copilot/tools/server/files/workspace-file.ts b/apps/sim/lib/copilot/tools/server/files/workspace-file.ts
index 5bbe8e3ddd8..35d0d91dd65 100644
--- a/apps/sim/lib/copilot/tools/server/files/workspace-file.ts
+++ b/apps/sim/lib/copilot/tools/server/files/workspace-file.ts
@@ -199,9 +199,10 @@ export const workspaceFileServerTool: BaseServerTool<WorkspaceFileArgs, Workspac
 
           const docInfo = getDocumentFormatInfo(fileName)
           let contentType = inferContentType(fileName, explicitType)
+          let fileBuffer: Buffer
           if (docInfo.isDoc) {
             try {
-              await runSandboxTask(
+              fileBuffer = await runSandboxTask(
                 docInfo.taskId!,
                 { code: content, workspaceId },
                 { ownerKey: `user:${context.userId}`, signal: context.abortSignal }
@@ -214,9 +215,9 @@ export const workspaceFileServerTool: BaseServerTool<WorkspaceFileArgs, Workspac
               }
             }
             contentType = docInfo.sourceMime!
+          } else {
+            fileBuffer = Buffer.from(content, 'utf-8')
           }
-
-          const fileBuffer = Buffer.from(content, 'utf-8')
           assertServerToolNotAborted(context)
           const result = await uploadWorkspaceFile(
             workspaceId,