45 lines
1.3 KiB
TypeScript
Raw Normal View History

2025-04-25 11:32:17 +08:00
import WorkflowWithDefaultContext from '@/app/components/workflow'
2025-04-18 14:56:34 +08:00
import {
WorkflowContextProvider,
} from '@/app/components/workflow/context'
2025-04-18 15:46:54 +08:00
import type { InjectWorkflowStoreSliceFn } from '@/app/components/workflow/store'
2025-04-29 16:11:54 +08:00
import { generateNewNode } from '@/app/components/workflow/utils'
import dataSourceNodeDefault from '@/app/components/workflow/nodes/data-source/default'
import {
NODE_WIDTH_X_OFFSET,
START_INITIAL_POSITION,
} from '@/app/components/workflow/constants'
2025-04-18 15:46:54 +08:00
import { createRagPipelineSliceSlice } from './store'
2025-04-25 11:32:17 +08:00
import RagPipelineMain from './components/rag-pipeline-main'
2025-04-18 14:56:34 +08:00
const RagPipeline = () => {
2025-04-29 16:11:54 +08:00
const { newNode: DataSourceNode } = generateNewNode({
data: {
type: dataSourceNodeDefault.metaData.type,
title: 'data-source',
...dataSourceNodeDefault.defaultValue,
},
position: {
x: START_INITIAL_POSITION.x + NODE_WIDTH_X_OFFSET,
y: START_INITIAL_POSITION.y,
},
} as any)
2025-04-18 14:56:34 +08:00
return (
2025-04-18 15:46:54 +08:00
<WorkflowContextProvider
injectWorkflowStoreSliceFn={createRagPipelineSliceSlice as InjectWorkflowStoreSliceFn}
>
2025-04-18 14:56:34 +08:00
<WorkflowWithDefaultContext
edges={[]}
2025-04-29 16:11:54 +08:00
nodes={[DataSourceNode]}
2025-04-18 14:56:34 +08:00
>
2025-04-25 11:32:17 +08:00
<RagPipelineMain
2025-04-18 14:56:34 +08:00
edges={[]}
2025-04-29 16:11:54 +08:00
nodes={[DataSourceNode]}
2025-04-25 11:32:17 +08:00
/>
2025-04-18 14:56:34 +08:00
</WorkflowWithDefaultContext>
</WorkflowContextProvider>
)
}
export default RagPipeline