whai/cherry-studio
1
1
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix: o1模型支持流式输出 #439

Browse Source
This commit is contained in:
kangfenmao 2024-12-16 14:48:51 +08:00
parent 978f1342e4
commit 7fdc2db522
1 changed files with 1 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -39,13 +39,6 @@ export default class OpenAIProvider extends BaseProvider {
}) })
} }
private isSupportStreamOutput(modelId: string): boolean {
if (modelId.includes('o1-')) {
return false
}
return true
}
private get isNotSupportFiles() { private get isNotSupportFiles() {
const providers = ['deepseek', 'baichuan', 'minimax', 'doubao'] const providers = ['deepseek', 'baichuan', 'minimax', 'doubao']
return providers.includes(this.provider.id) return providers.includes(this.provider.id)
@ -137,7 +130,7 @@ export default class OpenAIProvider extends BaseProvider {
} }
const isOpenAIo1 = model.id.includes('o1-') const isOpenAIo1 = model.id.includes('o1-')
const isSupportStreamOutput = streamOutput && this.isSupportStreamOutput(model.id) const isSupportStreamOutput = streamOutput
// @ts-ignore key is not typed // @ts-ignore key is not typed
const stream = await this.sdk.chat.completions.create({ const stream = await this.sdk.chat.completions.create({