diff --git a/src/lib/components/chat/Messages/ResponseMessage.svelte b/src/lib/components/chat/Messages/ResponseMessage.svelte index 765f5b17b..6c4dbc5d6 100644 --- a/src/lib/components/chat/Messages/ResponseMessage.svelte +++ b/src/lib/components/chat/Messages/ResponseMessage.svelte @@ -656,7 +656,7 @@ {/if} - {#if message.citations} + {#if message.citations && (model?.info?.meta?.capabilities?.citations ?? true)} {/if} diff --git a/src/lib/components/workspace/Models/Capabilities.svelte b/src/lib/components/workspace/Models/Capabilities.svelte index f5325d5f3..cbccef1ff 100644 --- a/src/lib/components/workspace/Models/Capabilities.svelte +++ b/src/lib/components/workspace/Models/Capabilities.svelte @@ -10,12 +10,14 @@ vision: $i18n.t('Model accepts image inputs'), usage: $i18n.t( 'Sends `stream_options: { include_usage: true }` in the request.\nSupported providers will return token usage information in the response when set.' - ) + ), + citations: $i18n.t('Displays citations in the response') }; export let capabilities: { vision?: boolean; usage?: boolean; + citations?: boolean; } = {}; diff --git a/src/lib/components/workspace/Models/ModelEditor.svelte b/src/lib/components/workspace/Models/ModelEditor.svelte index 7a7b376f5..91680d942 100644 --- a/src/lib/components/workspace/Models/ModelEditor.svelte +++ b/src/lib/components/workspace/Models/ModelEditor.svelte @@ -71,7 +71,8 @@ let params = {}; let capabilities = { vision: true, - usage: undefined + usage: undefined, + citations: true }; let knowledge = [];