feat: knowledge run var

pull/21369/head
Joel 1 year ago
parent d6d8d25026
commit 64c52d16dc

@ -3,11 +3,13 @@ import { useTranslation } from 'react-i18next'
import type { InputVar, Variable } from '@/app/components/workflow/types' import type { InputVar, Variable } from '@/app/components/workflow/types'
import { InputVarType } from '@/app/components/workflow/types' import { InputVarType } from '@/app/components/workflow/types'
import { useCallback, useMemo } from 'react' import { useCallback, useMemo } from 'react'
import type { KnowledgeRetrievalNodeType } from './types'
const i18nPrefix = 'workflow.nodes.knowledgeRetrieval' const i18nPrefix = 'workflow.nodes.knowledgeRetrieval'
type Params = { type Params = {
id: string, id: string,
payload: KnowledgeRetrievalNodeType
runInputData: Record<string, any> runInputData: Record<string, any>
runInputDataRef: MutableRefObject<Record<string, any>> runInputDataRef: MutableRefObject<Record<string, any>>
getInputVars: (textList: string[]) => InputVar[] getInputVars: (textList: string[]) => InputVar[]
@ -15,6 +17,7 @@ type Params = {
toVarInputs: (variables: Variable[]) => InputVar[] toVarInputs: (variables: Variable[]) => InputVar[]
} }
const useSingleRunFormParams = ({ const useSingleRunFormParams = ({
payload,
runInputData, runInputData,
setRunInputData, setRunInputData,
}: Params) => { }: Params) => {
@ -42,8 +45,19 @@ const useSingleRunFormParams = ({
] ]
}, [query, setQuery, t]) }, [query, setQuery, t])
const getDependentVars = () => {
console.log(payload.query_variable_selector)
return [payload.query_variable_selector]
}
const getDependentVar = (variable: string) => {
if(variable === 'query')
return payload.query_variable_selector
}
return { return {
forms, forms,
getDependentVars,
getDependentVar,
} }
} }

Loading…
Cancel
Save