fix(jest): add missing loadSession and saveSession mock

mybigday · Oct 20, 2023 · 80adc19 · 80adc19
1 parent eeb0e21
commit 80adc19
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/jest/mock.js b/jest/mock.js
@@ -144,6 +144,12 @@ if (!NativeModules.RNLlama) {
     detokenize: jest.fn(async () => ''),
     embedding: jest.fn(async () => []),
 
+    loadSession: jest.fn(async () => ({
+      tokens_loaded: 1,
+      prompt: 'Hello',
+    })),
+    saveSession: jest.fn(async () => 1),
+
     releaseContext: jest.fn(() => Promise.resolve()),
     releaseAllContexts: jest.fn(() => Promise.resolve()),