dify/web/app/components/rag-pipeline/index.tsx

import type { Features as FeaturesData } from '@/app/components/base/features/types'
import type { InjectWorkflowStoreSliceFn } from '@/app/components/workflow/store'
import { useMemo } from 'react'
import { FeaturesProvider } from '@/app/components/base/features'
import Loading from '@/app/components/base/loading'
import WorkflowWithDefaultContext from '@/app/components/workflow'
import {
  WorkflowContextProvider,
} from '@/app/components/workflow/context'
import {
  initialEdges,
  initialNodes,
} from '@/app/components/workflow/utils'
import { useDatasetDetailContextWithSelector } from '@/context/dataset-detail'
import Conversion from './components/conversion'
import RagPipelineMain from './components/rag-pipeline-main'
import { usePipelineInit } from './hooks'
import { createRagPipelineSliceSlice } from './store'
import { processNodesWithoutDataSource } from './utils'

const RagPipeline = () => {
  const {
    data,
    isLoading,
  } = usePipelineInit()
  const nodesData = useMemo(() => {
    if (data)
      return initialNodes(data.graph.nodes, data.graph.edges)

    return []
  }, [data])
  const edgesData = useMemo(() => {
    if (data)
      return initialEdges(data.graph.edges, data.graph.nodes)

    return []
  }, [data])

  const initialFeatures = useMemo(() => {
    const features = data?.features || {}
    return {
      sandbox: features.sandbox || { enabled: false },
    } as FeaturesData
  }, [data?.features])

  if (!data || isLoading) {
    return (
      <div className="relative flex h-full w-full items-center justify-center">
        <Loading />
      </div>
    )
  }

  const {
    nodes: processedNodes,
    viewport,
  } = processNodesWithoutDataSource(nodesData, data.graph.viewport)
  return (
    <WorkflowWithDefaultContext
      edges={edgesData}
      nodes={processedNodes}
    >
      <FeaturesProvider features={initialFeatures}>
        <RagPipelineMain
          edges={edgesData}
          nodes={processedNodes}
          viewport={viewport}
        />
      </FeaturesProvider>
    </WorkflowWithDefaultContext>
  )
}

const RagPipelineWrapper = () => {
  const pipelineId = useDatasetDetailContextWithSelector(s => s.dataset?.pipeline_id)

  if (!pipelineId)
    return <Conversion />

  return (
    <WorkflowContextProvider
      injectWorkflowStoreSliceFn={createRagPipelineSliceSlice as InjectWorkflowStoreSliceFn}
    >
      <RagPipeline />
    </WorkflowContextProvider>
  )
}

export default RagPipelineWrapper