run

web/app/components/workflow/header/index.tsx

@@ -5,7 +5,10 @@ import {
 } from 'react'
 import { useTranslation } from 'react-i18next'
 import { useStore } from '../store'
-import { useIsChatMode } from '../hooks'
+import {
+  useIsChatMode,
+  useWorkflow,
+} from '../hooks'
 import RunAndHistory from './run-and-history'
 import EditingTitle from './editing-title'
 import RunningTitle from './running-title'
@@ -21,11 +24,16 @@ const Header: FC = () => {
   const appSidebarExpand = useAppStore(s => s.appSidebarExpand)
   const isChatMode = useIsChatMode()
   const runningStatus = useStore(s => s.runningStatus)
+  const { handleRunInit } = useWorkflow()
 
   const handleShowFeatures = useCallback(() => {
     useStore.setState({ showFeaturesPanel: true })
   }, [])
 
+  const handleGoBackToEdit = useCallback(() => {
+    handleRunInit(true)
+  }, [handleRunInit])
+
   return (
     <div
       className='absolute top-0 left-0 flex items-center justify-between px-3 w-full h-14 z-10'
@@ -54,7 +62,7 @@ const Header: FC = () => {
                 mr-2 px-3 py-0 h-8 bg-white text-[13px] font-medium text-primary-600
                 border-[0.5px] border-gray-200 shadow-xs
               `}
-              onClick={() => useStore.setState({ runningStatus: undefined })}
+              onClick={handleGoBackToEdit}
             >
               <ArrowNarrowLeft className='mr-1 w-4 h-4' />
               {t('workflow.common.goBackToEdit')}

web/app/components/workflow/header/run-and-history.tsx

@@ -2,7 +2,10 @@ import type { FC } from 'react'
 import { memo } from 'react'
 import { useTranslation } from 'react-i18next'
 import { useStore } from '../store'
-import { useIsChatMode } from '../hooks'
+import {
+  useIsChatMode,
+  useWorkflow,
+} from '../hooks'
 import { WorkflowRunningStatus } from '../types'
 import { Play } from '@/app/components/base/icons/src/vender/line/mediaAndDevices'
 import { ClockPlay } from '@/app/components/base/icons/src/vender/line/time'
@@ -51,11 +54,12 @@ RunMode.displayName = 'RunMode'
 
 const PreviewMode = memo(() => {
   const { t } = useTranslation()
+  const { handleRunInit } = useWorkflow()
   const runningStatus = useStore(s => s.runningStatus)
   const isRunning = runningStatus === WorkflowRunningStatus.Running
 
   const handleClick = () => {
-    useStore.setState({ runningStatus: WorkflowRunningStatus.Succeeded })
+    handleRunInit()
   }
 
   return (