diff --git a/web/app/components/workflow/nodes/code/use-config.ts b/web/app/components/workflow/nodes/code/use-config.ts index fd72f6b006..a5498d6c62 100644 --- a/web/app/components/workflow/nodes/code/use-config.ts +++ b/web/app/components/workflow/nodes/code/use-config.ts @@ -58,9 +58,12 @@ const useConfig = (id: string, payload: CodeNodeType) => { }, [inputs, setInputs]) const handleCodeLanguageChange = useCallback((codeLanguage: CodeLanguage) => { + const currDefaultConfig = allLanguageDefault?.[codeLanguage] + const newInputs = produce(inputs, (draft) => { - const currDefaultConfig = allLanguageDefault![codeLanguage] draft.code_language = codeLanguage + if (!currDefaultConfig) + return draft.code = currDefaultConfig.code draft.variables = currDefaultConfig.variables draft.outputs = currDefaultConfig.outputs