✨ Add support for array type in Generate variables actions

baptisteArno · baptisteArno · commit de7f96ac0caf · 2025-07-07T16:13:19.000+02:00
Closes baptisteArno#1870
diff --git a/packages/ai/src/parseGenerateVariablesOptions.ts b/packages/ai/src/parseGenerateVariablesOptions.ts
@@ -37,6 +37,11 @@ export const variablesToExtractSchema = option
           type: option.literal("boolean"),
         })
         .extend(extractInfoBaseShape),
+      option
+        .object({
+          type: option.literal("array"),
+        })
+        .extend(extractInfoBaseShape),
       option
         .object({
           type: option.literal("enum"),
diff --git a/packages/ai/src/runGenerateVariables.ts b/packages/ai/src/runGenerateVariables.ts
@@ -85,6 +85,9 @@ const convertVariablesToExtractToSchema = ({
       case "boolean":
         shape[matchingVariable.name] = z.boolean();
         break;
+      case "array":
+        shape[matchingVariable.name] = z.array(z.string());
+        break;
       case "enum": {
         if (!variableToExtract.values || variableToExtract.values.length === 0)
           return;
