diff --git a/packages/ai/src/parseGenerateVariablesOptions.ts b/packages/ai/src/parseGenerateVariablesOptions.ts index b8bf63fd3..1ebfa4af4 100644 --- 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 index b75ed7178..5881f5cac 100644 --- 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;