mirror of
https://github.com/langgenius/dify.git
synced 2026-05-04 17:38:04 +08:00
refactor: replace SWR with custom hooks for dataset detail and related apps; update context usage in components
This commit is contained in:
@ -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`),
|
||||
})
|
||||
}
|
||||
|
||||
@ -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,
|
||||
})
|
||||
|
||||
Reference in New Issue
Block a user