fix: agent node output vars error (#12931)

pull/12934/head
Joel 1 year ago committed by GitHub
parent 6ee4eba86b
commit c8b4a76530
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -48,6 +48,7 @@ const AgentPanel: FC<NodePanelProps<AgentNodeType>> = (props) => {
runInputData, runInputData,
setRunInputData, setRunInputData,
varInputs, varInputs,
outputSchema,
} = useConfig(props.id, props.data) } = useConfig(props.id, props.data)
const { t } = useTranslation() const { t } = useTranslation()
const nodeInfo = useMemo(() => { const nodeInfo = useMemo(() => {
@ -121,12 +122,12 @@ const AgentPanel: FC<NodePanelProps<AgentNodeType>> = (props) => {
type='Array[Object]' type='Array[Object]'
description={t(`${i18nPrefix}.outputVars.json`)} description={t(`${i18nPrefix}.outputVars.json`)}
/> />
{inputs.output_schema && Object.entries(inputs.output_schema).map(([name, schema]) => ( {outputSchema.map(({ name, type, description }) => (
<VarItem <VarItem
key={name} key={name}
name={name} name={name}
type={schema.type} type={type}
description={schema.description} description={description}
/> />
))} ))}
</OutputVars> </OutputVars>

@ -158,6 +158,23 @@ const useConfig = (id: string, payload: AgentNodeType) => {
return vars return vars
})() })()
const outputSchema = useMemo(() => {
const res: any[] = []
if (!inputs.output_schema)
return []
Object.keys(inputs.output_schema.properties).forEach((outputKey) => {
const output = inputs.output_schema.properties[outputKey]
res.push({
name: outputKey,
type: output.type === 'array'
? `Array[${output.items?.type.slice(0, 1).toLocaleUpperCase()}${output.items?.type.slice(1)}]`
: `${output.type.slice(0, 1).toLocaleUpperCase()}${output.type.slice(1)}`,
description: output.description,
})
})
return res
}, [inputs.output_schema])
return { return {
readOnly, readOnly,
inputs, inputs,
@ -184,6 +201,7 @@ const useConfig = (id: string, payload: AgentNodeType) => {
setRunInputData, setRunInputData,
runResult, runResult,
varInputs, varInputs,
outputSchema,
} }
} }

Loading…
Cancel
Save