mirror of
https://github.com/langgenius/dify.git
synced 2025-07-15 21:24:23 +00:00

The frontend of feat: Persist Variables for Enhanced Debugging Workflow (#20699). Co-authored-by: jZonG <jzongcode@gmail.com>
64 lines
1.7 KiB
TypeScript
64 lines
1.7 KiB
TypeScript
import type { MutableRefObject } from 'react'
|
|
import { useTranslation } from 'react-i18next'
|
|
import type { InputVar, Variable } from '@/app/components/workflow/types'
|
|
import { InputVarType } from '@/app/components/workflow/types'
|
|
import { useCallback, useMemo } from 'react'
|
|
import type { KnowledgeRetrievalNodeType } from './types'
|
|
|
|
const i18nPrefix = 'workflow.nodes.knowledgeRetrieval'
|
|
|
|
type Params = {
|
|
id: string,
|
|
payload: KnowledgeRetrievalNodeType
|
|
runInputData: Record<string, any>
|
|
runInputDataRef: MutableRefObject<Record<string, any>>
|
|
getInputVars: (textList: string[]) => InputVar[]
|
|
setRunInputData: (data: Record<string, any>) => void
|
|
toVarInputs: (variables: Variable[]) => InputVar[]
|
|
}
|
|
const useSingleRunFormParams = ({
|
|
payload,
|
|
runInputData,
|
|
setRunInputData,
|
|
}: Params) => {
|
|
const { t } = useTranslation()
|
|
const query = runInputData.query
|
|
const setQuery = useCallback((newQuery: string) => {
|
|
setRunInputData({
|
|
...runInputData,
|
|
query: newQuery,
|
|
})
|
|
}, [runInputData, setRunInputData])
|
|
|
|
const forms = useMemo(() => {
|
|
return [
|
|
{
|
|
inputs: [{
|
|
label: t(`${i18nPrefix}.queryVariable`)!,
|
|
variable: 'query',
|
|
type: InputVarType.paragraph,
|
|
required: true,
|
|
}],
|
|
values: { query },
|
|
onChange: (keyValue: Record<string, any>) => setQuery(keyValue.query),
|
|
},
|
|
]
|
|
}, [query, setQuery, t])
|
|
|
|
const getDependentVars = () => {
|
|
return [payload.query_variable_selector]
|
|
}
|
|
const getDependentVar = (variable: string) => {
|
|
if(variable === 'query')
|
|
return payload.query_variable_selector
|
|
}
|
|
|
|
return {
|
|
forms,
|
|
getDependentVars,
|
|
getDependentVar,
|
|
}
|
|
}
|
|
|
|
export default useSingleRunFormParams
|