Merge pull request #10268 from hurxxxx/feat/hide-citations-in-message

feat: citations disabled model to exclude source citations in messages
This commit is contained in:
Timothy Jaeryang Baek 2025-02-18 19:37:12 -08:00 committed by GitHub
commit c658781995
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 17 additions and 10 deletions

View File

@ -120,6 +120,11 @@
sourceIds={(sources ?? []).reduce((acc, s) => { sourceIds={(sources ?? []).reduce((acc, s) => {
let ids = []; let ids = [];
s.document.forEach((document, index) => { s.document.forEach((document, index) => {
if (model?.info?.meta?.capabilities?.citations == false) {
ids.push('N/A');
return ids;
}
const metadata = s.metadata?.[index]; const metadata = s.metadata?.[index];
const id = metadata?.source ?? 'N/A'; const id = metadata?.source ?? 'N/A';

View File

@ -35,13 +35,15 @@
$: attributes = extractAttributes(token.text); $: attributes = extractAttributes(token.text);
</script> </script>
<button {#if attributes.title !== 'N/A'}
<button
class="text-xs font-medium w-fit translate-y-[2px] px-2 py-0.5 dark:bg-white/5 dark:text-white/60 dark:hover:text-white bg-gray-50 text-black/60 hover:text-black transition rounded-lg" class="text-xs font-medium w-fit translate-y-[2px] px-2 py-0.5 dark:bg-white/5 dark:text-white/60 dark:hover:text-white bg-gray-50 text-black/60 hover:text-black transition rounded-lg"
on:click={() => { on:click={() => {
onClick(attributes.data); onClick(attributes.data);
}} }}
> >
<span class="line-clamp-1"> <span class="line-clamp-1">
{attributes.title ? formattedTitle(attributes.title) : ''} {attributes.title ? formattedTitle(attributes.title) : ''}
</span> </span>
</button> </button>
{/if}