Fix CI test failed.

Author: daviddaytw

jest.config.js

@@ -1,6 +1,10 @@
 module.exports = {
   preset: 'react-native',
   moduleFileExtensions: ['ts', 'tsx', 'js', 'jsx', 'json', 'node'],
+  modulePathIgnorePatterns: [
+    '<rootDir>/example/node_modules',
+    '<rootDir>/lib/',
+  ],
   transformIgnorePatterns: [
     'node_modules/(?!(' +
       'react-native|' +

package.json

@@ -94,13 +94,6 @@
     "example"
   ],
   "packageManager": "yarn@3.6.1",
-  "jest": {
-    "preset": "react-native",
-    "modulePathIgnorePatterns": [
-      "<rootDir>/example/node_modules",
-      "<rootDir>/lib/"
-    ]
-  },
   "commitlint": {
     "extends": [
       "@commitlint/config-conventional"

src/__tests__/__mocks__/@huggingface/transformers.js

@@ -0,0 +1,15 @@
+module.exports = {
+  env: { allowRemoteModels: true, allowLocalModels: false },
+  AutoTokenizer: {
+    from_pretrained: jest.fn().mockResolvedValue(
+      Object.assign(
+        jest.fn((_text, _options) => ({ input_ids: [1, 2, 3, 4] })),
+        {
+          decode: jest.fn((_tokens, _options) => 'decoded text'),
+          encode: jest.fn((_text, _options) => ({ input_ids: [1, 2, 3, 4] })),
+          call: jest.fn((_text, _options) => ({ input_ids: [1, 2, 3, 4] })),
+        }
+      )
+    ),
+  },
+};

src/__tests__/setup.js

@@ -44,13 +44,3 @@ jest.mock('onnxruntime-react-native', () => ({
     dispose: jest.fn(),
   })),
 }));
-
-// Mock transformers
-jest.mock('@huggingface/transformers', () => ({
-  env: { allowRemoteModels: true, allowLocalModels: false },
-  AutoTokenizer: {
-    from_pretrained: jest.fn().mockResolvedValue({
-      decode: jest.fn((_tokens, _options) => 'decoded text'),
-    }),
-  },
-}));

src/__tests__/text-embedding.pipeline.test.tsx

@@ -17,13 +17,6 @@ const createCallableTokenizer = () => {
   return tokenizer;
 };
 
-jest.mock('@huggingface/transformers', () => ({
-  env: { allowRemoteModels: true, allowLocalModels: false },
-  AutoTokenizer: {
-    from_pretrained: jest.fn().mockResolvedValue(createCallableTokenizer()),
-  },
-}));
-
 describe('TextEmbedding Pipeline', () => {
   beforeEach(() => {
     jest.clearAllMocks();

src/__tests__/text-generation.pipeline.test.tsx

@@ -1,25 +1,4 @@
 import TextGenerationPipeline from '../pipelines/text-generation';
-import type { PreTrainedTokenizer } from '@huggingface/transformers';
-
-// Mock the transformers library
-jest.mock('@huggingface/transformers', () => {
-  // Create a mock tokenizer function with the correct type
-  const mockTokenizerFn = Object.assign(
-    jest
-      .fn<Promise<{ input_ids: bigint[] }>, [string, any]>()
-      .mockResolvedValue({ input_ids: [1n, 2n] }),
-    {
-      decode: jest.fn((_tokens: bigint[], _options: unknown) => 'decoded text'),
-    }
-  ) as unknown as PreTrainedTokenizer;
-
-  return {
-    env: { allowRemoteModels: true, allowLocalModels: false },
-    AutoTokenizer: {
-      from_pretrained: jest.fn().mockResolvedValue(mockTokenizerFn),
-    },
-  };
-});
 
 // Mock the model
 jest.mock('../models/text-generation', () => {