From 0d38fabfac3584b6a16d031e72b9b1b9d5850e3e Mon Sep 17 00:00:00 2001
From: SuYao <sy20010504@gmail.com>
Date: Wed, 14 May 2025 13:51:25 +0800
Subject: [PATCH] fix: improve citation deduplication logic for non-knowledge
 citations (#5981)

---
 src/renderer/src/store/messageBlock.ts | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/renderer/src/store/messageBlock.ts b/src/renderer/src/store/messageBlock.ts
index 2a4ac9845c..c9cc55cec3 100644
--- a/src/renderer/src/store/messageBlock.ts
+++ b/src/renderer/src/store/messageBlock.ts
@@ -236,10 +236,11 @@ const formatCitationsFromBlock = (block: CitationMessageBlock | undefined): Cita
       })
     )
   }
-  // 4. Deduplicate by URL and Renumber Sequentially
+  // 4. Deduplicate non-knowledge citations by URL and Renumber Sequentially
   const urlSet = new Set<string>()
   return formattedCitations
     .filter((citation) => {
+      if (citation.type === 'knowledge') return true
       if (!citation.url || urlSet.has(citation.url)) return false
       urlSet.add(citation.url)
       return true