From a430284aa21e3ae1f0d5654e55b2ad2852519cc2 Mon Sep 17 00:00:00 2001 From: wwf <yearningwang@iqtogether.com> Date: 星期三, 04 六月 2025 15:17:49 +0800 Subject: [PATCH] 初始化 --- service/knowledge/use-document.ts | 24 ++++++++++-------------- 1 files changed, 10 insertions(+), 14 deletions(-) diff --git a/service/knowledge/use-document.ts b/service/knowledge/use-document.ts index 6dabe7d..02e523b 100644 --- a/service/knowledge/use-document.ts +++ b/service/knowledge/use-document.ts @@ -4,37 +4,33 @@ } from '@tanstack/react-query' import { del, get, patch } from '../base' import { useInvalid } from '../use-base' -import type { MetadataType, SortType } from '../datasets' -import type { DocumentDetailResponse, DocumentListResponse, UpdateDocumentBatchParams } from '@/models/datasets' +import type { MetadataType } from '../datasets' +import type { DocumentDetailResponse, SimpleDocumentDetail, UpdateDocumentBatchParams } from '@/models/datasets' import { DocumentActionType } from '@/models/datasets' import type { CommonResponse } from '@/models/common' const NAME_SPACE = 'knowledge/document' -export const useDocumentListKey = [NAME_SPACE, 'documentList'] +const useDocumentListKey = [NAME_SPACE, 'documentList'] export const useDocumentList = (payload: { datasetId: string query: { keyword: string page: number limit: number - sort?: SortType - }, - refetchInterval?: number | false + } }) => { - const { query, datasetId, refetchInterval } = payload - const { keyword, page, limit, sort } = query - return useQuery<DocumentListResponse>({ - queryKey: [...useDocumentListKey, datasetId, keyword, page, limit, sort], - queryFn: () => get<DocumentListResponse>(`/datasets/${datasetId}/documents`, { + const { query, datasetId } = payload + return useQuery<{ data: SimpleDocumentDetail[] }>({ + queryKey: [...useDocumentListKey, datasetId, query], + queryFn: () => get<{ data: SimpleDocumentDetail[] }>(`/datasets/${datasetId}/documents`, { params: query, }), - refetchInterval, }) } -export const useInvalidDocumentList = (datasetId?: string) => { - return useInvalid(datasetId ? [...useDocumentListKey, datasetId] : useDocumentListKey) +export const useInvalidDocumentList = () => { + return useInvalid(useDocumentListKey) } const useAutoDisabledDocumentKey = [NAME_SPACE, 'autoDisabledDocument'] -- Gitblit v1.8.0