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)