feat: add content disposition to file endpoint (#1500)

* feat: add content disposition to file endpoint

* fix: use `mime-types` to get extension

package-lock.json

@@ -69,6 +69,7 @@
 				"@types/js-yaml": "^4.0.9",
 				"@types/jsdom": "^21.1.1",
 				"@types/jsonpath": "^0.2.4",
+				"@types/mime-types": "^2.1.4",
 				"@types/minimist": "^1.2.5",
 				"@types/node": "^22.1.0",
 				"@types/parquetjs": "^0.10.3",
@@ -4308,6 +4309,13 @@
 			"integrity": "sha512-/pyBZWSLD2n0dcHE3hq8s8ZvcETHtEuF+3E7XVt0Ig2nvsVQXdghHVcEkIWjy9A0wKfTn97a/PSDYohKIlnP/w==",
 			"dev": true
 		},
+		"node_modules/@types/mime-types": {
+			"version": "2.1.4",
+			"resolved": "https://registry.npmjs.org/@types/mime-types/-/mime-types-2.1.4.tgz",
+			"integrity": "sha512-lfU4b34HOri+kAY5UheuFMWPDOI+OPceBSHZKp69gEyTL/mmJ4cnU6Y/rlme3UL3GyOn6Y42hyIEw0/q8sWx5w==",
+			"dev": true,
+			"license": "MIT"
+		},
 		"node_modules/@types/minimist": {
 			"version": "1.2.5",
 			"resolved": "https://registry.npmjs.org/@types/minimist/-/minimist-1.2.5.tgz",

package.json

@@ -28,6 +28,7 @@
 		"@types/js-yaml": "^4.0.9",
 		"@types/jsdom": "^21.1.1",
 		"@types/jsonpath": "^0.2.4",
+		"@types/mime-types": "^2.1.4",
 		"@types/minimist": "^1.2.5",
 		"@types/node": "^22.1.0",
 		"@types/parquetjs": "^0.10.3",
@@ -108,9 +109,9 @@
 		"zod": "^3.22.3"
 	},
 	"optionalDependencies": {
-		"@aws-sdk/client-bedrock-runtime": "^3.631.0",
 		"@anthropic-ai/sdk": "^0.25.0",
 		"@anthropic-ai/vertex-sdk": "^0.4.1",
+		"@aws-sdk/client-bedrock-runtime": "^3.631.0",
 		"@google-cloud/vertexai": "^1.1.0",
 		"@google/generative-ai": "^0.14.1",
 		"aws4fetch": "^1.0.17",

src/routes/conversation/[id]/output/[sha256]/+server.ts

@@ -5,6 +5,7 @@ import { ObjectId } from "mongodb";
 import { z } from "zod";
 import type { RequestHandler } from "./$types";
 import { downloadFile } from "$lib/server/files/downloadFile";
+import mimeTypes from "mime-types";
 
 export const GET: RequestHandler = async ({ locals, params }) => {
 	const sha256 = z.string().parse(params.sha256);
@@ -47,6 +48,9 @@ export const GET: RequestHandler = async ({ locals, params }) => {
 			"Content-Type": mime ?? "application/octet-stream",
 			"Content-Security-Policy":
 				"default-src 'none'; script-src 'none'; style-src 'none'; sandbox;",
+			"Content-Disposition": `attachment; filename="${sha256.slice(0, 8)}.${
+				mime ? mimeTypes.extension(mime) || "bin" : "bin"
+			}"`,
 			"Content-Length": b64Value.length.toString(),
 			"Accept-Range": "bytes",
 		},