From ef5c0d8460a6323268bc786fbe33f436e49a3141 Mon Sep 17 00:00:00 2001 From: Asurada <43401755+ousugo@users.noreply.github.com> Date: Thu, 17 Apr 2025 15:19:14 +0800 Subject: [PATCH] fix(models): simplify OpenAI o-series model identification logic (#4985) * fix(models): simplify OpenAI o-series model identification logic * Update OpenAIProvider.ts --------- Co-authored-by: Pleasurecruise <3196812536@qq.com> --- src/renderer/src/config/models.ts | 2 +- src/renderer/src/providers/AiProvider/OpenAIProvider.ts | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/renderer/src/config/models.ts b/src/renderer/src/config/models.ts index fd726548d0..ed0f702025 100644 --- a/src/renderer/src/config/models.ts +++ b/src/renderer/src/config/models.ts @@ -2208,7 +2208,7 @@ export function isVisionModel(model: Model): boolean { } export function isOpenAIoSeries(model: Model): boolean { - return ['o1', 'o1-2024-12-17'].includes(model.id) || model.id.includes('o3') || model.id.includes('o4') + return model.id.includes('o1') || model.id.includes('o3') || model.id.includes('o4') } export function isOpenAIWebSearch(model: Model): boolean { diff --git a/src/renderer/src/providers/AiProvider/OpenAIProvider.ts b/src/renderer/src/providers/AiProvider/OpenAIProvider.ts index dfdbdf0bd7..4373c85c05 100644 --- a/src/renderer/src/providers/AiProvider/OpenAIProvider.ts +++ b/src/renderer/src/providers/AiProvider/OpenAIProvider.ts @@ -293,7 +293,7 @@ export default class OpenAIProvider extends BaseProvider { * @returns True if the model is an OpenAI reasoning model, false otherwise */ private isOpenAIReasoning(model: Model) { - return model.id.startsWith('o1') || model.id.startsWith('o3') + return model.id.startsWith('o1') || model.id.startsWith('o3') || model.id.startsWith('o4') } /**