From 8a94bb05eadca4ee74a954ed41261c410d0fd897 Mon Sep 17 00:00:00 2001 From: kangfenmao Date: Tue, 14 Jan 2025 20:32:04 +0800 Subject: [PATCH] fix: fix model type logic based on provider properties --- src/renderer/src/config/models.ts | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/src/renderer/src/config/models.ts b/src/renderer/src/config/models.ts index 2fb572e3..1a1ae956 100644 --- a/src/renderer/src/config/models.ts +++ b/src/renderer/src/config/models.ts @@ -1105,16 +1105,16 @@ export function isWebSearchModel(model: Model): boolean { const provider = getProviderByModel(model) - if (provider.type === 'openai') { + if (!provider) { + return false + } + + if (provider?.type === 'openai') { if (model?.id?.includes('gemini-2.0-flash-exp')) { return true } } - if (!provider) { - return false - } - if (provider.id === 'gemini' || provider?.type === 'gemini') { return model?.id === 'gemini-2.0-flash-exp' }