From 6af07c278d66387df8386d1e01642f2c245e2ab3 Mon Sep 17 00:00:00 2001 From: kangfenmao Date: Fri, 13 Dec 2024 10:28:09 +0800 Subject: [PATCH] fix: handle unknown models in iswebsearchmodel function --- src/renderer/src/config/models.ts | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/src/renderer/src/config/models.ts b/src/renderer/src/config/models.ts index 8702e1f1..7adef97d 100644 --- a/src/renderer/src/config/models.ts +++ b/src/renderer/src/config/models.ts @@ -1060,5 +1060,10 @@ export function isSupportedModel(model: OpenAI.Models.Model): boolean { export function isWebSearchModel(model: Model): boolean { const provider = getProviderByModel(model) + + if (!provider) { + return false + } + return (provider.id === 'gemini' || provider?.type === 'gemini') && model?.id === 'gemini-2.0-flash-exp' }