diff --git a/src/renderer/src/windows/selection/action/components/__tests__/ActionUtils.test.ts b/src/renderer/src/windows/selection/action/components/__tests__/ActionUtils.test.ts index d6b34021b8..d48e6cb665 100644 --- a/src/renderer/src/windows/selection/action/components/__tests__/ActionUtils.test.ts +++ b/src/renderer/src/windows/selection/action/components/__tests__/ActionUtils.test.ts @@ -10,6 +10,14 @@ vi.mock('@renderer/services/ApiService', () => ({ fetchChatCompletion: vi.fn() })) +vi.mock('@renderer/services/ConversationService', () => ({ + ConversationService: class { + static async prepareMessagesForModel() { + // do nothing + } + } +})) + vi.mock('@renderer/services/MessagesService', () => ({ getUserMessage: vi.fn(), getAssistantMessage: vi.fn() @@ -50,6 +58,20 @@ vi.mock('@renderer/utils/messageUtils/create', () => ({ createErrorBlock: vi.fn() })) +vi.mock('@renderer/config/models', () => ({ + SYSTEM_MODELS: { + defaultModel: [ + { id: 'gpt-4', name: 'GPT-4' }, + { id: 'gpt-4', name: 'GPT-4' }, + { id: 'gpt-4', name: 'GPT-4' } + ], + silicon: [], + openai: [], + anthropic: [], + gemini: [] + } +})) + // Import mocked modules import { fetchChatCompletion } from '@renderer/services/ApiService' import { getAssistantMessage, getUserMessage } from '@renderer/services/MessagesService'