feat(cloud): pass env vars to scheduled workflows (#935)

Co-authored-by: Proactive Runtime Bot <agent@agent-relay.com>

Author: khaliqgant

packages/cloud/src/types.ts

@@ -183,6 +183,7 @@ export type ScheduleWorkflowOptions = {
   cron?: string;
   at?: string;
   timezone?: string;
+  envSecrets?: Record<string, string>;
 };
 
 export type WorkflowLogsResponse = {

packages/cloud/src/workflows.test.ts

@@ -495,6 +495,10 @@ describe('workflow schedules', () => {
     const result = await scheduleWorkflow(workflowPath, {
       cron: '0 * * * *',
       name: 'Hourly eval',
+      envSecrets: {
+        AI_CLI_UPDATES_DRY_RUN: 'true',
+        AI_CLI_UPDATES_ONLY: 'codex',
+      },
     });
 
     expect(result.id).toBe('sched-1');
@@ -505,6 +509,10 @@ describe('workflow schedules', () => {
       timezone: 'UTC',
       workflowRequest: {
         fileType: 'yaml',
+        envSecrets: {
+          AI_CLI_UPDATES_DRY_RUN: 'true',
+          AI_CLI_UPDATES_ONLY: 'codex',
+        },
       },
     });
     expect(

packages/cloud/src/workflows.ts

@@ -685,6 +685,9 @@ export async function scheduleWorkflow(
       workflow: input.workflow,
       fileType: input.fileType,
       ...(input.sourceFileType ? { sourceFileType: input.sourceFileType } : {}),
+      ...(options.envSecrets && Object.keys(options.envSecrets).length > 0
+        ? { envSecrets: options.envSecrets }
+        : {}),
     },
   };
   if (options.description?.trim()) {

src/cli/commands/cloud.test.ts

@@ -137,18 +137,46 @@ describe('registerCloudCommands', () => {
       '0 * * * *',
       '--name',
       'Hourly eval',
+      '--env',
+      'AI_CLI_UPDATES_DRY_RUN=true',
+      '--env',
+      'AI_CLI_UPDATES_ONLY=codex',
     ]);
 
     expect(cloudMocks.scheduleWorkflow).toHaveBeenCalledWith(
       'workflow.yaml',
       expect.objectContaining({
         cron: '0 * * * *',
         name: 'Hourly eval',
+        envSecrets: {
+          AI_CLI_UPDATES_DRY_RUN: 'true',
+          AI_CLI_UPDATES_ONLY: 'codex',
+        },
       })
     );
     expect(deps.log).toHaveBeenCalledWith('Schedule created: sched-1');
   });
 
+  it('schedule rejects malformed environment assignments', async () => {
+    const { program } = createHarness();
+
+    await expect(
+      program.parseAsync([
+        'node',
+        'agent-relay',
+        'cloud',
+        'schedule',
+        'workflow.yaml',
+        '--cron',
+        '0 * * * *',
+        '--env',
+        'not-an-assignment',
+      ])
+    ).rejects.toThrow();
+
+    expect(cloudMocks.scheduleWorkflow).not.toHaveBeenCalled();
+  });
+
   it('schedule creates one-time workflow schedules', async () => {
     const { program, deps } = createHarness();
     cloudMocks.scheduleWorkflow.mockResolvedValueOnce({

src/cli/commands/cloud.ts

@@ -87,6 +87,23 @@ function parseWorkflowFileType(value: string): WorkflowFileType {
   throw new InvalidArgumentError('Expected workflow type to be one of: yaml, ts, py');
 }
 
+function parseEnvAssignment(value: string, previous: Record<string, string> = {}): Record<string, string> {
+  const equalsIndex = value.indexOf('=');
+  if (equalsIndex <= 0) {
+    throw new InvalidArgumentError('Expected environment assignment in KEY=VALUE form.');
+  }
+
+  const key = value.slice(0, equalsIndex).trim();
+  if (!/^[A-Za-z_][A-Za-z0-9_]*$/.test(key)) {
+    throw new InvalidArgumentError(`Invalid environment variable name: ${key || '(empty)'}`);
+  }
+
+  return {
+    ...previous,
+    [key]: value.slice(equalsIndex + 1),
+  };
+}
+
 function isObject(value: unknown): value is Record<string, unknown> {
   return value !== null && typeof value === 'object' && !Array.isArray(value);
 }
@@ -421,6 +438,12 @@ export function registerCloudCommands(program: Command, overrides: Partial<Cloud
     .option('--timezone <timezone>', 'IANA timezone for cron schedules', 'UTC')
     .option('--name <name>', 'Schedule name')
     .option('--description <description>', 'Schedule description')
+    .option(
+      '--env <KEY=VALUE>',
+      'Environment variable to pass to scheduled runs; repeat for multiple variables',
+      parseEnvAssignment,
+      {}
+    )
     .option('--json', 'Print raw JSON response', false)
     .action(
       async (
@@ -433,14 +456,19 @@ export function registerCloudCommands(program: Command, overrides: Partial<Cloud
           timezone?: string;
           name?: string;
           description?: string;
+          env?: Record<string, string>;
           json?: boolean;
         }
       ) => {
         const started = Date.now();
         let success = false;
         let errorClass: string | undefined;
         try {
-          const result = await scheduleWorkflow(workflow, options);
+          const { env, ...scheduleOptions } = options;
+          const result = await scheduleWorkflow(workflow, {
+            ...scheduleOptions,
+            ...(env && Object.keys(env).length > 0 ? { envSecrets: env } : {}),
+          });
           if (options.json) {
             deps.log(JSON.stringify(result, null, 2));
           } else {