diff --git a/examples/serverless/src/shared.ts b/examples/serverless/src/shared.ts index a70df2a..36dc87e 100644 --- a/examples/serverless/src/shared.ts +++ b/examples/serverless/src/shared.ts @@ -27,7 +27,16 @@ export async function getModel( ): Promise { if (process.env.IS_OFFLINE === "true") { console.log("model offline retrieving"); - return path.resolve(__dirname, "../../../", `models/${modelName}`); + + const standardModelName = modelName.endsWith(".pass") + ? modelName + : `${modelName}.pass`; + + return path.resolve( + __dirname, + "../../../", + `models/${standardModelName}`, + ); } const s3 = await getS3Instance(); @@ -127,7 +136,7 @@ export async function getSpecificFileInModel( const model = await getModel(modelName); if (typeof model === "string") { - return fs.readFile(path.resolve(`${model}.pass`, fileName)); + return fs.readFile(path.resolve(model, fileName)); } return model[fileName];