From 676ac21804fa555561eb417242dd4997f90ad4cd Mon Sep 17 00:00:00 2001 From: SuYao Date: Thu, 17 Apr 2025 00:00:35 +0800 Subject: [PATCH] fix(GeminiProvider): update content configuration based on model type (#4960) * fix(GeminiProvider): update content configuration based on model type * chore(ApiService): comment out debug log for message output --- src/renderer/src/providers/AiProvider/GeminiProvider.ts | 4 ++-- src/renderer/src/services/ApiService.ts | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/renderer/src/providers/AiProvider/GeminiProvider.ts b/src/renderer/src/providers/AiProvider/GeminiProvider.ts index 40bdfc5e..e84f806f 100644 --- a/src/renderer/src/providers/AiProvider/GeminiProvider.ts +++ b/src/renderer/src/providers/AiProvider/GeminiProvider.ts @@ -232,8 +232,8 @@ export default class GeminiProvider extends BaseProvider { } const generateContentConfig: GenerateContentConfig = { - responseModalities: [Modality.TEXT, Modality.IMAGE], - responseMimeType: 'text/plain', + responseModalities: isGenerateImageModel(model) ? [Modality.TEXT, Modality.IMAGE] : undefined, + responseMimeType: isGenerateImageModel(model) ? 'text/plain' : undefined, safetySettings: this.getSafetySettings(model.id), // generate image don't need system instruction systemInstruction: isGemmaModel(model) || isGenerateImageModel(model) ? undefined : systemInstruction, diff --git a/src/renderer/src/services/ApiService.ts b/src/renderer/src/services/ApiService.ts index ff1e3457..f27ee6ba 100644 --- a/src/renderer/src/services/ApiService.ts +++ b/src/renderer/src/services/ApiService.ts @@ -282,7 +282,7 @@ export async function fetchChatCompletion({ } } - console.log('message', message) + // console.log('message', message) // Emit chat completion event EventEmitter.emit(EVENT_NAMES.RECEIVE_MESSAGE, message) onResponse(message)