Northern_Lights/cherry-studio
1
0
Fork 0
mirror of https://github.com/CherryHQ/cherry-studio.git synced 2025-12-26 20:12:38 +08:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

fix:针对思考类模型的返回，总结标题仅截取</think>之后的内容 (#1415)

Browse Source
This commit is contained in:
linauror 2025-02-11 18:05:09 +08:00 committed by GitHub
parent 29b83862b7
commit 83d96c7927
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/renderer/src/providers/OpenAIProvider.ts
View File

@ -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> {