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 = [];