From c04fd62becd25d49c7b79903f049dba2f6af336f Mon Sep 17 00:00:00 2001
From: kangfenmao <kangfenmao@qq.com>
Date: Mon, 20 Jan 2025 12:55:24 +0800
Subject: [PATCH] feat: extended safety threshold check to include
 'thinking-exp' model ids

---
 src/renderer/src/providers/GeminiProvider.ts | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/renderer/src/providers/GeminiProvider.ts b/src/renderer/src/providers/GeminiProvider.ts
index e16ed522..c0d68d90 100644
--- a/src/renderer/src/providers/GeminiProvider.ts
+++ b/src/renderer/src/providers/GeminiProvider.ts
@@ -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 [
       {