Browse Source

enh: citations option in model editor

Timothy Jaeryang Baek 5 months ago
parent
commit
09c6e4b92f

+ 1 - 1
src/lib/components/chat/Messages/ResponseMessage.svelte

@@ -656,7 +656,7 @@
 									<Error content={message?.error?.content ?? message.content} />
 								{/if}
 
-								{#if message.citations}
+								{#if message.citations && (model?.info?.meta?.capabilities?.citations ?? true)}
 									<Citations citations={message.citations} />
 								{/if}
 

+ 3 - 1
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;
 	} = {};
 </script>
 

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