mirror of
https://github.com/langgenius/dify.git
synced 2025-11-22 16:06:53 +00:00
85cda47c70
Signed-off-by: -LAN- <laipz8200@outlook.com> Co-authored-by: twwu <twwu@dify.ai> Co-authored-by: crazywoola <100913391+crazywoola@users.noreply.github.com> Co-authored-by: jyong <718720800@qq.com> Co-authored-by: Wu Tianwei <30284043+WTW0313@users.noreply.github.com> Co-authored-by: QuantumGhost <obelisk.reg+git@gmail.com> Co-authored-by: lyzno1 <yuanyouhuilyz@gmail.com> Co-authored-by: quicksand <quicksandzn@gmail.com> Co-authored-by: Jyong <76649700+JohnJyong@users.noreply.github.com> Co-authored-by: lyzno1 <92089059+lyzno1@users.noreply.github.com> Co-authored-by: zxhlyh <jasonapring2015@outlook.com> Co-authored-by: Yongtao Huang <yongtaoh2022@gmail.com> Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com> Co-authored-by: Joel <iamjoel007@gmail.com> Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com> Co-authored-by: nite-knite <nkCoding@gmail.com> Co-authored-by: Hanqing Zhao <sherry9277@gmail.com> Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com> Co-authored-by: Harry <xh001x@hotmail.com>
53 lines
2.6 KiB
TypeScript
53 lines
2.6 KiB
TypeScript
import type { RAGPipelineVariables } from '@/models/pipeline'
|
|
import type { StateCreator } from 'zustand'
|
|
import type {
|
|
ToolWithProvider,
|
|
} from '@/app/components/workflow/types'
|
|
import type { DataSourceItem } from '@/app/components/workflow/block-selector/types'
|
|
import { transformDataSourceToTool } from '@/app/components/workflow/block-selector/utils'
|
|
import type { IconInfo } from '@/models/datasets'
|
|
import type { InputFieldEditorProps } from '../components/panel/input-field/editor'
|
|
|
|
export type RagPipelineSliceShape = {
|
|
pipelineId: string
|
|
knowledgeName: string
|
|
knowledgeIcon?: IconInfo
|
|
showInputFieldPanel: boolean
|
|
setShowInputFieldPanel: (showInputFieldPanel: boolean) => void
|
|
showInputFieldPreviewPanel: boolean
|
|
setShowInputFieldPreviewPanel: (showInputFieldPreviewPanel: boolean) => void
|
|
inputFieldEditPanelProps: InputFieldEditorProps | null
|
|
setInputFieldEditPanelProps: (showInputFieldEditPanel: InputFieldEditorProps | null) => void
|
|
nodesDefaultConfigs: Record<string, any>
|
|
setNodesDefaultConfigs: (nodesDefaultConfigs: Record<string, any>) => void
|
|
ragPipelineVariables: RAGPipelineVariables
|
|
setRagPipelineVariables: (ragPipelineVariables: RAGPipelineVariables) => void
|
|
dataSourceList: ToolWithProvider[]
|
|
setDataSourceList: (dataSourceList: DataSourceItem[]) => void
|
|
isPreparingDataSource: boolean
|
|
setIsPreparingDataSource: (isPreparingDataSource: boolean) => void
|
|
}
|
|
|
|
export type CreateRagPipelineSliceSlice = StateCreator<RagPipelineSliceShape>
|
|
export const createRagPipelineSliceSlice: StateCreator<RagPipelineSliceShape> = set => ({
|
|
pipelineId: '',
|
|
knowledgeName: '',
|
|
showInputFieldPanel: false,
|
|
setShowInputFieldPanel: showInputFieldPanel => set(() => ({ showInputFieldPanel })),
|
|
showInputFieldPreviewPanel: false,
|
|
setShowInputFieldPreviewPanel: showInputFieldPreviewPanel => set(() => ({ showInputFieldPreviewPanel })),
|
|
inputFieldEditPanelProps: null,
|
|
setInputFieldEditPanelProps: inputFieldEditPanelProps => set(() => ({ inputFieldEditPanelProps })),
|
|
nodesDefaultConfigs: {},
|
|
setNodesDefaultConfigs: nodesDefaultConfigs => set(() => ({ nodesDefaultConfigs })),
|
|
ragPipelineVariables: [],
|
|
setRagPipelineVariables: (ragPipelineVariables: RAGPipelineVariables) => set(() => ({ ragPipelineVariables })),
|
|
dataSourceList: [],
|
|
setDataSourceList: (dataSourceList: DataSourceItem[]) => {
|
|
const formattedDataSourceList = dataSourceList.map(item => transformDataSourceToTool(item))
|
|
set(() => ({ dataSourceList: formattedDataSourceList }))
|
|
},
|
|
isPreparingDataSource: false,
|
|
setIsPreparingDataSource: isPreparingDataSource => set(() => ({ isPreparingDataSource })),
|
|
})
|