fix: replace test models with lightweight mocks to resolve module errors (#1164)

This commit is contained in:
josh 2025-09-08 16:27:32 +01:00 committed by GitHub
parent 60b09d170c
commit 73dae0d10f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 39 additions and 1 deletions

38
lib/ai/models.mock.ts Normal file
View file

@ -0,0 +1,38 @@
import { type LanguageModel } from 'ai';
const createMockModel = (): LanguageModel => {
return {
specificationVersion: 'v2',
provider: 'mock',
modelId: 'mock-model',
defaultObjectGenerationMode: 'tool',
supportedUrls: [],
supportsImageUrls: false,
supportsStructuredOutputs: false,
doGenerate: async () => ({
rawCall: { rawPrompt: null, rawSettings: {} },
finishReason: 'stop',
usage: { inputTokens: 10, outputTokens: 20, totalTokens: 30 },
content: [{ type: 'text', text: 'Hello, world!' }],
warnings: [],
}),
doStream: async () => ({
stream: new ReadableStream({
start(controller) {
controller.enqueue({
type: 'text-delta',
id: 'mock-id',
delta: 'Mock response',
});
controller.close();
},
}),
rawCall: { rawPrompt: null, rawSettings: {} },
}),
} as unknown as LanguageModel;
};
export const chatModel = createMockModel();
export const reasoningModel = createMockModel();
export const titleModel = createMockModel();
export const artifactModel = createMockModel();