whai/cherry-studio
1
1
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix(GeminiProvider): update content configuration based on model type (#4960)

Browse Source 
* fix(GeminiProvider): update content configuration based on model type

* chore(ApiService): comment out debug log for message output
This commit is contained in:
SuYao 2025-04-17 00:00:35 +08:00 committed by GitHub
parent 24ddd69cd5
commit 676ac21804
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/renderer/src/providers/AiProvider/GeminiProvider.ts
View File

@ -232,8 +232,8 @@ export default class GeminiProvider extends BaseProvider {
} }
const generateContentConfig: GenerateContentConfig = { const generateContentConfig: GenerateContentConfig = {
responseModalities: [Modality.TEXT, Modality.IMAGE], responseModalities: isGenerateImageModel(model) ? [Modality.TEXT, Modality.IMAGE] : undefined,
responseMimeType: 'text/plain', responseMimeType: isGenerateImageModel(model) ? 'text/plain' : undefined,
safetySettings: this.getSafetySettings(model.id), safetySettings: this.getSafetySettings(model.id),
// generate image don't need system instruction // generate image don't need system instruction
systemInstruction: isGemmaModel(model) || isGenerateImageModel(model) ? undefined : systemInstruction, systemInstruction: isGemmaModel(model) || isGenerateImageModel(model) ? undefined : systemInstruction,

2
src/renderer/src/services/ApiService.ts
View File

@ -282,7 +282,7 @@ export async function fetchChatCompletion({
} }
} }
console.log('message', message) // console.log('message', message)
// Emit chat completion event // Emit chat completion event
EventEmitter.emit(EVENT_NAMES.RECEIVE_MESSAGE, message) EventEmitter.emit(EVENT_NAMES.RECEIVE_MESSAGE, message)
onResponse(message) onResponse(message)