mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-08-06 23:51:15 +08:00
Refa: migrate document preview/download to RESTful API (#14633)
### What problem does this PR solve? migrate document preview/download to RESTful API ### Type of change - [x] Refactoring
This commit is contained in:
@@ -26,8 +26,7 @@ import kbService, {
|
||||
uploadDocument,
|
||||
webCrawlDocument,
|
||||
} from '@/services/knowledge-service';
|
||||
import { restAPIv1, webAPI } from '@/utils/api';
|
||||
import { getSearchValue } from '@/utils/common-util';
|
||||
import { restAPIv1 } from '@/utils/api';
|
||||
import { buildChunkHighlights } from '@/utils/document-util';
|
||||
import { useMutation, useQuery, useQueryClient } from '@tanstack/react-query';
|
||||
import { useDebounce } from 'ahooks';
|
||||
@@ -214,6 +213,7 @@ export const useGetDocumentFilter = (): {
|
||||
const { id } = useParams();
|
||||
const debouncedSearchString = useDebounce(searchString, { wait: 500 });
|
||||
const [open, setOpen] = useState<number>(0);
|
||||
const datasetId = knowledgeId || id;
|
||||
const { data } = useQuery({
|
||||
queryKey: [
|
||||
DocumentApiAction.FetchDocumentFilter,
|
||||
@@ -221,7 +221,10 @@ export const useGetDocumentFilter = (): {
|
||||
knowledgeId,
|
||||
],
|
||||
queryFn: async () => {
|
||||
const { data } = await documentFilter(knowledgeId || id);
|
||||
if (!datasetId) {
|
||||
return;
|
||||
}
|
||||
const { data } = await documentFilter(datasetId);
|
||||
if (data.code === 0) {
|
||||
return data.data;
|
||||
}
|
||||
@@ -504,14 +507,11 @@ export const useCreateDocument = () => {
|
||||
};
|
||||
|
||||
export const useGetDocumentUrl = (documentId?: string) => {
|
||||
const auth = getSearchValue('auth');
|
||||
const getDocumentUrl = useCallback(
|
||||
(id?: string) => {
|
||||
return auth
|
||||
? `${restAPIv1}/documents/${id || documentId}`
|
||||
: `${webAPI}/document/get/${id || documentId}`;
|
||||
return `${restAPIv1}/documents/${id || documentId}/preview`;
|
||||
},
|
||||
[documentId, auth],
|
||||
[documentId],
|
||||
);
|
||||
|
||||
return getDocumentUrl;
|
||||
|
||||
Reference in New Issue
Block a user