Magic-Mirror/gecko-dev
1
0
Fork 0
mirror of https://github.com/mozilla/gecko-dev.git synced 2024-12-03 02:25:34 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Bug 1886675 - store the files as blobs r=gregtatum

Browse Source 
Depends on D205290

Differential Revision: https://phabricator.services.mozilla.com/D205308
This commit is contained in:
Tarek Ziadé 2024-03-25 16:57:35 +00:00
parent b7ea7b37c3
commit 80b6e8b58a
1 changed files with 24 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
toolkit/components/ml/content/ModelHub.sys.mjs
View File

@ -538,13 +538,13 @@ export class ModelHub {
modelVersion,
file,
}) {
const [arrayBuffer, headers] = await this.getModelFileAsArrayBuffer({
const [blob, headers] = await this.getModelFileAsBlob({
organization,
modelName,
modelVersion,
file,
});
return new Response(arrayBuffer, { headers });
return new Response(blob, { headers });
}
/**
@ -563,6 +563,26 @@ export class ModelHub {
modelVersion,
file,
}) {
const [blob, headers] = await this.getModelFileAsBlob({
organization,
modelName,
modelVersion,
file,
});
return [await blob.arrayBuffer(), headers];
}
/**
* Given an organization, model, and version, fetch a model file in the hub as blob.
*
* @param {object} config
* @param {string} config.organization
* @param {string} config.modelName
* @param {string} config.modelVersion
* @param {string} config.file
* @returns {Promise<[Blob, object]>} The file content
*/
async getModelFileAsBlob({ organization, modelName, modelVersion, file }) {
// Make sure inputs are clean. We don't sanitize them but throw an exception
let checkError = this.#checkInput(
organization,
@ -623,10 +643,10 @@ export class ModelHub {
modelName,
modelVersion,
file,
await clone.arrayBuffer(),
await clone.blob(),
headers
);
return [await response.arrayBuffer(), headers];
return [await response.blob(), headers];
}
} catch (error) {
lazy.console.error(`Failed to fetch ${url}:`, error);