Northern_Lights/cherry-studio
1
0
Fork 0
mirror of https://github.com/CherryHQ/cherry-studio.git synced 2026-01-05 20:41:30 +08:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

fix(InputbarTools): allow url context for gemini endpoint type model (#10926)

Browse Source 
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 0081a0740f)
This commit is contained in:
Phantom 2025-10-24 13:55:10 +08:00 committed by kangfenmao
parent 2173b677e7
commit 88ffee3f38
2 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/renderer/src/pages/home/Inputbar/InputbarTools.tsx
View File

@ -383,7 +383,9 @@ const InputbarTools = ({
key: 'url_context', key: 'url_context',
label: t('chat.input.url_context'), label: t('chat.input.url_context'),
component: <UrlContextButton ref={urlContextButtonRef} assistantId={assistant.id} />, component: <UrlContextButton ref={urlContextButtonRef} assistantId={assistant.id} />,
condition: isGeminiModel(model) && isSupportUrlContextProvider(getProviderByModel(model)) condition:
isGeminiModel(model) &&
(isSupportUrlContextProvider(getProviderByModel(model)) || model.endpoint_type === 'gemini')
}, },
{ {
key: 'knowledge_base', key: 'knowledge_base',

3
src/renderer/src/services/AssistantService.ts
View File

@ -133,6 +133,8 @@ export function getAssistantProvider(assistant: Assistant): Provider {
return provider || getDefaultProvider() 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 { export function getProviderByModel(model?: Model): Provider {
const providers = getStoreProviders() const providers = getStoreProviders()
const provider = providers.find((p) => p.id === model?.provider) const provider = providers.find((p) => p.id === model?.provider)
@ -145,6 +147,7 @@ export function getProviderByModel(model?: Model): Provider {
return provider return provider
} }
// FIXME: This function may return undefined but as Provider
export function getProviderByModelId(modelId?: string) { export function getProviderByModelId(modelId?: string) {
const providers = getStoreProviders() const providers = getStoreProviders()
const _modelId = modelId || getDefaultModel().id const _modelId = modelId || getDefaultModel().id