whai/cherry-studio
1
1
Fork 0
Code Issues Pull Requests Actions 2 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 220046cc95
commit de1976d984
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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 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> { public async generateText({ prompt, content }: { prompt: string; content: string }): Promise<string> {