From aafe6c5534123e94e3d169ab9a1756a9fabd0511 Mon Sep 17 00:00:00 2001 From: balibabu Date: Wed, 10 Jun 2026 19:11:31 +0800 Subject: [PATCH] Fix: The dataset retrieval test returned an incorrect total number. (#15901) ### What problem does this PR solve? Fix: The dataset retrieval test returned an incorrect total number. ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) --------- Co-authored-by: balibabu --- api/apps/services/dataset_api_service.py | 1 - web/src/hooks/use-knowledge-request.ts | 11 +++++++---- web/src/pages/dataset/testing/index.tsx | 2 +- 3 files changed, 8 insertions(+), 6 deletions(-) diff --git a/api/apps/services/dataset_api_service.py b/api/apps/services/dataset_api_service.py index a02e3900b5..6b4c3811a9 100644 --- a/api/apps/services/dataset_api_service.py +++ b/api/apps/services/dataset_api_service.py @@ -1429,7 +1429,6 @@ async def search_datasets(tenant_id: str, req: dict): except Exception: logging.warning("search_datasets KG retrieval failed: datasets=%s tenant=%s", kb_ids, tenant_id, exc_info=True) ranks["chunks"] = settings.retriever.retrieval_by_children(ranks["chunks"], tenant_ids) - ranks["total"] = len(ranks["chunks"]) for c in ranks["chunks"]: c.pop("vector", None) diff --git a/web/src/hooks/use-knowledge-request.ts b/web/src/hooks/use-knowledge-request.ts index bde0d7ef6f..d894771ade 100644 --- a/web/src/hooks/use-knowledge-request.ts +++ b/web/src/hooks/use-knowledge-request.ts @@ -87,21 +87,24 @@ export const useTestRetrieval = () => { }); const refetch = useCallback(() => { + setPage(1); if (queryParams.question) { - mutation.mutate(queryParams); + const newParams = { ...queryParams, page: 1 }; + mutation.mutate(newParams); } }, [mutation, queryParams]); const onPaginationChange = useCallback( (newPage: number, newPageSize: number) => { - setPage(newPage); + const nextPage = newPageSize !== pageSize ? 1 : newPage; + setPage(nextPage); setPageSize(newPageSize); if (mutation.data && queryParams.question) { - const newParams = { ...queryParams, page: newPage, size: newPageSize }; + const newParams = { ...queryParams, page: nextPage, size: newPageSize }; mutation.mutate(newParams); } }, - [mutation, queryParams], + [mutation, queryParams, pageSize], ); const handleFilterSubmit = useCallback( diff --git a/web/src/pages/dataset/testing/index.tsx b/web/src/pages/dataset/testing/index.tsx index 1f4b61fff3..effb13a00d 100644 --- a/web/src/pages/dataset/testing/index.tsx +++ b/web/src/pages/dataset/testing/index.tsx @@ -24,7 +24,7 @@ export default function RetrievalTesting() { filterValue, } = useTestRetrieval(); - const [count] = useState(1); + const [count] = useState(1); // TODO: Different layouts are needed; if they are no longer required, consider deleting them. return (