youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-06 10:28:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: add pipeline template details and import functionality, enhance dataset pipeline management

Browse Source
This commit is contained in:
twwu
2025-05-07 18:09:38 +08:00
parent 3f7f21ce70
commit 7ce9710229
9 changed files with 231 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
web/service/use-pipeline.ts
View File

@ -1,9 +1,11 @@
import type { MutationOptions } from '@tanstack/react-query'
import { useMutation, useQuery } from '@tanstack/react-query'
import { del, get, patch } from './base'
import { del, get, patch, post } from './base'
import type {
DeletePipelineResponse,
ExportPipelineDSLResponse,
ImportPipelineDSLRequest,
ImportPipelineDSLResponse,
PipelineTemplateByIdResponse,
PipelineTemplateListParams,
PipelineTemplateListResponse,
@ -22,12 +24,13 @@ export const usePipelineTemplateList = (params: PipelineTemplateListParams) => {
})
}
export const usePipelineTemplateById = (templateId: string) => {
export const usePipelineTemplateById = (templateId: string, enabled: boolean) => {
return useQuery<PipelineTemplateByIdResponse>({
queryKey: [NAME_SPACE, 'template', templateId],
queryFn: () => {
return get<PipelineTemplateByIdResponse>(`/rag/pipeline/template/${templateId}`)
},
enabled,
})
}
@ -69,3 +72,16 @@ export const useExportPipelineDSL = (
...mutationOptions,
})
}
// TODO: replace with real API
export const useImportPipelineDSL = (
mutationOptions: MutationOptions<ImportPipelineDSLResponse, Error, ImportPipelineDSLRequest> = {},
) => {
return useMutation({
mutationKey: [NAME_SPACE, 'template', 'import'],
mutationFn: (request: ImportPipelineDSLRequest) => {
return post<ImportPipelineDSLResponse>('/rag/pipeline/import', { body: request })
},
...mutationOptions,
})
}