From de1976d984cbc891c53955eea4d011977f2fa958 Mon Sep 17 00:00:00 2001
From: linauror <linauror@gmail.com>
Date: Tue, 11 Feb 2025 18:05:09 +0800
Subject: [PATCH] =?UTF-8?q?fix:=E9=92=88=E5=AF=B9=E6=80=9D=E8=80=83?=
 =?UTF-8?q?=E7=B1=BB=E6=A8=A1=E5=9E=8B=E7=9A=84=E8=BF=94=E5=9B=9E=EF=BC=8C?=
 =?UTF-8?q?=E6=80=BB=E7=BB=93=E6=A0=87=E9=A2=98=E4=BB=85=E6=88=AA=E5=8F=96?=
 =?UTF-8?q?</think>=E4=B9=8B=E5=90=8E=E7=9A=84=E5=86=85=E5=AE=B9=20(#1415)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/renderer/src/providers/OpenAIProvider.ts | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/renderer/src/providers/OpenAIProvider.ts b/src/renderer/src/providers/OpenAIProvider.ts
index 2e4837fc..c3c3c75c 100644
--- a/src/renderer/src/providers/OpenAIProvider.ts
+++ b/src/renderer/src/providers/OpenAIProvider.ts
@@ -341,7 +341,11 @@ export default class OpenAIProvider extends BaseProvider {
       max_tokens: 1000
     })
 
-    return removeSpecialCharacters(response.choices[0].message?.content?.substring(0, 50) || '')
+    // 针对思考类模型的返回，总结仅截取</think>之后的内容
+    let content = response.choices[0].message?.content || ''
+    content = content.replace(/^<think>(.*?)<\/think>/s, '')
+    
+    return removeSpecialCharacters(content.substring(0, 50))
   }
 
   public async generateText({ prompt, content }: { prompt: string; content: string }): Promise<string> {