From 3596b0cc0e06ca5d8ce1ddfab4eaa731f9695f7c Mon Sep 17 00:00:00 2001
From: zhaoying <zhaoyingyz@126.com>
Date: Tue, 2 Jun 2026 11:51:49 +0800
Subject: [PATCH] fix(web): kb document chunk total size

---
 .../views/KnowledgeBase/[knowledgeBaseId]/DocumentDetails.tsx | 3 +++
 web/src/views/KnowledgeBase/components/RecallTestResult.tsx   | 4 +++-
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/web/src/views/KnowledgeBase/[knowledgeBaseId]/DocumentDetails.tsx b/web/src/views/KnowledgeBase/[knowledgeBaseId]/DocumentDetails.tsx
index 0c3eaffa..fdd735e9 100644
--- a/web/src/views/KnowledgeBase/[knowledgeBaseId]/DocumentDetails.tsx
+++ b/web/src/views/KnowledgeBase/[knowledgeBaseId]/DocumentDetails.tsx
@@ -50,6 +50,7 @@ const DocumentDetails: FC = () => {
   const [infoItems, setInfoItems] = useState<InfoItem[]>([]);
   const [page, setPage] = useState(1);
   const [hasMore, setHasMore] = useState(true);
+  const [total, setTotal] = useState(0);
   const [chunkLoading, setChunkLoading] = useState(false);
   const [keywords, setKeywords] = useState('');
   const [fileUrl, setFileUrl] = useState('');
@@ -219,6 +220,7 @@ const DocumentDetails: FC = () => {
       }
       
       setHasMore(response.page?.has_next ?? false);
+      setTotal(response.page?.total ?? 0);
     } catch (error) {
       console.error('Failed to fetch document details:', error);
       message.error(t('common.loadFailed') || '加载失败');
@@ -472,6 +474,7 @@ const DocumentDetails: FC = () => {
           <RecallTestResult 
             refresh={refreshChunks}
             data={chunkList} 
+            total={total}
             showEmpty={false}
             hasMore={hasMore}
             loadMore={loadMoreChunks}
diff --git a/web/src/views/KnowledgeBase/components/RecallTestResult.tsx b/web/src/views/KnowledgeBase/components/RecallTestResult.tsx
index 407b74d1..934e2b9a 100644
--- a/web/src/views/KnowledgeBase/components/RecallTestResult.tsx
+++ b/web/src/views/KnowledgeBase/components/RecallTestResult.tsx
@@ -23,6 +23,7 @@ interface RecallTestResultProps {
   data: RecallTestData[];
   showEmpty?: boolean;
   hasMore?: boolean;
+  total?: number;
   loadMore?: () => void;
   refresh?: () => void;
   loading?: boolean;
@@ -38,6 +39,7 @@ const RecallTestResult = ({
   data, 
   showEmpty = true,
   hasMore = false,
+  total = 0,
   loadMore,
   refresh,
   loading = false,
@@ -395,7 +397,7 @@ const RecallTestResult = ({
         <div className='rb:flex rb:items-center rb:justify-start rb:gap-2'>
           <span className='rb:text-lg rb:font-medium'>{t('knowledgeBase.recallResult')}</span>
           <span className='rb:text-gray-500 rb:text-xs rb:pt-0.5'>
-            (<span className='rb:text-[#155EEF]'>{data.length}</span> results)
+            (<span className='rb:text-[#155EEF]'>{total || data.length}</span> results)
           </span>
         </div>
         <InfiniteScroll