50 lines
1.5 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'
2025-05-06 17:24:30 +08:00
import knowledgeBaseNodeDefault from '@/app/components/workflow/nodes/knowledge-base/default'
2025-04-29 16:11:54 +08:00
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-05-09 15:53:31 +08:00
// import { usePipelineInit } from './hooks'
2025-04-18 14:56:34 +08:00
const RagPipeline = () => {
2025-05-09 15:53:31 +08:00
// const {
// data,
// isLoading,
// } = usePipelineInit()
2025-05-06 17:24:30 +08:00
const { newNode: knowledgeBaseNode } = generateNewNode({
2025-04-29 16:11:54 +08:00
data: {
2025-05-06 17:24:30 +08:00
type: knowledgeBaseNodeDefault.metaData.type,
title: 'knowledge-base',
...knowledgeBaseNodeDefault.defaultValue,
2025-04-29 16:11:54 +08:00
},
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-05-06 17:24:30 +08:00
nodes={[knowledgeBaseNode]}
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-05-06 17:24:30 +08:00
nodes={[knowledgeBaseNode]}
2025-04-25 11:32:17 +08:00
/>
2025-04-18 14:56:34 +08:00
</WorkflowWithDefaultContext>
</WorkflowContextProvider>
)
}
export default RagPipeline