fix(ts): missing mock methods

mybigday · Dec 20, 2024 · a1d5413 · a1d5413
1 parent 361e673
commit a1d5413
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/jest/mock.js b/jest/mock.js
@@ -2,10 +2,15 @@ const { NativeModules, DeviceEventEmitter } = require('react-native')
 
 if (!NativeModules.RNLlama) {
   NativeModules.RNLlama = {
+    setContextLimit: jest.fn(),
+
+    modelInfo: jest.fn(async () => ({})),
+
     initContext: jest.fn(() =>
       Promise.resolve({
         gpu: false,
         reasonNoGPU: 'Test',
+        model: {},
       }),
     ),