From c3b0beb37f6d58e301c7d6c66011c86dc9a2d5fb Mon Sep 17 00:00:00 2001 From: Phantom Date: Fri, 24 Oct 2025 13:55:10 +0800 Subject: [PATCH] fix(InputbarTools): allow url context for gemini endpoint type model (#10926) fix(InputbarTools): allow url context for gemini endpoint type Add condition to check for gemini endpoint type when determining URL context support (cherry picked from commit 0081a0740fbc9e000ba192ec6a6d289017424fbe) --- src/renderer/src/pages/home/Inputbar/InputbarTools.tsx | 4 +++- src/renderer/src/services/AssistantService.ts | 3 +++ 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/src/renderer/src/pages/home/Inputbar/InputbarTools.tsx b/src/renderer/src/pages/home/Inputbar/InputbarTools.tsx index 36b17a8bc2..42d8445d5e 100644 --- a/src/renderer/src/pages/home/Inputbar/InputbarTools.tsx +++ b/src/renderer/src/pages/home/Inputbar/InputbarTools.tsx @@ -383,7 +383,9 @@ const InputbarTools = ({ key: 'url_context', label: t('chat.input.url_context'), component: , - condition: isGeminiModel(model) && isSupportUrlContextProvider(getProviderByModel(model)) + condition: + isGeminiModel(model) && + (isSupportUrlContextProvider(getProviderByModel(model)) || model.endpoint_type === 'gemini') }, { key: 'knowledge_base', diff --git a/src/renderer/src/services/AssistantService.ts b/src/renderer/src/services/AssistantService.ts index f9a4e47515..c3e1425c71 100644 --- a/src/renderer/src/services/AssistantService.ts +++ b/src/renderer/src/services/AssistantService.ts @@ -133,6 +133,8 @@ export function getAssistantProvider(assistant: Assistant): Provider { return provider || getDefaultProvider() } +// FIXME: This function fails in silence. +// TODO: Refactor it to make it return exactly valid value or null, and update all usage. export function getProviderByModel(model?: Model): Provider { const providers = getStoreProviders() const provider = providers.find((p) => p.id === model?.provider) @@ -145,6 +147,7 @@ export function getProviderByModel(model?: Model): Provider { return provider } +// FIXME: This function may return undefined but as Provider export function getProviderByModelId(modelId?: string) { const providers = getStoreProviders() const _modelId = modelId || getDefaultModel().id