🚑️ Fix set variable return injection for multi line statements

baptisteArno · Oct 23, 2024 · caa397d · caa397d
1 parent 4fdc80f
commit caa397d
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/packages/bot-engine/src/blocks/logic/setVariable/executeSetVariable.ts b/packages/bot-engine/src/blocks/logic/setVariable/executeSetVariable.ts
@@ -151,11 +151,8 @@ const evaluateSetVariableExpression =
     return { value: output };
   };
 
-const injectReturnKeywordIfNeeded = (code: string) => {
-  const isOneLineCode = code.split("\n").length === 1;
-  if (!isOneLineCode) return code;
-  return code.startsWith("return ") ? code : `return ${code}`;
-};
+const injectReturnKeywordIfNeeded = (code: string) =>
+  code.includes("return ") ? code : `return ${code}`;
 
 const getExpressionToEvaluate =
   (state: SessionState) =>