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

feat: extended safety threshold check to include 'thinking-exp' model ids

Browse Source
This commit is contained in:
kangfenmao 2025-01-20 12:55:24 +08:00
parent f86a274cd3
commit c04fd62bec
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/renderer/src/providers/GeminiProvider.ts
View File

@ -114,9 +114,10 @@ export default class GeminiProvider extends BaseProvider {
}
private getSafetySettings(modelId: string): SafetySetting[] {
const safetyThreshold = modelId.includes('gemini-exp-')
? HarmBlockThreshold.BLOCK_NONE
: ('OFF' as HarmBlockThreshold)
const safetyThreshold =
modelId.includes('gemini-exp-') || modelId.includes('thinking-exp')
? HarmBlockThreshold.BLOCK_NONE
: ('OFF' as HarmBlockThreshold)
return [
{