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

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

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