pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2025-12-08 20:42:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: Fetch chunk list by @tanstack/react-query #1306 (#1738)

Browse Source 
### What problem does this PR solve?

fix: Fetch chunk list by @tanstack/react-query #1306

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
balibabu
2024-07-29 19:20:13 +08:00
committed by GitHub
parent aac460ad29
commit 8468031e39
8 changed files with 240 additions and 101 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
web/src/pages/add-knowledge/components/knowledge-chunk/hooks.ts
View File

@ -1,24 +1,17 @@
import { useSelectChunkList } from '@/hooks/chunk-hooks';
import { useOneNamespaceEffectsLoading } from '@/hooks/store-hooks';
import { IChunk, IKnowledgeFile } from '@/interfaces/database/knowledge';
import { IChunk } from '@/interfaces/database/knowledge';
import { buildChunkHighlights } from '@/utils/document-util';
import { useCallback, useMemo, useState } from 'react';
import { IHighlight } from 'react-pdf-highlighter';
import { useSelector } from 'umi';
import { ChunkTextMode } from './constant';
export const useSelectDocumentInfo = () => {
const documentInfo: IKnowledgeFile = useSelector(
(state: any) => state.chunkModel.documentInfo,
);
return documentInfo;
};
export const useSelectChunkList = () => {
const chunkList: IChunk[] = useSelector(
(state: any) => state.chunkModel.data,
);
return chunkList;
};
// export const useSelectChunkList = () => {
// const chunkList: IChunk[] = useSelector(
// (state: any) => state.chunkModel.data,
// );
// return chunkList;
// };
export const useHandleChunkCardClick = () => {
const [selectedChunkId, setSelectedChunkId] = useState<string>('');
@ -31,9 +24,9 @@ export const useHandleChunkCardClick = () => {
};
export const useGetSelectedChunk = (selectedChunkId: string) => {
const chunkList: IChunk[] = useSelectChunkList();
const data = useSelectChunkList();
return (
chunkList.find((x) => x.chunk_id === selectedChunkId) ?? ({} as IChunk)
data?.data?.find((x) => x.chunk_id === selectedChunkId) ?? ({} as IChunk)
);
};
@ -45,14 +38,14 @@ export const useGetChunkHighlights = (selectedChunkId: string) => {
return buildChunkHighlights(selectedChunk, size);
}, [selectedChunk, size]);
const setWidthAndHeight = (width: number, height: number) => {
const setWidthAndHeight = useCallback((width: number, height: number) => {
setSize((pre) => {
if (pre.height !== height || pre.width !== width) {
return { height, width };
}
return pre;
});
};
}, []);
return { highlights, setWidthAndHeight };
};