You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
67 lines
2.1 KiB
TypeScript
67 lines
2.1 KiB
TypeScript
import {
|
|
memo,
|
|
useCallback,
|
|
useMemo,
|
|
} from 'react'
|
|
import { useShallow } from 'zustand/react/shallow'
|
|
import type { HeaderProps } from '@/app/components/workflow/header'
|
|
import Header from '@/app/components/workflow/header'
|
|
import { useStore as useAppStore } from '@/app/components/app/store'
|
|
import {
|
|
fetchWorkflowRunHistory,
|
|
} from '@/service/workflow'
|
|
import ChatVariableTrigger from './chat-variable-trigger'
|
|
import FeaturesTrigger from './features-trigger'
|
|
import { useResetWorkflowVersionHistory } from '@/service/use-workflow'
|
|
import { useIsChatMode } from '../../hooks'
|
|
|
|
const WorkflowHeader = () => {
|
|
const { appDetail, setCurrentLogItem, setShowMessageLogModal } = useAppStore(useShallow(state => ({
|
|
appDetail: state.appDetail,
|
|
setCurrentLogItem: state.setCurrentLogItem,
|
|
setShowMessageLogModal: state.setShowMessageLogModal,
|
|
})))
|
|
const resetWorkflowVersionHistory = useResetWorkflowVersionHistory(appDetail!.id)
|
|
const isChatMode = useIsChatMode()
|
|
|
|
const handleClearLogAndMessageModal = useCallback(() => {
|
|
setCurrentLogItem()
|
|
setShowMessageLogModal(false)
|
|
}, [setCurrentLogItem, setShowMessageLogModal])
|
|
|
|
const viewHistoryProps = useMemo(() => {
|
|
return {
|
|
onClearLogAndMessageModal: handleClearLogAndMessageModal,
|
|
historyUrl: isChatMode ? `/apps/${appDetail!.id}/advanced-chat/workflow-runs` : `/apps/${appDetail!.id}/workflow-runs`,
|
|
historyFetcher: fetchWorkflowRunHistory,
|
|
}
|
|
}, [appDetail, isChatMode, handleClearLogAndMessageModal])
|
|
|
|
const headerProps: HeaderProps = useMemo(() => {
|
|
return {
|
|
normal: {
|
|
components: {
|
|
left: <ChatVariableTrigger />,
|
|
middle: <FeaturesTrigger />,
|
|
},
|
|
runAndHistoryProps: {
|
|
showRunButton: !isChatMode,
|
|
showPreviewButton: isChatMode,
|
|
viewHistoryProps,
|
|
},
|
|
},
|
|
viewHistory: {
|
|
viewHistoryProps,
|
|
},
|
|
restoring: {
|
|
onRestoreSettled: resetWorkflowVersionHistory,
|
|
},
|
|
}
|
|
}, [resetWorkflowVersionHistory, isChatMode, viewHistoryProps])
|
|
return (
|
|
<Header {...headerProps} />
|
|
)
|
|
}
|
|
|
|
export default memo(WorkflowHeader)
|