foliko 2.0.22 → 2.0.24

package/plugins/core/workflow/context.js

@@ -1,941 +1,941 @@
-/**
- * FlowContext - Shared context for v2 workflow execution
- * Provides $.push(), $.next(), $.send() API for JS format
- * and manages stage results for JSON format
- *
- * Chain-style API support:
- * - $.tool - Direct tool registry/execution (list, execute, register)
- * - $.extension - Extension tools (list, register, remove, execute)
- * - $.workflow - Nested workflow calls (list, execute)
- * - $.agent - Agent creation/calls (create, chat, list)
- * - $.prompt - Prompt/chat operations
- * - $.skill - Skill commands (list, execute)
- * - $.logger - Logging operations (debug, info, warn, error, trace, fatal)
- * - $.storage - Key-value storage operations (set, get, delete, list, clear)
- */
-
-const { runScriptSafely } = require('../../../src/utils/sandbox');
-
-class FlowContext {
-  constructor(input = {}, sessionId = null) {
-    // Input from workflow caller
-    this.input = input;
-
-    // Flow-level persistent data (like $.data)
-    this.data = {};
-
-    // Stage results storage (stageName -> result)
-    this._results = new Map();
-
-    // Current stage info
-    this.current = null;
-    this.prev = null;
-
-    // Session ID for tool execution
-    this._sessionId = sessionId;
-
-    // Framework reference for tool execution
-    this._framework = null;
-
-    // Destruction flag
-    this._destroyed = false;
-
-    // Pending next stage (for async transitions)
-    this._pendingNext = null;
-
-    // Log prefix
-    this._log = require('../../../src/common/logger').logger.child('FlowContext');
-
-    // Lazy-loaded chain-style API objects
-    this._toolAccessor = null;
-    this._extensionAccessor = null;
-    this._workflowAccessor = null;
-    this._agentAccessor = null;
-    this._promptAccessor = null;
-    this._skillAccessor = null;
-    this._loggerAccessor = null;
-    this._storageAccessor = null;
-  }
-
-  /**
-   * Set framework reference (called by engine)
-   */
-  setFramework(framework) {
-    this._framework = framework;
-  }
-
-  // ============================================================
-  // $.tool - Direct tool registry and execution
-  // ============================================================
-  get tool() {
-    if (!this._toolAccessor) {
-      this._toolAccessor = this._createToolAccessor();
-    }
-    return this._toolAccessor;
-  }
-
-  _createToolAccessor() {
-    const context = this;
-    const log = this._log;
-
-    return {
-      /**
-       * List all registered tools
-       */
-      list: () => {
-        if (!context._framework) return [];
-        return context._framework.toolRegistry ? context._framework.toolRegistry.listTools() : [];
-      },
-
-      /**
-       * Execute a tool directly
-       */
-      execute: async (name, args = {}) => {
-        return await context._executeTool(name, args);
-      },
-
-      /**
-       * Register a new tool (temporary, workflow-scoped)
-       */
-      register: (toolDef) => {
-        log.warn('tool.register() is not implemented in workflow context - tools must be registered via plugin');
-      },
-
-      /**
-       * Get tool definition
-       */
-      get: (name) => {
-        if (!context._framework?.toolRegistry) return null;
-        return context._framework.toolRegistry.getTool(name);
-      },
-    };
-  }
-
-  // ============================================================
-  // $.extension - Extension tools (like Plugin.extension accessor)
-  // ============================================================
-  get extension() {
-    if (!this._extensionAccessor) {
-      this._extensionAccessor = this._createExtensionAccessor();
-    }
-    return this._extensionAccessor;
-  }
-
-  _createExtensionAccessor() {
-    const context = this;
-    const log = this._log;
-
-    return {
-      /**
-       * List all registered extensions
-       */
-      list: () => {
-        if (!context._framework) return [];
-        const extExec = context._framework.pluginManager?.get('extension-executor');
-        if (!extExec?._registry) return [];
-        return extExec._registry.listExtensions();
-      },
-
-      /**
-       * Register an extension tool
-       */
-      register: (toolDef) => {
-        if (!context._framework) {
-          log.warn('extension.register() called before framework is set');
-          return;
-        }
-        if (!toolDef || !toolDef.name) {
-          log.warn('extension.register() missing tool name');
-          return;
-        }
-
-        const extExec = context._framework.pluginManager?.get('extension-executor');
-        if (!extExec) {
-          log.warn('extension-executor plugin not found');
-          return;
-        }
-
-        extExec.registerTool('workflow', {
-          name: 'workflow',
-          description: 'Workflow scoped extensions',
-          version: '1.0.0',
-        }, toolDef);
-
-        extExec._invalidatePromptCaches(context._framework);
-        log.debug(`[Workflow] Registered extension tool: workflow/${toolDef.name}`);
-      },
-
-      /**
-       * Remove an extension tool
-       */
-      remove: (toolName) => {
-        if (!context._framework) {
-          log.warn('extension.remove() called before framework is set');
-          return;
-        }
-
-        const extExec = context._framework.pluginManager?.get('extension-executor');
-        if (extExec) {
-          extExec._unregisterTool('workflow', toolName);
-        }
-        log.debug(`[Workflow] Removed extension tool: workflow/${toolName}`);
-      },
-
-      /**
-       * Execute an extension tool
-       * Supports: extension.execute('tool-name', args) or extension.execute('plugin', 'tool-name', args)
-       */
-      execute: async (pluginNameOrToolName, toolNameOrArgs, args = {}) => {
-        if (!context._framework) {
-          throw new Error('extension.execute() called before framework is set');
-        }
-
-        const extExec = context._framework.pluginManager?.get('extension-executor');
-        if (!extExec) {
-          return { success: false, error: 'extension-executor not found' };
-        }
-
-        let pluginName, toolName;
-        if (typeof toolNameOrArgs === 'object') {
-          // extension.execute('tool-name', args)
-          pluginName = 'workflow';
-          toolName = pluginNameOrToolName;
-          args = toolNameOrArgs;
-        } else {
-          // extension.execute('plugin', 'tool-name', args)
-          pluginName = pluginNameOrToolName;
-          toolName = toolNameOrArgs;
-        }
-
-        const result = await extExec._executeTool(pluginName, toolName, args, context._framework);
-        return result;
-      },
-
-      /**
-       * Get extension info
-       */
-      get: (pluginName) => {
-        if (!context._framework) return null;
-        const extExec = context._framework.pluginManager?.get('extension-executor');
-        if (!extExec?._registry) return null;
-        return extExec._registry.getExtension(pluginName);
-      },
-    };
-  }
-
-  // ============================================================
-  // $.workflow - Nested workflow operations
-  // ============================================================
-  get workflow() {
-    if (!this._workflowAccessor) {
-      this._workflowAccessor = this._createWorkflowAccessor();
-    }
-    return this._workflowAccessor;
-  }
-
-  _createWorkflowAccessor() {
-    const context = this;
-    const log = this._log;
-
-    return {
-      /**
-       * List all available workflows
-       */
-      list: () => {
-        if (!context._framework) return [];
-        const workflowPlugin = context._framework.pluginManager?.get('workflow');
-        if (!workflowPlugin?._workflows) return [];
-        return Array.from(workflowPlugin._workflows.keys());
-      },
-
-      /**
-       * Execute a workflow by name
-       */
-      execute: async (workflowName, input = {}, sessionId = null) => {
-        if (!context._framework) {
-          throw new Error('workflow.execute() called before framework is set');
-        }
-
-        const workflowPlugin = context._framework.pluginManager?.get('workflow');
-        if (!workflowPlugin) {
-          return { success: false, error: 'workflow plugin not found' };
-        }
-
-        const sid = sessionId || context._sessionId;
-        return await workflowPlugin.executeWorkflow(workflowName, input, sid);
-      },
-
-      /**
-       * Get workflow definition
-       */
-      get: (workflowName) => {
-        if (!context._framework) return null;
-        const workflowPlugin = context._framework.pluginManager?.get('workflow');
-        if (!workflowPlugin?._workflows) return null;
-        return workflowPlugin._workflows.get(workflowName);
-      },
-    };
-  }
-
-  // ============================================================
-  // $.agent - Agent operations
-  // ============================================================
-  get agent() {
-    if (!this._agentAccessor) {
-      this._agentAccessor = this._createAgentAccessor();
-    }
-    return this._agentAccessor;
-  }
-
-  _createAgentAccessor() {
-    const context = this;
-    const log = this._log;
-
-    return {
-      /**
-       * Create a new agent
-       */
-      create: (config = {}) => {
-        if (!context._framework) {
-          throw new Error('agent.create() called before framework is set');
-        }
-
-        const resolvedConfig = context.resolveArgs(config);
-        let sessionId = context._sessionId;
-
-        if (sessionId && context._framework.runWithContext) {
-          return context._framework.runWithContext(
-            { sessionId },
-            () => context._framework.createAgent(resolvedConfig)
-          );
-        }
-        return context._framework.createAgent(resolvedConfig);
-      },
-
-      /**
-       * Chat with an agent
-       */
-      chat: async (config = {}) => {
-        if (!context._framework) {
-          throw new Error('agent.chat() called before framework is set');
-        }
-
-        const resolvedConfig = context.resolveArgs(config);
-        const { agentId, message, ...options } = resolvedConfig;
-
-        if (!message) {
-          throw new Error('agent.chat() requires message');
-        }
-
-        let sessionId = context._sessionId;
-
-        if (sessionId && context._framework.runWithContext) {
-          return await context._framework.runWithContext(
-            { sessionId },
-            async () => {
-              const agent = agentId
-                ? context._framework.createAgent({ id: agentId, hidden: true, ...options })
-                : context._framework.createSubAgent({ hidden: true, ...options });
-              return await agent.chat(message, options);
-            }
-          );
-        }
-
-        const agent = agentId
-          ? context._framework.createAgent({ id: agentId, hidden: true, ...options })
-          : context._framework.createSubAgent({ hidden: true, ...options });
-        return await agent.chat(message, options);
-      },
-
-      /**
-       * List active agents
-       */
-      list: () => {
-        if (!context._framework) return [];
-        return context._framework._agents || [];
-      },
-    };
-  }
-
-  // ============================================================
-  // $.prompt - Prompt operations
-  // ============================================================
-  get prompt() {
-    if (!this._promptAccessor) {
-      this._promptAccessor = this._createPromptAccessor();
-    }
-    return this._promptAccessor;
-  }
-
-  _createPromptAccessor() {
-    const context = this;
-    const log = this._log;
-
-    return {
-      /**
-       * Send a prompt and get response (creates temporary agent)
-       */
-      send: async (config = {}) => {
-        if (!context._framework) {
-          throw new Error('prompt.send() called before framework is set');
-        }
-
-        const resolvedConfig = context.resolveArgs(config);
-        const { message, role, systemPrompt, ...options } = resolvedConfig;
-
-        if (!message) {
-          throw new Error('prompt.send() requires message');
-        }
-
-        let sessionId = context._sessionId;
-
-        if (sessionId && context._framework.runWithContext) {
-          return await context._framework.runWithContext(
-            { sessionId },
-            async () => {
-              const agent = context._framework.createSubAgent({
-                role: role || 'Assistant',
-                systemPrompt: systemPrompt || '',
-                hidden: true,
-              });
-              return await agent.chat(message, options);
-            }
-          );
-        }
-
-        const agent = context._framework.createSubAgent({
-          role: role || 'Assistant',
-          systemPrompt: systemPrompt || '',
-          hidden: true,
-        });
-        return await agent.chat(message, options);
-      },
-
-      /**
-       * Chat with a prompt (alias for send)
-       */
-      chat: async (config = {}) => {
-        return this.send(config);
-      },
-    };
-  }
-
-  // ============================================================
-  // $.skill - Skill commands
-  // ============================================================
-  get skill() {
-    if (!this._skillAccessor) {
-      this._skillAccessor = this._createSkillAccessor();
-    }
-    return this._skillAccessor;
-  }
-
-  _createSkillAccessor() {
-    const context = this;
-    const log = this._log;
-
-    return {
-      /**
-       * List available skills
-       */
-      list: () => {
-        if (!context._framework) return [];
-        const extExec = context._framework.pluginManager?.get('extension-executor');
-        if (!extExec?._registry) return [];
-        const extensions = extExec._registry.listExtensions();
-        return extensions
-          .filter(e => e.name.startsWith('skill:'))
-          .map(e => e.name.replace('skill:', ''));
-      },
-
-      /**
-       * Execute a skill command
-       * Supports: skill.execute('skill-name', 'command', args)
-       */
-      execute: async (skillName, commandOrArgs, args = {}) => {
-        if (!context._framework) {
-          throw new Error('skill.execute() called before framework is set');
-        }
-
-        let command, toolArgs;
-        if (typeof commandOrArgs === 'object') {
-          command = commandOrArgs.command || commandOrArgs.tool;
-          toolArgs = commandOrArgs;
-        } else {
-          command = commandOrArgs;
-          toolArgs = args;
-        }
-
-        if (!command) {
-          throw new Error('skill.execute() requires command');
-        }
-
-        const plugin = `skill:${skillName}`;
-        const resolvedArgs = context.resolveArgs(toolArgs);
-
-        if (context._sessionId && context._framework.runWithContext) {
-          return await context._framework.runWithContext(
-            { sessionId: context._sessionId },
-            async () => await context._framework.executeTool('ext_call', {
-              plugin,
-              tool: command,
-              args: resolvedArgs,
-            })
-          );
-        }
-        return await context._framework.executeTool('ext_call', {
-          plugin,
-          tool: command,
-          args: resolvedArgs,
-        });
-      },
-
-      /**
-       * Get skill info
-       */
-      get: (skillName) => {
-        if (!context._framework) return null;
-        const extExec = context._framework.pluginManager?.get('extension-executor');
-        if (!extExec?._registry) return null;
-        return extExec._registry.getExtension(`skill:${skillName}`);
-      },
-    };
-  }
-
-  // ============================================================
-  // $.logger - Logging operations
-  // ============================================================
-  get logger() {
-    if (!this._loggerAccessor) {
-      this._loggerAccessor = this._createLoggerAccessor();
-    }
-    return this._loggerAccessor;
-  }
-
-  _createLoggerAccessor() {
-    const context = this;
-    const flowLog = this._log;
-
-    // Get the underlying logger methods
-    const LEVELS = ['debug', 'info', 'warn', 'error', 'trace', 'fatal'];
-
-    const loggerObj = {
-      /**
-       * Log debug message
-       */
-      debug: (...args) => {
-        flowLog.debug(...args);
-      },
-
-      /**
-       * Log info message
-       */
-      info: (...args) => {
-        flowLog.info(...args);
-      },
-
-      /**
-       * Log warning message
-       */
-      warn: (...args) => {
-        flowLog.warn(...args);
-      },
-
-      /**
-       * Log error message
-       */
-      error: (...args) => {
-        flowLog.error(...args);
-      },
-
-      /**
-       * Log trace message
-       */
-      trace: (...args) => {
-        flowLog.trace(...args);
-      },
-
-      /**
-       * Log fatal message
-       */
-      fatal: (...args) => {
-        flowLog.fatal(...args);
-      },
-
-      /**
-       * Create a child logger with namespace
-       */
-      child: (namespace) => {
-        return flowLog.child(namespace);
-      },
-
-      /**
-       * Log with custom level
-       */
-      log: (level, ...args) => {
-        if (typeof flowLog[level] === 'function') {
-          flowLog[level](...args);
-        } else {
-          flowLog.info(...args);
-        }
-      },
-    };
-
-    return loggerObj;
-  }
-
-  // ============================================================
-  // $.storage - Storage operations
-  // ============================================================
-  get storage() {
-    if (!this._storageAccessor) {
-      this._storageAccessor = this._createStorageAccessor();
-    }
-    return this._storageAccessor;
-  }
-
-  _createStorageAccessor() {
-    const context = this;
-
-    return {
-      /**
-       * Set a value in storage
-       */
-      set: async (key, value, namespace) => {
-        return await context._executeTool('storage_set', { key, value, namespace });
-      },
-
-      /**
-       * Get a value from storage
-       */
-      get: async (key, namespace) => {
-        return await context._executeTool('storage_get', { key, namespace });
-      },
-
-      /**
-       * Delete a value from storage
-       */
-      delete: async (key, namespace) => {
-        return await context._executeTool('storage_delete', { key, namespace });
-      },
-
-      /**
-       * List all keys in a namespace
-       */
-      list: async (namespace) => {
-        return await context._executeTool('storage_list', { namespace });
-      },
-
-      /**
-       * Clear all values in a namespace
-       */
-      clear: async (namespace) => {
-        return await context._executeTool('storage_clear', { namespace });
-      },
-
-      /**
-       * Watch a key for changes
-       */
-      watch: async (key, namespace) => {
-        return await context._executeTool('storage_watch', { key, namespace });
-      },
-
-      /**
-       * Get storage statistics
-       */
-      stats: async (namespace) => {
-        return await context._executeTool('storage_stats', { namespace });
-      },
-
-      /**
-       * Trigger storage compaction
-       */
-      compact: async (namespace) => {
-        return await context._executeTool('storage_compact', { namespace });
-      },
-    };
-  }
-
-  // ============================================================
-  // NOTE: 之前这里有 `async tool(name, args)` legacy method，
-  //       但与上方的 `get tool()` 冲突 (同名)，JavaScript class
-  //       让后定义的方法覆盖 getter，导致 $.tool 返回的永远是
-  //       async function 而非 accessor object。
-  //       已删除此 legacy method；新 API 用 `$.tool.execute(name, args)`
-  //       走 get tool() 返回的 accessor 对象。
-  // ============================================================
-
-  /**
-   * Execute a tool directly
-   */
-  async _executeTool(toolName, args) {
-    if (!this._framework) {
-      throw new Error('Framework not set');
-    }
-
-    // Resolve args with variable interpolation
-    const resolvedArgs = this.resolveArgs(args || {});
-
-    let sessionId = this._sessionId;
-    if (!sessionId && this._framework.getExecutionContext) {
-      const ctx = this._framework.getExecutionContext();
-      sessionId = ctx?.sessionId;
-    }
-
-    if (sessionId && this._framework.runWithContext) {
-      return await this._framework.runWithContext(
-        { sessionId },
-        async () => await this._framework.executeTool(toolName, resolvedArgs)
-      );
-    }
-    return await this._framework.executeTool(toolName, resolvedArgs);
-  }
-
-  // ============================================================
-  // Basic Flow API
-  // ============================================================
-
-  /**
-   * Register a stage (JS format - $.push)
-   */
-  push(name, fn) {
-    if (!this._stages) {
-      this._stages = new Map();
-    }
-    this._stages.set(name, fn);
-    return this;
-  }
-
-  /**
-   * Go to next stage (sync) - $.next
-   * 支持直接传参: $.next('stage-name', arg1, arg2, ...)
-   * 等同�?$.send({ arg1, arg2, ... }) 后再 $.next('stage-name')
-   */
-  next(name, ...args) {
-    this._pendingNext = name;
-    if (args.length > 0) {
-      this._sentData = args.length === 1 ? args[0] : args;
-    }
-    return this;
-  }
-
-  /**
-   * Go to next stage (async) - $.next2
-   * Returns a function for deferred transition
-   */
-  next2(name) {
-    return () => {
-      this.next(name);
-    };
-  }
-
-  /**
-   * Pass data to next stage - $.send
-   */
-  send(data) {
-    if (data !== undefined) {
-      this._sentData = data;
-    }
-    return this;
-  }
-
-  /**
-   * Built-in delay helper - $.wait
-   */
-  wait(ms) {
-    return new Promise((resolve) => setTimeout(resolve, ms));
-  }
-
-  /**
-   * Execute a script - $.script
-   */
-  async script(code) {
-    const scriptContext = {
-      input: this.input,
-      data: this.data,
-      variables: Object.fromEntries(this._results),
-      previousResult: this._sentData,
-      console: {
-        log: (...args) => {
-          this._log.debug('[Script]', ...args);
-        },
-      },
-    };
-    return await runScriptSafely(code, scriptContext, { timeout: 10000 });
-  }
-
-  /**
-   * End workflow - $.destroy
-   */
-  destroy() {
-    this._destroyed = true;
-    this._pendingNext = null;
-  }
-
-  /**
-   * Get result from a named stage
-   */
-  getResult(stageName) {
-    return this._results.get(stageName);
-  }
-
-  /**
-   * Set result for a stage
-   */
-  setResult(stageName, result) {
-    this._results.set(stageName, result);
-  }
-
-  // ============================================================
-  // Variable Resolution
-  // ============================================================
-
-  /**
-   * Resolve a value with variable interpolation
-   * Supports {{stageName}}, {{stage.field}}, {{input.field}}, {{data.field}}
-   *
-   * @param {*} value - Value to resolve (string templates only)
-   * @param {Object} [options]
-   * @param {boolean} [options.raw=false] - If true, return raw resolved value (preserve type).
-   *                       Used by stages like `each` that need actual arrays/objects
-   *                       instead of stringified versions.
-   * @returns {*} Resolved value
-   */
-  resolveValue(value, options = {}) {
-    if (typeof value !== 'string') {
-      return value;
-    }
-
-    const { raw = false } = options;
-
-    // Replace {{stageName}} or {{stage.field}} with stage result
-    return value.replace(/\{\{([^}]+)\}\}/g, (match, path) => {
-      const resolved = this._resolvePath(path);
-
-      // Path not found - keep the original placeholder
-      if (resolved === undefined) {
-        return match;
-      }
-
-      // Raw mode: return the value as-is (caller handles serialization)
-      if (raw) {
-        return resolved;
-      }
-
-      // String mode: serialize for interpolation
-      if (resolved === null) {
-        return 'null';
-      }
-      // For arrays and objects, JSON-encode so the template works as a string
-      // (Raw arrays/objects can't be inlined into strings via String.replace without coercion)
-      if (typeof resolved === 'object') {
-        try {
-          return JSON.stringify(resolved);
-        } catch {
-          return String(resolved);
-        }
-      }
-
-      return resolved;
-    });
-  }
-
-  /**
-   * Resolve a template path to its value.
-   * Internal helper used by resolveValue and other template consumers.
-   * Supports: input.field, data.field, stageName, stage.field
-   * Also searches context.data when a simple path is not found in _results.
-   *
-   * @param {string} path - Template path (without {{ }} markers)
-   * @returns {*} Resolved value or undefined if not found
-   */
-  _resolvePath(path) {
-    // input.field access
-    if (path.startsWith('input.')) {
-      const val = this._getNestedValue(this.input, path.substring(6));
-      return val !== undefined ? val : undefined;
-    }
-
-    // data.field access
-    if (path.startsWith('data.')) {
-      const val = this._getNestedValue(this.data, path.substring(5));
-      return val !== undefined ? val : undefined;
-    }
-
-    // stage.field or stageName access
-    const dotIndex = path.indexOf('.');
-    if (dotIndex > 0) {
-      const stageName = path.substring(0, dotIndex);
-      const fieldPath = path.substring(dotIndex + 1);
-      // Look up the root value (try stage results first, then loop/data context)
-      // so that nested property access works for each-loop variables too,
-      // e.g. inside an each body: {{u.name}} where u is the current item.
-      let stageResult = this._results.get(stageName);
-      if (stageResult === undefined) {
-        stageResult = this.data[stageName];
-      }
-      if (stageResult !== undefined) {
-        const val = this._getNestedValue(stageResult, fieldPath);
-        if (val !== undefined) return val;
-      }
-      return undefined;
-    }
-
-    // Simple stageName reference: try _results first, then data
-    // (data lookup supports each-loop variables, e.g., {{fruit}} inside an each body)
-    let val = this._results.get(path);
-    if (val === undefined) {
-      val = this.data[path];
-    }
-    return val !== undefined ? val : undefined;
-  }
-
-  /**
-   * Resolve args object with variable interpolation
-   */
-  resolveArgs(args) {
-    if (!args || typeof args !== 'object') {
-      return args;
-    }
-    const resolved = {};
-    for (const [key, value] of Object.entries(args)) {
-      if (typeof value === 'string') {
-        resolved[key] = this.resolveValue(value);
-      } else if (Array.isArray(value)) {
-        // Resolve each element as a template (raw mode preserves types)
-        resolved[key] = value.map((v) =>
-          typeof v === 'string' ? this.resolveValue(v, { raw: true }) : v
-        );
-      } else if (typeof value === 'object' && value !== null) {
-        resolved[key] = this.resolveArgs(value);
-      } else {
-        resolved[key] = value;
-      }
-    }
-    return resolved;
-  }
-
-  /**
-   * Get nested value from object by dot notation path
-   */
-  _getNestedValue(obj, path) {
-    if (!obj) return undefined;
-    const parts = path.split('.');
-    let current = obj;
-    for (const part of parts) {
-      if (current === null || current === undefined) return undefined;
-      // Handle array index in path like "items.0.name"
-      if (typeof current === 'string') {
-        try {
-          current = JSON.parse(current);
-        } catch {
-          // Not JSON, continue
-        }
-      }
-      current = current[part];
-    }
-    return current;
-  }
-}
-
-module.exports = { FlowContext };
+/**
+ * FlowContext - Shared context for v2 workflow execution
+ * Provides $.push(), $.next(), $.send() API for JS format
+ * and manages stage results for JSON format
+ *
+ * Chain-style API support:
+ * - $.tool - Direct tool registry/execution (list, execute, register)
+ * - $.extension - Extension tools (list, register, remove, execute)
+ * - $.workflow - Nested workflow calls (list, execute)
+ * - $.agent - Agent creation/calls (create, chat, list)
+ * - $.prompt - Prompt/chat operations
+ * - $.skill - Skill commands (list, execute)
+ * - $.logger - Logging operations (debug, info, warn, error, trace, fatal)
+ * - $.storage - Key-value storage operations (set, get, delete, list, clear)
+ */
+
+const { runScriptSafely } = require('../../../src/utils/sandbox');
+
+class FlowContext {
+  constructor(input = {}, sessionId = null) {
+    // Input from workflow caller
+    this.input = input;
+
+    // Flow-level persistent data (like $.data)
+    this.data = {};
+
+    // Stage results storage (stageName -> result)
+    this._results = new Map();
+
+    // Current stage info
+    this.current = null;
+    this.prev = null;
+
+    // Session ID for tool execution
+    this._sessionId = sessionId;
+
+    // Framework reference for tool execution
+    this._framework = null;
+
+    // Destruction flag
+    this._destroyed = false;
+
+    // Pending next stage (for async transitions)
+    this._pendingNext = null;
+
+    // Log prefix
+    this._log = require('../../../src/common/logger').logger.child('FlowContext');
+
+    // Lazy-loaded chain-style API objects
+    this._toolAccessor = null;
+    this._extensionAccessor = null;
+    this._workflowAccessor = null;
+    this._agentAccessor = null;
+    this._promptAccessor = null;
+    this._skillAccessor = null;
+    this._loggerAccessor = null;
+    this._storageAccessor = null;
+  }
+
+  /**
+   * Set framework reference (called by engine)
+   */
+  setFramework(framework) {
+    this._framework = framework;
+  }
+
+  // ============================================================
+  // $.tool - Direct tool registry and execution
+  // ============================================================
+  get tool() {
+    if (!this._toolAccessor) {
+      this._toolAccessor = this._createToolAccessor();
+    }
+    return this._toolAccessor;
+  }
+
+  _createToolAccessor() {
+    const context = this;
+    const log = this._log;
+
+    return {
+      /**
+       * List all registered tools
+       */
+      list: () => {
+        if (!context._framework) return [];
+        return context._framework.toolRegistry ? context._framework.toolRegistry.listTools() : [];
+      },
+
+      /**
+       * Execute a tool directly
+       */
+      execute: async (name, args = {}) => {
+        return await context._executeTool(name, args);
+      },
+
+      /**
+       * Register a new tool (temporary, workflow-scoped)
+       */
+      register: (toolDef) => {
+        log.warn('tool.register() is not implemented in workflow context - tools must be registered via plugin');
+      },
+
+      /**
+       * Get tool definition
+       */
+      get: (name) => {
+        if (!context._framework?.toolRegistry) return null;
+        return context._framework.toolRegistry.getTool(name);
+      },
+    };
+  }
+
+  // ============================================================
+  // $.extension - Extension tools (like Plugin.extension accessor)
+  // ============================================================
+  get extension() {
+    if (!this._extensionAccessor) {
+      this._extensionAccessor = this._createExtensionAccessor();
+    }
+    return this._extensionAccessor;
+  }
+
+  _createExtensionAccessor() {
+    const context = this;
+    const log = this._log;
+
+    return {
+      /**
+       * List all registered extensions
+       */
+      list: () => {
+        if (!context._framework) return [];
+        const extExec = context._framework.pluginManager?.get('extension-executor');
+        if (!extExec?._registry) return [];
+        return extExec._registry.listExtensions();
+      },
+
+      /**
+       * Register an extension tool
+       */
+      register: (toolDef) => {
+        if (!context._framework) {
+          log.warn('extension.register() called before framework is set');
+          return;
+        }
+        if (!toolDef || !toolDef.name) {
+          log.warn('extension.register() missing tool name');
+          return;
+        }
+
+        const extExec = context._framework.pluginManager?.get('extension-executor');
+        if (!extExec) {
+          log.warn('extension-executor plugin not found');
+          return;
+        }
+
+        extExec.registerTool('workflow', {
+          name: 'workflow',
+          description: 'Workflow scoped extensions',
+          version: '1.0.0',
+        }, toolDef);
+
+        extExec._invalidatePromptCaches(context._framework);
+        log.debug(`[Workflow] Registered extension tool: workflow/${toolDef.name}`);
+      },
+
+      /**
+       * Remove an extension tool
+       */
+      remove: (toolName) => {
+        if (!context._framework) {
+          log.warn('extension.remove() called before framework is set');
+          return;
+        }
+
+        const extExec = context._framework.pluginManager?.get('extension-executor');
+        if (extExec) {
+          extExec._unregisterTool('workflow', toolName);
+        }
+        log.debug(`[Workflow] Removed extension tool: workflow/${toolName}`);
+      },
+
+      /**
+       * Execute an extension tool
+       * Supports: extension.execute('tool-name', args) or extension.execute('plugin', 'tool-name', args)
+       */
+      execute: async (pluginNameOrToolName, toolNameOrArgs, args = {}) => {
+        if (!context._framework) {
+          throw new Error('extension.execute() called before framework is set');
+        }
+
+        const extExec = context._framework.pluginManager?.get('extension-executor');
+        if (!extExec) {
+          return { success: false, error: 'extension-executor not found' };
+        }
+
+        let pluginName, toolName;
+        if (typeof toolNameOrArgs === 'object') {
+          // extension.execute('tool-name', args)
+          pluginName = 'workflow';
+          toolName = pluginNameOrToolName;
+          args = toolNameOrArgs;
+        } else {
+          // extension.execute('plugin', 'tool-name', args)
+          pluginName = pluginNameOrToolName;
+          toolName = toolNameOrArgs;
+        }
+
+        const result = await extExec._executeTool(pluginName, toolName, args, context._framework);
+        return result;
+      },
+
+      /**
+       * Get extension info
+       */
+      get: (pluginName) => {
+        if (!context._framework) return null;
+        const extExec = context._framework.pluginManager?.get('extension-executor');
+        if (!extExec?._registry) return null;
+        return extExec._registry.getExtension(pluginName);
+      },
+    };
+  }
+
+  // ============================================================
+  // $.workflow - Nested workflow operations
+  // ============================================================
+  get workflow() {
+    if (!this._workflowAccessor) {
+      this._workflowAccessor = this._createWorkflowAccessor();
+    }
+    return this._workflowAccessor;
+  }
+
+  _createWorkflowAccessor() {
+    const context = this;
+    const log = this._log;
+
+    return {
+      /**
+       * List all available workflows
+       */
+      list: () => {
+        if (!context._framework) return [];
+        const workflowPlugin = context._framework.pluginManager?.get('workflow');
+        if (!workflowPlugin?._workflows) return [];
+        return Array.from(workflowPlugin._workflows.keys());
+      },
+
+      /**
+       * Execute a workflow by name
+       */
+      execute: async (workflowName, input = {}, sessionId = null) => {
+        if (!context._framework) {
+          throw new Error('workflow.execute() called before framework is set');
+        }
+
+        const workflowPlugin = context._framework.pluginManager?.get('workflow');
+        if (!workflowPlugin) {
+          return { success: false, error: 'workflow plugin not found' };
+        }
+
+        const sid = sessionId || context._sessionId;
+        return await workflowPlugin.executeWorkflow(workflowName, input, sid);
+      },
+
+      /**
+       * Get workflow definition
+       */
+      get: (workflowName) => {
+        if (!context._framework) return null;
+        const workflowPlugin = context._framework.pluginManager?.get('workflow');
+        if (!workflowPlugin?._workflows) return null;
+        return workflowPlugin._workflows.get(workflowName);
+      },
+    };
+  }
+
+  // ============================================================
+  // $.agent - Agent operations
+  // ============================================================
+  get agent() {
+    if (!this._agentAccessor) {
+      this._agentAccessor = this._createAgentAccessor();
+    }
+    return this._agentAccessor;
+  }
+
+  _createAgentAccessor() {
+    const context = this;
+    const log = this._log;
+
+    return {
+      /**
+       * Create a new agent
+       */
+      create: (config = {}) => {
+        if (!context._framework) {
+          throw new Error('agent.create() called before framework is set');
+        }
+
+        const resolvedConfig = context.resolveArgs(config);
+        let sessionId = context._sessionId;
+
+        if (sessionId && context._framework.runWithContext) {
+          return context._framework.runWithContext(
+            { sessionId },
+            () => context._framework.createAgent(resolvedConfig)
+          );
+        }
+        return context._framework.createAgent(resolvedConfig);
+      },
+
+      /**
+       * Chat with an agent
+       */
+      chat: async (config = {}) => {
+        if (!context._framework) {
+          throw new Error('agent.chat() called before framework is set');
+        }
+
+        const resolvedConfig = context.resolveArgs(config);
+        const { agentId, message, ...options } = resolvedConfig;
+
+        if (!message) {
+          throw new Error('agent.chat() requires message');
+        }
+
+        let sessionId = context._sessionId;
+
+        if (sessionId && context._framework.runWithContext) {
+          return await context._framework.runWithContext(
+            { sessionId },
+            async () => {
+              const agent = agentId
+                ? context._framework.createAgent({ id: agentId, hidden: true, ...options })
+                : context._framework.createSubAgent({ hidden: true, ...options });
+              return await agent.chat(message, options);
+            }
+          );
+        }
+
+        const agent = agentId
+          ? context._framework.createAgent({ id: agentId, hidden: true, ...options })
+          : context._framework.createSubAgent({ hidden: true, ...options });
+        return await agent.chat(message, options);
+      },
+
+      /**
+       * List active agents
+       */
+      list: () => {
+        if (!context._framework) return [];
+        return context._framework._agents || [];
+      },
+    };
+  }
+
+  // ============================================================
+  // $.prompt - Prompt operations
+  // ============================================================
+  get prompt() {
+    if (!this._promptAccessor) {
+      this._promptAccessor = this._createPromptAccessor();
+    }
+    return this._promptAccessor;
+  }
+
+  _createPromptAccessor() {
+    const context = this;
+    const log = this._log;
+
+    return {
+      /**
+       * Send a prompt and get response (creates temporary agent)
+       */
+      send: async (config = {}) => {
+        if (!context._framework) {
+          throw new Error('prompt.send() called before framework is set');
+        }
+
+        const resolvedConfig = context.resolveArgs(config);
+        const { message, role, systemPrompt, ...options } = resolvedConfig;
+
+        if (!message) {
+          throw new Error('prompt.send() requires message');
+        }
+
+        let sessionId = context._sessionId;
+
+        if (sessionId && context._framework.runWithContext) {
+          return await context._framework.runWithContext(
+            { sessionId },
+            async () => {
+              const agent = context._framework.createSubAgent({
+                role: role || 'Assistant',
+                systemPrompt: systemPrompt || '',
+                hidden: true,
+              });
+              return await agent.chat(message, options);
+            }
+          );
+        }
+
+        const agent = context._framework.createSubAgent({
+          role: role || 'Assistant',
+          systemPrompt: systemPrompt || '',
+          hidden: true,
+        });
+        return await agent.chat(message, options);
+      },
+
+      /**
+       * Chat with a prompt (alias for send)
+       */
+      chat: async (config = {}) => {
+        return this.send(config);
+      },
+    };
+  }
+
+  // ============================================================
+  // $.skill - Skill commands
+  // ============================================================
+  get skill() {
+    if (!this._skillAccessor) {
+      this._skillAccessor = this._createSkillAccessor();
+    }
+    return this._skillAccessor;
+  }
+
+  _createSkillAccessor() {
+    const context = this;
+    const log = this._log;
+
+    return {
+      /**
+       * List available skills
+       */
+      list: () => {
+        if (!context._framework) return [];
+        const extExec = context._framework.pluginManager?.get('extension-executor');
+        if (!extExec?._registry) return [];
+        const extensions = extExec._registry.listExtensions();
+        return extensions
+          .filter(e => e.name.startsWith('skill:'))
+          .map(e => e.name.replace('skill:', ''));
+      },
+
+      /**
+       * Execute a skill command
+       * Supports: skill.execute('skill-name', 'command', args)
+       */
+      execute: async (skillName, commandOrArgs, args = {}) => {
+        if (!context._framework) {
+          throw new Error('skill.execute() called before framework is set');
+        }
+
+        let command, toolArgs;
+        if (typeof commandOrArgs === 'object') {
+          command = commandOrArgs.command || commandOrArgs.tool;
+          toolArgs = commandOrArgs;
+        } else {
+          command = commandOrArgs;
+          toolArgs = args;
+        }
+
+        if (!command) {
+          throw new Error('skill.execute() requires command');
+        }
+
+        const plugin = `skill:${skillName}`;
+        const resolvedArgs = context.resolveArgs(toolArgs);
+
+        if (context._sessionId && context._framework.runWithContext) {
+          return await context._framework.runWithContext(
+            { sessionId: context._sessionId },
+            async () => await context._framework.executeTool('ext_call', {
+              plugin,
+              tool: command,
+              args: resolvedArgs,
+            })
+          );
+        }
+        return await context._framework.executeTool('ext_call', {
+          plugin,
+          tool: command,
+          args: resolvedArgs,
+        });
+      },
+
+      /**
+       * Get skill info
+       */
+      get: (skillName) => {
+        if (!context._framework) return null;
+        const extExec = context._framework.pluginManager?.get('extension-executor');
+        if (!extExec?._registry) return null;
+        return extExec._registry.getExtension(`skill:${skillName}`);
+      },
+    };
+  }
+
+  // ============================================================
+  // $.logger - Logging operations
+  // ============================================================
+  get logger() {
+    if (!this._loggerAccessor) {
+      this._loggerAccessor = this._createLoggerAccessor();
+    }
+    return this._loggerAccessor;
+  }
+
+  _createLoggerAccessor() {
+    const context = this;
+    const flowLog = this._log;
+
+    // Get the underlying logger methods
+    const LEVELS = ['debug', 'info', 'warn', 'error', 'trace', 'fatal'];
+
+    const loggerObj = {
+      /**
+       * Log debug message
+       */
+      debug: (...args) => {
+        flowLog.debug(...args);
+      },
+
+      /**
+       * Log info message
+       */
+      info: (...args) => {
+        flowLog.info(...args);
+      },
+
+      /**
+       * Log warning message
+       */
+      warn: (...args) => {
+        flowLog.warn(...args);
+      },
+
+      /**
+       * Log error message
+       */
+      error: (...args) => {
+        flowLog.error(...args);
+      },
+
+      /**
+       * Log trace message
+       */
+      trace: (...args) => {
+        flowLog.trace(...args);
+      },
+
+      /**
+       * Log fatal message
+       */
+      fatal: (...args) => {
+        flowLog.fatal(...args);
+      },
+
+      /**
+       * Create a child logger with namespace
+       */
+      child: (namespace) => {
+        return flowLog.child(namespace);
+      },
+
+      /**
+       * Log with custom level
+       */
+      log: (level, ...args) => {
+        if (typeof flowLog[level] === 'function') {
+          flowLog[level](...args);
+        } else {
+          flowLog.info(...args);
+        }
+      },
+    };
+
+    return loggerObj;
+  }
+
+  // ============================================================
+  // $.storage - Storage operations
+  // ============================================================
+  get storage() {
+    if (!this._storageAccessor) {
+      this._storageAccessor = this._createStorageAccessor();
+    }
+    return this._storageAccessor;
+  }
+
+  _createStorageAccessor() {
+    const context = this;
+
+    return {
+      /**
+       * Set a value in storage
+       */
+      set: async (key, value, namespace) => {
+        return await context._executeTool('storage_set', { key, value, namespace });
+      },
+
+      /**
+       * Get a value from storage
+       */
+      get: async (key, namespace) => {
+        return await context._executeTool('storage_get', { key, namespace });
+      },
+
+      /**
+       * Delete a value from storage
+       */
+      delete: async (key, namespace) => {
+        return await context._executeTool('storage_delete', { key, namespace });
+      },
+
+      /**
+       * List all keys in a namespace
+       */
+      list: async (namespace) => {
+        return await context._executeTool('storage_list', { namespace });
+      },
+
+      /**
+       * Clear all values in a namespace
+       */
+      clear: async (namespace) => {
+        return await context._executeTool('storage_clear', { namespace });
+      },
+
+      /**
+       * Watch a key for changes
+       */
+      watch: async (key, namespace) => {
+        return await context._executeTool('storage_watch', { key, namespace });
+      },
+
+      /**
+       * Get storage statistics
+       */
+      stats: async (namespace) => {
+        return await context._executeTool('storage_stats', { namespace });
+      },
+
+      /**
+       * Trigger storage compaction
+       */
+      compact: async (namespace) => {
+        return await context._executeTool('storage_compact', { namespace });
+      },
+    };
+  }
+
+  // ============================================================
+  // NOTE: 之前这里有 `async tool(name, args)` legacy method，
+  //       但与上方的 `get tool()` 冲突 (同名)，JavaScript class
+  //       让后定义的方法覆盖 getter，导致 $.tool 返回的永远是
+  //       async function 而非 accessor object。
+  //       已删除此 legacy method；新 API 用 `$.tool.execute(name, args)`
+  //       走 get tool() 返回的 accessor 对象。
+  // ============================================================
+
+  /**
+   * Execute a tool directly
+   */
+  async _executeTool(toolName, args) {
+    if (!this._framework) {
+      throw new Error('Framework not set');
+    }
+
+    // Resolve args with variable interpolation
+    const resolvedArgs = this.resolveArgs(args || {});
+
+    let sessionId = this._sessionId;
+    if (!sessionId && this._framework.getExecutionContext) {
+      const ctx = this._framework.getExecutionContext();
+      sessionId = ctx?.sessionId;
+    }
+
+    if (sessionId && this._framework.runWithContext) {
+      return await this._framework.runWithContext(
+        { sessionId },
+        async () => await this._framework.executeTool(toolName, resolvedArgs)
+      );
+    }
+    return await this._framework.executeTool(toolName, resolvedArgs);
+  }
+
+  // ============================================================
+  // Basic Flow API
+  // ============================================================
+
+  /**
+   * Register a stage (JS format - $.push)
+   */
+  push(name, fn) {
+    if (!this._stages) {
+      this._stages = new Map();
+    }
+    this._stages.set(name, fn);
+    return this;
+  }
+
+  /**
+   * Go to next stage (sync) - $.next
+   * 支持直接传参: $.next('stage-name', arg1, arg2, ...)
+   * 等同�?$.send({ arg1, arg2, ... }) 后再 $.next('stage-name')
+   */
+  next(name, ...args) {
+    this._pendingNext = name;
+    if (args.length > 0) {
+      this._sentData = args.length === 1 ? args[0] : args;
+    }
+    return this;
+  }
+
+  /**
+   * Go to next stage (async) - $.next2
+   * Returns a function for deferred transition
+   */
+  next2(name) {
+    return () => {
+      this.next(name);
+    };
+  }
+
+  /**
+   * Pass data to next stage - $.send
+   */
+  send(data) {
+    if (data !== undefined) {
+      this._sentData = data;
+    }
+    return this;
+  }
+
+  /**
+   * Built-in delay helper - $.wait
+   */
+  wait(ms) {
+    return new Promise((resolve) => setTimeout(resolve, ms));
+  }
+
+  /**
+   * Execute a script - $.script
+   */
+  async script(code) {
+    const scriptContext = {
+      input: this.input,
+      data: this.data,
+      variables: Object.fromEntries(this._results),
+      previousResult: this._sentData,
+      console: {
+        log: (...args) => {
+          this._log.debug('[Script]', ...args);
+        },
+      },
+    };
+    return await runScriptSafely(code, scriptContext, { timeout: 10000 });
+  }
+
+  /**
+   * End workflow - $.destroy
+   */
+  destroy() {
+    this._destroyed = true;
+    this._pendingNext = null;
+  }
+
+  /**
+   * Get result from a named stage
+   */
+  getResult(stageName) {
+    return this._results.get(stageName);
+  }
+
+  /**
+   * Set result for a stage
+   */
+  setResult(stageName, result) {
+    this._results.set(stageName, result);
+  }
+
+  // ============================================================
+  // Variable Resolution
+  // ============================================================
+
+  /**
+   * Resolve a value with variable interpolation
+   * Supports {{stageName}}, {{stage.field}}, {{input.field}}, {{data.field}}
+   *
+   * @param {*} value - Value to resolve (string templates only)
+   * @param {Object} [options]
+   * @param {boolean} [options.raw=false] - If true, return raw resolved value (preserve type).
+   *                       Used by stages like `each` that need actual arrays/objects
+   *                       instead of stringified versions.
+   * @returns {*} Resolved value
+   */
+  resolveValue(value, options = {}) {
+    if (typeof value !== 'string') {
+      return value;
+    }
+
+    const { raw = false } = options;
+
+    // Replace {{stageName}} or {{stage.field}} with stage result
+    return value.replace(/\{\{([^}]+)\}\}/g, (match, path) => {
+      const resolved = this._resolvePath(path);
+
+      // Path not found - keep the original placeholder
+      if (resolved === undefined) {
+        return match;
+      }
+
+      // Raw mode: return the value as-is (caller handles serialization)
+      if (raw) {
+        return resolved;
+      }
+
+      // String mode: serialize for interpolation
+      if (resolved === null) {
+        return 'null';
+      }
+      // For arrays and objects, JSON-encode so the template works as a string
+      // (Raw arrays/objects can't be inlined into strings via String.replace without coercion)
+      if (typeof resolved === 'object') {
+        try {
+          return JSON.stringify(resolved);
+        } catch {
+          return String(resolved);
+        }
+      }
+
+      return resolved;
+    });
+  }
+
+  /**
+   * Resolve a template path to its value.
+   * Internal helper used by resolveValue and other template consumers.
+   * Supports: input.field, data.field, stageName, stage.field
+   * Also searches context.data when a simple path is not found in _results.
+   *
+   * @param {string} path - Template path (without {{ }} markers)
+   * @returns {*} Resolved value or undefined if not found
+   */
+  _resolvePath(path) {
+    // input.field access
+    if (path.startsWith('input.')) {
+      const val = this._getNestedValue(this.input, path.substring(6));
+      return val !== undefined ? val : undefined;
+    }
+
+    // data.field access
+    if (path.startsWith('data.')) {
+      const val = this._getNestedValue(this.data, path.substring(5));
+      return val !== undefined ? val : undefined;
+    }
+
+    // stage.field or stageName access
+    const dotIndex = path.indexOf('.');
+    if (dotIndex > 0) {
+      const stageName = path.substring(0, dotIndex);
+      const fieldPath = path.substring(dotIndex + 1);
+      // Look up the root value (try stage results first, then loop/data context)
+      // so that nested property access works for each-loop variables too,
+      // e.g. inside an each body: {{u.name}} where u is the current item.
+      let stageResult = this._results.get(stageName);
+      if (stageResult === undefined) {
+        stageResult = this.data[stageName];
+      }
+      if (stageResult !== undefined) {
+        const val = this._getNestedValue(stageResult, fieldPath);
+        if (val !== undefined) return val;
+      }
+      return undefined;
+    }
+
+    // Simple stageName reference: try _results first, then data
+    // (data lookup supports each-loop variables, e.g., {{fruit}} inside an each body)
+    let val = this._results.get(path);
+    if (val === undefined) {
+      val = this.data[path];
+    }
+    return val !== undefined ? val : undefined;
+  }
+
+  /**
+   * Resolve args object with variable interpolation
+   */
+  resolveArgs(args) {
+    if (!args || typeof args !== 'object') {
+      return args;
+    }
+    const resolved = {};
+    for (const [key, value] of Object.entries(args)) {
+      if (typeof value === 'string') {
+        resolved[key] = this.resolveValue(value);
+      } else if (Array.isArray(value)) {
+        // Resolve each element as a template (raw mode preserves types)
+        resolved[key] = value.map((v) =>
+          typeof v === 'string' ? this.resolveValue(v, { raw: true }) : v
+        );
+      } else if (typeof value === 'object' && value !== null) {
+        resolved[key] = this.resolveArgs(value);
+      } else {
+        resolved[key] = value;
+      }
+    }
+    return resolved;
+  }
+
+  /**
+   * Get nested value from object by dot notation path
+   */
+  _getNestedValue(obj, path) {
+    if (!obj) return undefined;
+    const parts = path.split('.');
+    let current = obj;
+    for (const part of parts) {
+      if (current === null || current === undefined) return undefined;
+      // Handle array index in path like "items.0.name"
+      if (typeof current === 'string') {
+        try {
+          current = JSON.parse(current);
+        } catch {
+          // Not JSON, continue
+        }
+      }
+      current = current[part];
+    }
+    return current;
+  }
+}
+
+module.exports = { FlowContext };