diff --git a/tests/frontend/components/modelMetadata.renamePath.test.js b/tests/frontend/components/modelMetadata.renamePath.test.js index e5e2f764..ef3b6b41 100644 --- a/tests/frontend/components/modelMetadata.renamePath.test.js +++ b/tests/frontend/components/modelMetadata.renamePath.test.js @@ -89,6 +89,14 @@ vi.mock(I18N_HELPERS_MODULE, () => ({ translate: vi.fn((_, __, fallback) => fallback || ''), })); +vi.mock('../../../static/js/api/apiConfig.js', () => ({ + MODEL_TYPES: { + LORA: 'loras', + CHECKPOINT: 'checkpoints', + EMBEDDING: 'embeddings' + } +})); + vi.mock(API_FACTORY, () => ({ getModelApiClient: vi.fn(), })); diff --git a/tests/frontend/components/modelModal.licenseIcons.test.js b/tests/frontend/components/modelModal.licenseIcons.test.js index dfd1d837..7a9ad1d7 100644 --- a/tests/frontend/components/modelModal.licenseIcons.test.js +++ b/tests/frontend/components/modelModal.licenseIcons.test.js @@ -87,6 +87,14 @@ vi.mock(I18N_HELPERS_MODULE, () => ({ translate: vi.fn((_, __, fallback) => fallback || ''), })); +vi.mock('../../../static/js/api/apiConfig.js', () => ({ + MODEL_TYPES: { + LORA: 'loras', + CHECKPOINT: 'checkpoints', + EMBEDDING: 'embeddings' + } +})); + vi.mock(API_FACTORY, () => ({ getModelApiClient: vi.fn(), }));