refactor: replace SWR with custom hooks for dataset detail and related apps; update context usage in components

This commit is contained in:
twwu
2025-05-16 16:32:25 +08:00
parent 87dc80f6fa
commit 7de5585da6
5 changed files with 28 additions and 20 deletions

View File

@ -1,5 +1,5 @@
import { useInfiniteQuery } from '@tanstack/react-query'
import type { DataSetListResponse, DatasetListRequest } from '@/models/datasets'
import { useInfiniteQuery, useQuery } from '@tanstack/react-query'
import type { DataSet, DataSetListResponse, DatasetListRequest, RelatedAppResponse } from '@/models/datasets'
import { get } from '../base'
import { useReset } from '../use-base'
import qs from 'qs'
@ -30,3 +30,17 @@ export const useDatasetList = (params: DatasetListRequest) => {
export const useResetDatasetList = () => {
return useReset([...DatasetListKey])
}
export const useDatasetDetail = (datasetId: string) => {
return useQuery({
queryKey: [NAME_SPACE, 'detail', datasetId],
queryFn: () => get<DataSet>(`/datasets/${datasetId}`),
})
}
export const useDatasetRelatedApps = (datasetId: string) => {
return useQuery({
queryKey: [NAME_SPACE, 'related-apps', datasetId],
queryFn: () => get<RelatedAppResponse>(`/datasets/${datasetId}/related_apps`),
})
}

View File

@ -66,7 +66,7 @@ export const useDeleteTemplate = (
}
export const useExportTemplateDSL = (
mutationOptions: MutationOptions<ExportTemplateDSLResponse, Error, ExportTemplateDSLRequest> = {},
mutationOptions: MutationOptions<ExportTemplateDSLResponse, Error, string> = {},
) => {
return useMutation({
mutationKey: [NAME_SPACE, 'dsl-export'],
@ -83,7 +83,7 @@ export const useImportPipelineDSL = (
return useMutation({
mutationKey: [NAME_SPACE, 'dsl-import'],
mutationFn: (request: ImportPipelineDSLRequest) => {
return post<ImportPipelineDSLResponse>('/rag/pipeline/imports', { body: request })
return post<ImportPipelineDSLResponse>('/rag/pipelines/imports', { body: request })
},
...mutationOptions,
})