From 15c0a3881c4896acee0a672284acbf5593c5aba0 Mon Sep 17 00:00:00 2001 From: suyao Date: Thu, 27 Nov 2025 21:28:24 +0800 Subject: [PATCH] fix: test --- src/renderer/src/config/models/__tests__/utils.test.ts | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/renderer/src/config/models/__tests__/utils.test.ts b/src/renderer/src/config/models/__tests__/utils.test.ts index f3f4d402af..97dbf755d1 100644 --- a/src/renderer/src/config/models/__tests__/utils.test.ts +++ b/src/renderer/src/config/models/__tests__/utils.test.ts @@ -120,7 +120,7 @@ describe('model utils', () => { rerankMock.mockReturnValue(false) visionMock.mockReturnValue(true) textToImageMock.mockReturnValue(false) - generateImageMock.mockReturnValue(true) + generateImageMock.mockReturnValue(false) reasoningMock.mockReturnValue(false) openAIWebSearchOnlyMock.mockReturnValue(false) }) @@ -274,6 +274,7 @@ describe('model utils', () => { visionMock.mockReturnValueOnce(true).mockReturnValueOnce(false) expect(isVisionModels(models)).toBe(false) + generateImageMock.mockReturnValue(true) expect(isGenerateImageModels(models)).toBe(true) generateImageMock.mockReturnValueOnce(true).mockReturnValueOnce(false) expect(isGenerateImageModels(models)).toBe(false) @@ -292,6 +293,10 @@ describe('model utils', () => { rerankMock.mockReturnValue(false) textToImageMock.mockReturnValueOnce(true) expect(agentModelFilter(createModel({ id: 'gpt-image-1' }))).toBe(false) + + textToImageMock.mockReturnValue(false) + generateImageMock.mockReturnValueOnce(true) + expect(agentModelFilter(createModel({ id: 'dall-e-3' }))).toBe(false) }) it('identifies models with maximum temperature of 1.0', () => {