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