feat: add child segment response type and enhance dataset components with improved translations and UI updates

web/service/knowledge/use-segment.ts

@@ -1,11 +1,11 @@
 import { useMutation, useQuery } from '@tanstack/react-query'
 import { del, get, patch } from '../base'
 import type { CommonResponse } from '@/models/common'
-import type { SegmentsResponse } from '@/models/datasets'
+import type { ChildSegmentResponse, SegmentsResponse } from '@/models/datasets'
 
 const NAME_SPACE = 'segment'
 
-const useSegmentListKey = [NAME_SPACE, 'list']
+const useSegmentListKey = [NAME_SPACE, 'chunkList']
 
 export const useSegmentList = (
   payload: {
@@ -28,7 +28,7 @@ export const useSegmentList = (
       return get<SegmentsResponse>(`/datasets/${datasetId}/documents/${documentId}/segments`, { params })
     },
     enabled: !disable,
-    initialData: disable ? { data: [], has_more: false, total: 0, total_pages: 0, limit: 10 } : undefined,
+    initialData: disable ? { data: [], has_more: false, page: 1, total: 0, total_pages: 0, limit: 10 } : undefined,
   })
 }
 
@@ -64,3 +64,30 @@ export const useDeleteSegment = () => {
     },
   })
 }
+
+const useChildSegmentListKey = [NAME_SPACE, 'childChunkList']
+
+export const useChildSegmentList = (
+  payload: {
+    datasetId: string
+    documentId: string
+    segmentId: string
+    params: {
+      page: number
+      limit: number
+      keyword: string
+    }
+  },
+  disable?: boolean,
+) => {
+  const { datasetId, documentId, segmentId, params } = payload
+  const { page, limit, keyword } = params
+  return useQuery({
+    queryKey: [...useChildSegmentListKey, datasetId, documentId, segmentId, page, limit, keyword],
+    queryFn: () => {
+      return get<ChildSegmentResponse>(`/datasets/${datasetId}/documents/${documentId}/segment/${segmentId}/child_chunks`, { params })
+    },
+    enabled: !disable,
+    initialData: disable ? { data: [], total: 0, page: 1, total_pages: 0, limit: 10 } : undefined,
+  })
+}