Miscellaneous fixes

Author: akaashrp

examples/integrity-verification/package.json

@@ -15,6 +15,6 @@
     "url": "^0.11.3"
   },
   "dependencies": {
-    "@mlc-ai/web-llm": "^0.2.80"
+    "@mlc-ai/web-llm": "^0.2.82"
   }
 }

src/cache_util.ts

@@ -10,7 +10,7 @@ import { ModelNotFoundError, UnsupportedTokenizerFilesError } from "./error";
 import { Tokenizer } from "@mlc-ai/web-tokenizers";
 import { ModelIntegrity, verifyIntegrity } from "./integrity";
 
-async function verifyTokenizerIfNeeded(
+async function maybeVerifyTokenizerIntegrity(
   data: ArrayBuffer,
   filename: string,
   url: string,
@@ -147,7 +147,12 @@ export async function asyncLoadTokenizer(
   if (config.tokenizer_files.includes("tokenizer.json")) {
     const url = new URL("tokenizer.json", baseUrl).href;
     const model = await modelCache.fetchWithCache(url, "arraybuffer");
-    await verifyTokenizerIfNeeded(model, "tokenizer.json", url, integrity);
+    await maybeVerifyTokenizerIntegrity(
+      model,
+      "tokenizer.json",
+      url,
+      integrity,
+    );
     return Tokenizer.fromJSON(model);
   } else if (config.tokenizer_files.includes("tokenizer.model")) {
     logger(
@@ -159,7 +164,12 @@ export async function asyncLoadTokenizer(
     );
     const url = new URL("tokenizer.model", baseUrl).href;
     const model = await modelCache.fetchWithCache(url, "arraybuffer");
-    await verifyTokenizerIfNeeded(model, "tokenizer.model", url, integrity);
+    await maybeVerifyTokenizerIntegrity(
+      model,
+      "tokenizer.model",
+      url,
+      integrity,
+    );
     return Tokenizer.fromSentencePiece(model);
   }
   throw new UnsupportedTokenizerFilesError(config.tokenizer_files);

src/error.ts

@@ -614,14 +614,14 @@ export class ReloadModelIdNotUniqueError extends Error {
 
 export class IntegrityError extends Error {
   constructor(
-    public readonly url: string,
-    public readonly expected: string,
-    public readonly actual: string,
+    readonly url: string,
+    readonly expected: string,
+    readonly actual: string,
   ) {
     super(
       `Integrity verification failed for ${url}\n` +
         `  Expected: ${expected}\n` +
-        `  Got:      ${actual}\n` +
+        `  Actual: ${actual}\n` +
         `This may indicate file corruption or tampering.`,
     );
     this.name = "IntegrityError";