mirror of
https://github.com/langgenius/dify.git
synced 2025-07-23 09:32:36 +00:00
31 lines
1.3 KiB
TypeScript
31 lines
1.3 KiB
TypeScript
import type { RAGPipelineVariables } from '@/models/pipeline'
|
|
import type { StateCreator } from 'zustand'
|
|
import type {
|
|
ToolWithProvider,
|
|
} from '@/app/components/workflow/types'
|
|
|
|
export type RagPipelineSliceShape = {
|
|
pipelineId: string
|
|
showInputFieldDialog: boolean
|
|
setShowInputFieldDialog: (showInputFieldPanel: boolean) => void
|
|
nodesDefaultConfigs: Record<string, any>
|
|
setNodesDefaultConfigs: (nodesDefaultConfigs: Record<string, any>) => void
|
|
ragPipelineVariables: RAGPipelineVariables
|
|
setRagPipelineVariables: (ragPipelineVariables: RAGPipelineVariables) => void
|
|
dataSourceList: ToolWithProvider[]
|
|
setDataSourceList: (dataSourceList: ToolWithProvider[]) => void
|
|
}
|
|
|
|
export type CreateRagPipelineSliceSlice = StateCreator<RagPipelineSliceShape>
|
|
export const createRagPipelineSliceSlice: StateCreator<RagPipelineSliceShape> = set => ({
|
|
pipelineId: '',
|
|
showInputFieldDialog: false,
|
|
setShowInputFieldDialog: showInputFieldDialog => set(() => ({ showInputFieldDialog })),
|
|
nodesDefaultConfigs: {},
|
|
setNodesDefaultConfigs: nodesDefaultConfigs => set(() => ({ nodesDefaultConfigs })),
|
|
ragPipelineVariables: [],
|
|
setRagPipelineVariables: (ragPipelineVariables: RAGPipelineVariables) => set(() => ({ ragPipelineVariables })),
|
|
dataSourceList: [],
|
|
setDataSourceList: (dataSourceList: ToolWithProvider[]) => set(() => ({ dataSourceList })),
|
|
})
|