fix(tools): better parse MIME types in tool outputs

package-lock.json

@@ -24,7 +24,7 @@
 				"date-fns": "^2.29.3",
 				"dotenv": "^16.0.3",
 				"express": "^4.19.2",
-				"file-type": "^19.0.0",
+				"file-type": "^19.4.1",
 				"google-auth-library": "^9.13.0",
 				"handlebars": "^4.7.8",
 				"highlight.js": "^11.7.0",
@@ -7438,13 +7438,13 @@
 			}
 		},
 		"node_modules/file-type": {
-			"version": "19.4.0",
-			"resolved": "https://registry.npmjs.org/file-type/-/file-type-19.4.0.tgz",
-			"integrity": "sha512-7N7Pu0UzVYV8YP6WXhN+kcvtp/P00eWKVo76nMAK+RasRO/ICzuJzjoG+aSLqbY6khDAFQuqsyImGaSdkm49Iw==",
+			"version": "19.4.1",
+			"resolved": "https://registry.npmjs.org/file-type/-/file-type-19.4.1.tgz",
+			"integrity": "sha512-RuWzwF2L9tCHS76KR/Mdh+DwJZcFCzrhrPXpOw6MlEfl/o31fjpTikzcKlYuyeV7e7ftdCGVJTNOCzkYD/aLbw==",
 			"license": "MIT",
 			"dependencies": {
 				"get-stream": "^9.0.1",
-				"strtok3": "^8.0.0",
+				"strtok3": "^8.1.0",
 				"token-types": "^6.0.0",
 				"uint8array-extras": "^1.3.0"
 			},
@@ -12267,9 +12267,9 @@
 			"license": "MIT"
 		},
 		"node_modules/strtok3": {
-			"version": "8.0.5",
-			"resolved": "https://registry.npmjs.org/strtok3/-/strtok3-8.0.5.tgz",
-			"integrity": "sha512-yybH4XOcYsQLSiPCRSqCdC6Nz6mtceq84y1RtBPfPSM2yKjeQKQDZGvQ2JwcT9jlCX1Eb5Gu0OqRhCfhOFhRvQ==",
+			"version": "8.1.0",
+			"resolved": "https://registry.npmjs.org/strtok3/-/strtok3-8.1.0.tgz",
+			"integrity": "sha512-ExzDvHYPj6F6QkSNe/JxSlBxTh3OrI6wrAIz53ulxo1c4hBJ1bT9C/JrAthEKHWG9riVH3Xzg7B03Oxty6S2Lw==",
 			"license": "MIT",
 			"dependencies": {
 				"@tokenizer/token": "^0.3.0",

package.json

@@ -68,13 +68,13 @@
 		"@playwright/browser-chromium": "^1.43.1",
 		"@resvg/resvg-js": "^2.6.2",
 		"autoprefixer": "^10.4.14",
-		"aws4": "^1.13.0",
 		"aws-sigv4-fetch": "^4.0.1",
+		"aws4": "^1.13.0",
 		"browser-image-resizer": "^2.4.1",
 		"date-fns": "^2.29.3",
 		"dotenv": "^16.0.3",
 		"express": "^4.19.2",
-		"file-type": "^19.0.0",
+		"file-type": "^19.4.1",
 		"google-auth-library": "^9.13.0",
 		"handlebars": "^4.7.8",
 		"highlight.js": "^11.7.0",

src/lib/server/tools/index.ts

@@ -26,6 +26,7 @@ import { sha256 } from "$lib/utils/sha256";
 import { ObjectId } from "mongodb";
 import { isValidOutputComponent, ToolOutputPaths } from "./outputs";
 import { downloadFile } from "../files/downloadFile";
+import { fileTypeFromBlob } from "file-type";
 
 export type BackendToolContext = Pick<
 	TextGenerationContext,
@@ -253,13 +254,13 @@ export function getCallMethod(tool: Omit<BaseTool, "call">): BackendCall {
 								await fetch(output)
 									.then((res) => res.blob())
 									.then(async (blob) => {
-										const mimeType = blob.type;
-										const fileType = blob.type.split("/")[1] ?? mimeType?.split("/")[1];
+										const { ext, mime } = (await fileTypeFromBlob(blob)) ?? { ext: "octet-stream" };
+
 										return new File(
 											[blob],
-											`${idx}-${await sha256(JSON.stringify(params))}.${fileType}`,
+											`${idx}-${await sha256(JSON.stringify(params))}.${ext}`,
 											{
-												type: fileType,
+												type: mime,
 											}
 										);
 									})