show test run panel

pull/21398/head
zxhlyh 12 months ago
parent fa8ab4ea04
commit 3f7f21ce70

@ -5,10 +5,10 @@ import { useStore } from '@/app/components/workflow/store'
import TestRunPanel from './test-run' import TestRunPanel from './test-run'
const RagPipelinePanelOnRight = () => { const RagPipelinePanelOnRight = () => {
const showTestRunPanel = useStore(s => s.showTestRunPanel) const showDebugAndPreviewPanel = useStore(s => s.showDebugAndPreviewPanel)
return ( return (
<> <>
{showTestRunPanel && <TestRunPanel />} {showDebugAndPreviewPanel && <TestRunPanel />}
</> </>
) )
} }

@ -1,4 +1,4 @@
import { useStore } from '@/app/components/workflow/store' import { useWorkflowStore } from '@/app/components/workflow/store'
import { RiCloseLine } from '@remixicon/react' import { RiCloseLine } from '@remixicon/react'
import { useCallback, useMemo, useState } from 'react' import { useCallback, useMemo, useState } from 'react'
import StepIndicator from './step-indicator' import StepIndicator from './step-indicator'
@ -20,6 +20,7 @@ import Actions from './data-source/actions'
import DocumentProcessing from './document-processing' import DocumentProcessing from './document-processing'
const TestRunPanel = () => { const TestRunPanel = () => {
const workflowStore = useWorkflowStore()
const [currentStep, setCurrentStep] = useState(2) const [currentStep, setCurrentStep] = useState(2)
const [dataSource, setDataSource] = useState<string>(DataSourceProvider.waterCrawl) const [dataSource, setDataSource] = useState<string>(DataSourceProvider.waterCrawl)
const [fileList, setFiles] = useState<FileItem[]>([]) const [fileList, setFiles] = useState<FileItem[]>([])
@ -28,7 +29,6 @@ const TestRunPanel = () => {
const [websiteCrawlJobId, setWebsiteCrawlJobId] = useState('') const [websiteCrawlJobId, setWebsiteCrawlJobId] = useState('')
const [crawlOptions, setCrawlOptions] = useState<CrawlOptions>(DEFAULT_CRAWL_OPTIONS) const [crawlOptions, setCrawlOptions] = useState<CrawlOptions>(DEFAULT_CRAWL_OPTIONS)
const setShowTestRunPanel = useStore(s => s.setShowTestRunPanel)
const plan = useProviderContextSelector(state => state.plan) const plan = useProviderContextSelector(state => state.plan)
const enableBilling = useProviderContextSelector(state => state.enableBilling) const enableBilling = useProviderContextSelector(state => state.enableBilling)
@ -60,7 +60,8 @@ const TestRunPanel = () => {
}, [dataSource, nextDisabled, isShowVectorSpaceFull, notionPages.length, websitePages.length]) }, [dataSource, nextDisabled, isShowVectorSpaceFull, notionPages.length, websitePages.length])
const handleClose = () => { const handleClose = () => {
setShowTestRunPanel?.(false) const { setShowDebugAndPreviewPanel } = workflowStore.getState()
setShowDebugAndPreviewPanel(false)
} }
const handleDataSourceSelect = useCallback((option: string) => { const handleDataSourceSelect = useCallback((option: string) => {

@ -31,16 +31,15 @@ export const useWorkflowStartRun = () => {
const startNode = nodes.find(node => node.data.type === BlockEnum.Start) const startNode = nodes.find(node => node.data.type === BlockEnum.Start)
const startVariables = startNode?.data.variables || [] const startVariables = startNode?.data.variables || []
const { const {
showTestRunPanel, showDebugAndPreviewPanel,
setShowInputsPanel, setShowInputsPanel,
setShowEnvPanel, setShowEnvPanel,
setShowTestRunPanel, setShowDebugAndPreviewPanel,
} = workflowStore.getState() } = workflowStore.getState()
setShowEnvPanel(false) setShowEnvPanel(false)
if (showTestRunPanel) { if (showDebugAndPreviewPanel) {
setShowTestRunPanel?.(false)
handleCancelDebugAndPreviewPanel() handleCancelDebugAndPreviewPanel()
return return
} }
@ -48,11 +47,11 @@ export const useWorkflowStartRun = () => {
if (!startVariables.length) { if (!startVariables.length) {
await doSyncWorkflowDraft() await doSyncWorkflowDraft()
handleRun({ inputs: {}, files: [] }) handleRun({ inputs: {}, files: [] })
setShowTestRunPanel?.(true) setShowDebugAndPreviewPanel(true)
setShowInputsPanel(false) setShowInputsPanel(false)
} }
else { else {
setShowTestRunPanel?.(true) setShowDebugAndPreviewPanel(true)
setShowInputsPanel(true) setShowInputsPanel(true)
} }
}, [store, workflowStore, handleCancelDebugAndPreviewPanel, handleRun, doSyncWorkflowDraft]) }, [store, workflowStore, handleCancelDebugAndPreviewPanel, handleRun, doSyncWorkflowDraft])

@ -6,8 +6,6 @@ export type RagPipelineSliceShape = {
setShowInputFieldDialog: (showInputFieldPanel: boolean) => void setShowInputFieldDialog: (showInputFieldPanel: boolean) => void
nodesDefaultConfigs: Record<string, any> nodesDefaultConfigs: Record<string, any>
setNodesDefaultConfigs: (nodesDefaultConfigs: Record<string, any>) => void setNodesDefaultConfigs: (nodesDefaultConfigs: Record<string, any>) => void
showTestRunPanel: boolean
setShowTestRunPanel: (showTestRunPanel: boolean) => void
} }
export type CreateRagPipelineSliceSlice = StateCreator<RagPipelineSliceShape> export type CreateRagPipelineSliceSlice = StateCreator<RagPipelineSliceShape>
@ -17,6 +15,4 @@ export const createRagPipelineSliceSlice: StateCreator<RagPipelineSliceShape> =
setShowInputFieldDialog: showInputFieldDialog => set(() => ({ showInputFieldDialog })), setShowInputFieldDialog: showInputFieldDialog => set(() => ({ showInputFieldDialog })),
nodesDefaultConfigs: {}, nodesDefaultConfigs: {},
setNodesDefaultConfigs: nodesDefaultConfigs => set(() => ({ nodesDefaultConfigs })), setNodesDefaultConfigs: nodesDefaultConfigs => set(() => ({ nodesDefaultConfigs })),
showTestRunPanel: false,
setShowTestRunPanel: showTestRunPanel => set(() => ({ showTestRunPanel })),
}) })

Loading…
Cancel
Save