test: fix tests

Author: giladgd

src/bindings/types.ts

@@ -103,13 +103,17 @@ export enum LlamaVocabularyType {
     bpe = "bpe",
     wpm = "wpm",
     ugm = "ugm",
-    rwkv = "rwkv"
+    rwkv = "rwkv",
+    plamo2 = "plamo2"
 }
 export const LlamaVocabularyTypeValues = Object.freeze([
     LlamaVocabularyType.none,
     LlamaVocabularyType.spm,
     LlamaVocabularyType.bpe,
-    LlamaVocabularyType.wpm
+    LlamaVocabularyType.wpm,
+    LlamaVocabularyType.ugm,
+    LlamaVocabularyType.rwkv,
+    LlamaVocabularyType.plamo2
 ] as const);
 
 /**

test/modelDependent/bgeReranker/rank.test.ts

@@ -5,7 +5,7 @@ import {getTestLlama} from "../../utils/getTestLlama.js";
 describe("bgeReranker", () => {
     describe("rank", () => {
         test("simple ranking", {timeout: 1000 * 60 * 60 * 2}, async (test) => {
-            if (process.platform !== "darwin")
+            if (process.platform !== "darwin" && process.arch !== "arm64")
                 test.skip(); // the scores are a bit different on different platforms, so skipping on other platforms due to flakiness
 
             const modelPath = await getModelFile("bge-reranker-v2-m3-Q8_0.gguf");
@@ -61,7 +61,7 @@ describe("bgeReranker", () => {
         });
 
         test("rank all", {timeout: 1000 * 60 * 60 * 2}, async (test) => {
-            if (process.platform !== "darwin")
+            if (process.platform !== "darwin" && process.arch !== "arm64")
                 test.skip(); // the scores are a bit different on different platforms, so skipping on other platforms due to flakiness
 
             const modelPath = await getModelFile("bge-reranker-v2-m3-Q8_0.gguf");
@@ -115,7 +115,7 @@ describe("bgeReranker", () => {
         });
 
         test("rank and sort", {timeout: 1000 * 60 * 60 * 2}, async (test) => {
-            if (process.platform !== "darwin")
+            if (process.platform !== "darwin" && process.arch !== "arm64")
                 test.skip(); // the scores are a bit different on different platforms, so skipping on other platforms due to flakiness
 
             const modelPath = await getModelFile("bge-reranker-v2-m3-Q8_0.gguf");