agent-state-machine 2.0.10 → 2.0.12

package/README.md

@@ -56,6 +56,7 @@ Workflows live in:
 ```text
 workflows/<name>/
 ├── workflow.js        # Native JS workflow (async/await)
+├── config.js          # Model/API key configuration
 ├── package.json       # Sets "type": "module" for this workflow folder
 ├── agents/            # Custom agents (.js/.mjs/.cjs or .md)
 ├── interactions/      # Human-in-the-loop files (auto-created)
@@ -67,6 +68,8 @@ workflows/<name>/
 
 ## Writing workflows (native JS)
 
+Edit `config.js` to set models and API keys for the workflow.
+
 ```js
 /**
 /**
@@ -83,20 +86,6 @@ workflows/<name>/
 import { agent, memory, askHuman, parallel } from 'agent-state-machine';
 import { notify } from './scripts/mac-notification.js';
 
-// Model configuration (also supports models in a separate config export)
-export const config = {
-  models: {
-    low: "gemini",
-    med: "codex --model gpt-5.2",
-    high: "claude -m claude-opus-4-20250514 -p",
-  },
-  apiKeys: {
-    gemini: process.env.GEMINI_API_KEY,
-    anthropic: process.env.ANTHROPIC_API_KEY,
-    openai: process.env.OPENAI_API_KEY,
-  }
-};
-
 export default async function() {
   console.log('Starting project-builder workflow...');
 
@@ -298,7 +287,7 @@ export const config = {
 };
 ```
 
-The runtime captures the fully-built prompt in `state/history.jsonl`, viewable in the browser with live updates when running with the `--local` flag or via the remote URL. Remote follow links persist across runs (stored in `workflow.js` config) unless you pass `-n`/`--new` to regenerate.
+The runtime captures the fully-built prompt in `state/history.jsonl`, viewable in the browser with live updates when running with the `--local` flag or via the remote URL. Remote follow links persist across runs (stored in `config.js`) unless you pass `-n`/`--new` to regenerate.
 
 ---
 

package/bin/cli.js

@@ -2,6 +2,7 @@
 
 import path from 'path';
 import fs from 'fs';
+import readline from 'readline';
 import { pathToFileURL, fileURLToPath } from 'url';
 import { WorkflowRuntime } from '../lib/index.js';
 import { setup } from '../lib/setup.js';
@@ -62,6 +63,7 @@ Environment Variables:
 Workflow Structure:
   workflows/<name>/
   ├── workflow.js        # Native JS workflow (async/await)
+  ├── config.js          # Model/API key configuration
   ├── package.json       # Sets "type": "module" for this workflow folder
   ├── agents/            # Custom agents (.js/.mjs/.cjs or .md)
   ├── interactions/      # Human-in-the-loop files (auto-created)
@@ -70,6 +72,28 @@ Workflow Structure:
 `);
 }
 
+async function confirmHardReset(workflowName) {
+  if (!process.stdin.isTTY) {
+    console.error('Error: Hard reset requires confirmation in a TTY.');
+    return false;
+  }
+
+  const rl = readline.createInterface({ input: process.stdin, output: process.stdout });
+  const ask = (q) => new Promise((resolve) => rl.question(q, resolve));
+  try {
+    const answer = String(
+      await ask(
+        `Hard reset deletes history, interactions, and memory for '${workflowName}'. Type 'y' to continue: `
+      )
+    )
+      .trim()
+      .toLowerCase();
+    return answer === 'y' || answer === 'yes';
+  } finally {
+    rl.close();
+  }
+}
+
 function workflowsRoot() {
   return path.join(process.cwd(), 'workflows');
 }
@@ -162,8 +186,8 @@ function findConfigObjectRange(source) {
   return null;
 }
 
-function readRemotePathFromWorkflow(workflowFile) {
-  const source = fs.readFileSync(workflowFile, 'utf-8');
+function readRemotePathFromConfig(configFile) {
+  const source = fs.readFileSync(configFile, 'utf-8');
   const range = findConfigObjectRange(source);
   if (!range) return null;
   const configSource = source.slice(range.start, range.end + 1);
@@ -171,19 +195,19 @@ function readRemotePathFromWorkflow(workflowFile) {
   return match ? match[2] : null;
 }
 
-function writeRemotePathToWorkflow(workflowFile, remotePath) {
-  const source = fs.readFileSync(workflowFile, 'utf-8');
+function writeRemotePathToConfig(configFile, remotePath) {
+  const source = fs.readFileSync(configFile, 'utf-8');
   const range = findConfigObjectRange(source);
   const remoteLine = `remotePath: "${remotePath}"`;
 
   if (!range) {
     const hasConfigExport = /export\s+const\s+config\s*=/.test(source);
     if (hasConfigExport) {
-      throw new Error('Workflow config export is not an object literal; add remotePath manually.');
+      throw new Error('Config export is not an object literal; add remotePath manually.');
     }
     const trimmed = source.replace(/\s*$/, '');
     const appended = `${trimmed}\n\nexport const config = {\n  ${remoteLine}\n};\n`;
-    fs.writeFileSync(workflowFile, appended);
+    fs.writeFileSync(configFile, appended);
     return;
   }
 
@@ -224,15 +248,15 @@ function writeRemotePathToWorkflow(workflowFile, remotePath) {
     source.slice(0, range.start) +
     updatedConfigSource +
     source.slice(range.end + 1);
-  fs.writeFileSync(workflowFile, updatedSource);
+  fs.writeFileSync(configFile, updatedSource);
 }
 
-function ensureRemotePath(workflowFile, { forceNew = false } = {}) {
-  const existing = readRemotePathFromWorkflow(workflowFile);
+function ensureRemotePath(configFile, { forceNew = false } = {}) {
+  const existing = readRemotePathFromConfig(configFile);
   if (existing && !forceNew) return existing;
 
   const remotePath = generateSessionToken();
-  writeRemotePathToWorkflow(workflowFile, remotePath);
+  writeRemotePathToConfig(configFile, remotePath);
   return remotePath;
 }
 
@@ -253,6 +277,7 @@ function summarizeStatus(state) {
   if (s === 'COMPLETED') return ' [completed]';
   if (s === 'FAILED') return ' [failed - can resume]';
   if (s === 'PAUSED') return ' [paused - can resume]';
+  if (s === 'STOPPED') return ' [stopped - can resume]';
   if (s === 'RUNNING') return ' [running]';
   if (s === 'IDLE') return ' [idle]';
   return state.status ? ` [${state.status}]` : '';
@@ -327,12 +352,18 @@ async function runOrResume(
 
   const runtime = new WorkflowRuntime(workflowDir);
   if (preResetHard) {
+    const confirmed = await confirmHardReset(workflowName);
+    if (!confirmed) {
+      console.log('Hard reset cancelled.');
+      return;
+    }
     runtime.resetHard();
   } else if (preReset) {
     runtime.reset();
   }
 
   const workflowUrl = pathToFileURL(entry).href;
+  const configFile = path.join(workflowDir, 'config.js');
 
   let localServer = null;
   let remoteUrl = null;
@@ -354,7 +385,7 @@ async function runOrResume(
 
   // Enable remote follow mode if we have a URL
   if (remoteUrl) {
-    const sessionToken = ensureRemotePath(entry, { forceNew: forceNewRemotePath });
+    const sessionToken = ensureRemotePath(configFile, { forceNew: forceNewRemotePath });
     await runtime.enableRemote(remoteUrl, { sessionToken });
   }
 
@@ -478,6 +509,11 @@ async function main() {
       {
         const workflowDir = resolveWorkflowDir(workflowName);
         const runtime = new WorkflowRuntime(workflowDir);
+        const confirmed = await confirmHardReset(workflowName);
+        if (!confirmed) {
+          console.log('Hard reset cancelled.');
+          process.exit(0);
+        }
         runtime.resetHard();
       }
       break;

package/lib/llm.js

@@ -314,7 +314,7 @@ async function executeAPI(provider, model, prompt, apiKey, options = {}) {
  *
  * @param {object} context - The workflow context (contains _config, _steering, etc.)
  * @param {object} options - Options for the LLM call
- * @param {string} options.model - Model key from workflow.js models config
+ * @param {string} options.model - Model key from config.js models config
  * @param {string} options.prompt - The prompt to send
  * @param {boolean} options.includeContext - Whether to include context in prompt (default: true)
  * @param {number} options.maxTokens - Max tokens for API calls (default: 4096)
@@ -359,7 +359,7 @@ export async function llm(context, options) {
 
     if (!apiKey) {
       throw new Error(
-        `No API key found for ${provider}. Set in workflow.js apiKeys or ${provider.toUpperCase()}_API_KEY env var`
+        `No API key found for ${provider}. Set in config.js apiKeys or ${provider.toUpperCase()}_API_KEY env var`
       );
     }
 
@@ -459,4 +459,4 @@ export async function llmJSON(context, options) {
     ...response,
     data: parseJSON(response.text)
   };
-}
+}

package/lib/runtime/runtime.js

@@ -9,6 +9,7 @@
 import fs from 'fs';
 import path from 'path';
 import readline from 'readline';
+import { pathToFileURL } from 'url';
 import { createMemoryProxy } from './memory.js';
 import { RemoteClient } from '../remote/client.js';
 
@@ -218,6 +219,50 @@ export class WorkflowRuntime {
   async runWorkflow(workflowPath) {
     setCurrentRuntime(this);
 
+    // Handle Ctrl+C and termination signals to update status before exit
+    const handleShutdown = async (signal) => {
+      this.status = 'STOPPED';
+      this._error = `Workflow interrupted by ${signal}`;
+      this.persist();
+
+      // Log to history (local file)
+      const historyEntry = {
+        timestamp: new Date().toISOString(),
+        event: 'WORKFLOW_STOPPED',
+        reason: signal
+      };
+      const line = JSON.stringify(historyEntry) + '\n';
+      let existing = '';
+      if (fs.existsSync(this.historyFile)) {
+        existing = fs.readFileSync(this.historyFile, 'utf-8');
+      }
+      fs.writeFileSync(this.historyFile, line + existing);
+
+      // Send to remote and wait for it to complete before exiting
+      if (this.remoteClient && this.remoteEnabled) {
+        try {
+          await this.remoteClient.sendEvent(historyEntry);
+        } catch {
+          // Ignore errors during shutdown
+        }
+      }
+
+      console.log(`\n${C.yellow}⚠ Workflow '${this.workflowName}' stopped (${signal})${C.reset}`);
+      cleanupSignalHandlers();
+      clearCurrentRuntime();
+      process.exit(130); // 128 + SIGINT (2)
+    };
+
+    const sigintHandler = () => handleShutdown('SIGINT');
+    const sigtermHandler = () => handleShutdown('SIGTERM');
+    process.on('SIGINT', sigintHandler);
+    process.on('SIGTERM', sigtermHandler);
+
+    const cleanupSignalHandlers = () => {
+      process.removeListener('SIGINT', sigintHandler);
+      process.removeListener('SIGTERM', sigtermHandler);
+    };
+
     try {
       this.status = 'RUNNING';
       this._error = null;
@@ -226,18 +271,24 @@ export class WorkflowRuntime {
 
       this.prependHistory({ event: 'WORKFLOW_STARTED' });
 
-      // Import workflow module
-      const workflowModule = await import(workflowPath);
-      const runFn = workflowModule.default || workflowModule.run || workflowModule;
-
-      // Load workflow config from module export
-      const cfg = workflowModule.config || {};
+      const configPath = path.join(this.workflowDir, 'config.js');
+      if (!fs.existsSync(configPath)) {
+        throw new Error(`config.js not found in ${this.workflowDir}`);
+      }
+      const configUrl = pathToFileURL(configPath);
+      configUrl.searchParams.set('t', Date.now().toString());
+      const configModule = await import(configUrl.href);
+      const cfg = configModule.config || configModule.default || {};
       this.workflowConfig = {
         models: cfg.models || {},
         apiKeys: cfg.apiKeys || {},
         description: cfg.description || ''
       };
 
+      // Import workflow module
+      const workflowModule = await import(workflowPath);
+      const runFn = workflowModule.default || workflowModule.run || workflowModule;
+
       if (typeof runFn !== 'function') {
         throw new Error('Workflow module must export a default async function');
       }
@@ -263,6 +314,7 @@ export class WorkflowRuntime {
       console.error(`\n${C.red}✗ Workflow '${this.workflowName}' failed: ${err.message}${C.reset}`);
       throw err;
     } finally {
+      cleanupSignalHandlers();
       clearCurrentRuntime();
     }
   }
@@ -406,6 +458,7 @@ export class WorkflowRuntime {
     let statusColor = C.reset;
     if (this.status === 'COMPLETED') statusColor = C.green;
     if (this.status === 'FAILED') statusColor = C.red;
+    if (this.status === 'STOPPED') statusColor = C.yellow;
     if (this.status === 'RUNNING') statusColor = C.blue;
     if (this.status === 'IDLE') statusColor = C.gray;
     

package/lib/setup.js

@@ -62,20 +62,6 @@ async function setup(workflowName) {
 import { agent, memory, askHuman, parallel } from 'agent-state-machine';
 import { notify } from './scripts/mac-notification.js';
 
-// Model configuration (also supports models in a separate config export)
-export const config = {
-  models: {
-    low: "gemini",
-    med: "codex --model gpt-5.2",
-    high: "claude -m claude-opus-4-20250514 -p",
-  },
-  apiKeys: {
-    gemini: process.env.GEMINI_API_KEY,
-    anthropic: process.env.ANTHROPIC_API_KEY,
-    openai: process.env.OPENAI_API_KEY,
-  }
-};
-
 export default async function() {
   console.log('Starting ${workflowName} workflow...');
 
@@ -119,6 +105,24 @@ export default async function() {
   fs.writeFileSync(workflowFile, workflowJs);
   console.log(`  Created: ${path.relative(process.cwd(), workflowFile)}`);
 
+  const configJs = `export const config = {
+  models: {
+    low: "gemini",
+    med: "codex --model gpt-5.2",
+    high: "claude -m claude-opus-4-20250514 -p",
+  },
+  apiKeys: {
+    gemini: process.env.GEMINI_API_KEY,
+    anthropic: process.env.ANTHROPIC_API_KEY,
+    openai: process.env.OPENAI_API_KEY,
+  }
+};
+`;
+
+  const configFile = path.join(workflowDir, 'config.js');
+  fs.writeFileSync(configFile, configJs);
+  console.log(`  Created: ${path.relative(process.cwd(), configFile)}`);
+
   // Create example JS agent (ESM)
   // Create example JS agent (ESM)
   const exampleAgent = `/**
@@ -138,7 +142,7 @@ export default async function handler(context) {
     console.log('[Agent: example] Steering loaded (' + context._steering.global.length + ' chars)');
   }
 
-  // Example: Call an LLM (configure models in workflow.js)
+  // Example: Call an LLM (configure models in config.js)
   // const response = await llm(context, {
   //   model: 'smart',
   //   prompt: 'Say hello and describe what you can help with.'
@@ -295,6 +299,7 @@ A workflow created with agent-state-machine (native JS format).
 \\\`\\\`\\\`
 ${workflowName}/
 ├── workflow.js      # Native JS workflow (async/await)
+├── config.js        # Model/API key configuration
 ├── package.json     # Sets "type": "module" for this workflow folder
 ├── agents/          # Custom agents (.js/.mjs/.cjs or .md)
 ├── interactions/    # Human-in-the-loop inputs (created at runtime)
@@ -304,6 +309,8 @@ ${workflowName}/
 
 ## Usage
 
+Edit \`config.js\` to set models and API keys for this workflow.
+
 Run the workflow (or resume if interrupted):
 \\\`\\\`\\\`bash
 state-machine run ${workflowName}
@@ -411,8 +418,9 @@ Generate a greeting for {{name}}.
   console.log(`\n✓ Workflow '${workflowName}' created successfully!\n`);
   console.log('Next steps:');
   console.log(`  1. Edit workflows/${workflowName}/workflow.js to implement your flow`);
-  console.log(`  2. Add custom agents in workflows/${workflowName}/agents/`);
-  console.log(`  3. Run: state-machine run ${workflowName}\n`);
+  console.log(`  2. Edit workflows/${workflowName}/config.js to set models/API keys`);
+  console.log(`  3. Add custom agents in workflows/${workflowName}/agents/`);
+  console.log(`  4. Run: state-machine run ${workflowName}\n`);
 }
 
 export { setup };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agent-state-machine",
-  "version": "2.0.10",
+  "version": "2.0.12",
   "type": "module",
   "description": "A workflow orchestrator for running agents and scripts in sequence with state management",
   "main": "lib/index.js",

package/vercel-server/ui/index.html

@@ -486,11 +486,24 @@
 
             useEffect(() => localStorage.setItem("rf_theme", theme), [theme]);
 
+            // Helper to check if workflow is currently running based on history
+            const isWorkflowRunning = (entries) => {
+                // Find the most recent workflow lifecycle event (history is newest-first)
+                for (const entry of entries) {
+                    if (entry.event === "WORKFLOW_STARTED") return true;
+                    if (entry.event === "WORKFLOW_STOPPED" ||
+                        entry.event === "WORKFLOW_COMPLETED" ||
+                        entry.event === "WORKFLOW_FAILED") return false;
+                }
+                return false; // No lifecycle events found
+            };
+
             useEffect(() => {
                 if (history.length === 0) { setPendingInteraction(null); return; }
 
                 const resolvedSlugs = new Set();
                 let pending = null;
+                const workflowRunning = isWorkflowRunning(history);
 
                 for (const entry of history) {
                     const isResolution =
@@ -510,7 +523,8 @@
                     }
                 }
 
-                setPendingInteraction(pending);
+                // Only show pending interaction if workflow is running
+                setPendingInteraction(workflowRunning ? pending : null);
             }, [history]);
 
             const fetchData = async () => {
@@ -520,8 +534,18 @@
                     if (data.entries) setHistory(data.entries);
                     if (data.workflowName) setWorkflowName(data.workflowName);
 
-                    if (token && data.cliConnected !== undefined) setStatus(data.cliConnected ? "connected" : "disconnected");
-                    else if (!token) setStatus("connected");
+                    // Check if workflow is currently running based on most recent lifecycle event
+                    const workflowRunning = isWorkflowRunning(data.entries || []);
+
+                    if (workflowRunning) {
+                        setStatus("connected");
+                    } else if (token && data.cliConnected !== undefined) {
+                        setStatus(data.cliConnected ? "connected" : "disconnected");
+                    } else if (!token) {
+                        setStatus("connected");
+                    } else {
+                        setStatus("disconnected");
+                    }
 
                     setLoading(false);
                     return true;
@@ -569,6 +593,8 @@
                                     break;
                                 case "history":
                                     setHistory(data.entries || []);
+                                    // Update status based on workflow lifecycle
+                                    setStatus(isWorkflowRunning(data.entries || []) ? "connected" : "disconnected");
                                     break;
                                 case "event":
                                     setHistory((prev) => {
@@ -578,6 +604,14 @@
                                         }
                                         return [data, ...prev];
                                     });
+                                    // Update status based on workflow lifecycle events
+                                    if (data.event === "WORKFLOW_STARTED") {
+                                        setStatus("connected");
+                                    } else if (data.event === "WORKFLOW_STOPPED" ||
+                                               data.event === "WORKFLOW_COMPLETED" ||
+                                               data.event === "WORKFLOW_FAILED") {
+                                        setStatus("disconnected");
+                                    }
                                     break;
                                 case "cli_connected":
                                 case "cli_reconnected":