groundswell 0.0.2 → 0.0.3

package/dist/core/agent.js

@@ -0,0 +1,426 @@
+/**
+ * Agent - Lightweight wrapper around Anthropic's Agent SDK
+ *
+ * Agents execute prompts and manage tool invocation cycles.
+ * All configuration properties map 1:1 to Anthropic SDK.
+ */
+import Anthropic from '@anthropic-ai/sdk';
+import { MCPHandler } from './mcp-handler.js';
+import { generateId } from '../utils/id.js';
+import { getExecutionContext } from './context.js';
+import { generateCacheKey, defaultCache } from '../cache/index.js';
+/**
+ * Agent class - executes prompts via Anthropic SDK
+ */
+export class Agent {
+    /** Unique identifier for this agent instance */
+    id;
+    /** Agent name */
+    name;
+    /** Stored configuration */
+    config;
+    /** Anthropic client instance */
+    client;
+    /** MCP handler for tool management */
+    mcpHandler;
+    /** Direct MCPHandler instances for delegated execution */
+    mcpHandlers = [];
+    /** Default model to use */
+    model;
+    /**
+     * Create a new Agent instance
+     * @param config Agent configuration
+     */
+    constructor(config = {}) {
+        this.id = generateId();
+        this.name = config.name ?? 'Agent';
+        this.config = config;
+        this.model = config.model ?? 'claude-sonnet-4-20250514';
+        // Create Anthropic client
+        this.client = new Anthropic({
+            apiKey: process.env.ANTHROPIC_API_KEY,
+        });
+        // Initialize MCP handler
+        this.mcpHandler = new MCPHandler();
+        // Register MCP servers
+        if (config.mcps) {
+            for (const mcp of config.mcps) {
+                // If the MCP is already an MCPHandler instance, store it directly
+                // for delegated tool execution (preserves registered executors)
+                if (mcp instanceof MCPHandler) {
+                    this.mcpHandlers.push(mcp);
+                }
+                // Always register with main handler for tool discovery
+                this.mcpHandler.registerServer(mcp);
+            }
+        }
+    }
+    /**
+     * Execute a prompt and return validated response
+     * @param prompt Prompt to execute
+     * @param overrides Optional overrides for this execution
+     * @returns Validated response of type T
+     */
+    async prompt(prompt, overrides) {
+        const result = await this.executePrompt(prompt, overrides);
+        return result.data;
+    }
+    /**
+     * Execute a prompt with full result metadata
+     * @param prompt Prompt to execute
+     * @param overrides Optional overrides for this execution
+     * @returns Full result including metadata
+     */
+    async promptWithMetadata(prompt, overrides) {
+        return this.executePrompt(prompt, overrides);
+    }
+    /**
+     * Execute a prompt with reflection capabilities
+     * @param prompt Prompt to execute
+     * @param overrides Optional overrides for this execution
+     * @returns Validated response of type T
+     */
+    async reflect(prompt, overrides) {
+        // Add reflection system prefix if reflection is enabled
+        const reflectionEnabled = prompt.enableReflection ??
+            overrides?.enableReflection ??
+            this.config.enableReflection;
+        const systemPrefix = reflectionEnabled
+            ? 'Before answering, reflect on your reasoning step by step. Consider alternative approaches and potential errors. Then provide your final answer.\n\n'
+            : '';
+        const effectiveOverrides = {
+            ...overrides,
+            system: systemPrefix +
+                (prompt.systemOverride ?? overrides?.system ?? this.config.system ?? ''),
+        };
+        const result = await this.executePrompt(prompt, effectiveOverrides);
+        return result.data;
+    }
+    /**
+     * Get the MCP handler for custom tool registration
+     */
+    getMcpHandler() {
+        return this.mcpHandler;
+    }
+    /**
+     * Emit an event if within workflow context
+     */
+    emitWorkflowEvent(event) {
+        const ctx = getExecutionContext();
+        if (ctx) {
+            ctx.emitEvent(event);
+        }
+    }
+    /**
+     * Internal prompt execution with full flow
+     */
+    async executePrompt(prompt, overrides) {
+        const startTime = Date.now();
+        let toolCallCount = 0;
+        let totalUsage = { input_tokens: 0, output_tokens: 0 };
+        // Get execution context for event emission
+        const ctx = getExecutionContext();
+        // Merge configuration: Prompt > Overrides > Config
+        const effectiveSystem = prompt.systemOverride ?? overrides?.system ?? this.config.system;
+        const effectiveModel = overrides?.model ?? this.model;
+        const effectiveMaxTokens = overrides?.maxTokens ?? this.config.maxTokens ?? 4096;
+        const effectiveTemperature = overrides?.temperature ?? this.config.temperature;
+        // Check cache if enabled
+        const cacheEnabled = this.config.enableCache && !overrides?.disableCache;
+        let cacheKey;
+        if (cacheEnabled) {
+            const cacheInputs = {
+                user: prompt.buildUserMessage(),
+                data: prompt.getData(),
+                system: effectiveSystem,
+                model: effectiveModel,
+                temperature: effectiveTemperature,
+                maxTokens: effectiveMaxTokens,
+                tools: this.config.tools,
+                mcps: this.config.mcps,
+                skills: this.config.skills,
+                responseFormat: prompt.getResponseFormat(),
+            };
+            cacheKey = generateCacheKey(cacheInputs);
+            const cached = await defaultCache.get(cacheKey);
+            if (cached) {
+                // Emit cache hit event
+                if (ctx) {
+                    this.emitWorkflowEvent({
+                        type: 'cacheHit',
+                        key: cacheKey,
+                        node: ctx.workflowNode,
+                    });
+                }
+                return cached;
+            }
+            // Emit cache miss event
+            if (ctx) {
+                this.emitWorkflowEvent({
+                    type: 'cacheMiss',
+                    key: cacheKey,
+                    node: ctx.workflowNode,
+                });
+            }
+        }
+        // Emit prompt start event if in workflow context
+        if (ctx) {
+            this.emitWorkflowEvent({
+                type: 'agentPromptStart',
+                agentId: this.id,
+                agentName: this.name,
+                promptId: prompt.id,
+                node: ctx.workflowNode,
+            });
+        }
+        const effectiveTools = this.mergeTools(prompt.toolsOverride ?? overrides?.tools ?? this.config.tools);
+        const effectiveHooks = this.mergeHooks(prompt.hooksOverride, overrides?.hooks, this.config.hooks);
+        const effectiveStop = overrides?.stop;
+        // Set up environment variables
+        const originalEnv = this.setupEnvironment(overrides?.env ?? this.config.env);
+        try {
+            // Call session start hooks
+            await this.callHooks(effectiveHooks?.sessionStart, {
+                agentId: this.id,
+                agentName: this.name,
+            });
+            // Build initial messages
+            const messages = [
+                { role: 'user', content: prompt.buildUserMessage() },
+            ];
+            // Execute conversation loop
+            let response = await this.callApi(messages, effectiveSystem, effectiveTools, effectiveModel, effectiveMaxTokens, effectiveTemperature, effectiveStop);
+            totalUsage = this.addUsage(totalUsage, response.usage);
+            // Handle tool use loop
+            while (response.stop_reason === 'tool_use') {
+                const toolUseBlocks = response.content.filter((block) => block.type === 'tool_use');
+                const toolResults = [];
+                for (const toolUse of toolUseBlocks) {
+                    toolCallCount++;
+                    // Call pre-tool hooks
+                    await this.callHooks(effectiveHooks?.preToolUse, {
+                        toolName: toolUse.name,
+                        toolInput: toolUse.input,
+                        agentId: this.id,
+                    });
+                    const toolStartTime = Date.now();
+                    // Execute tool
+                    const result = await this.executeTool(toolUse.name, toolUse.input);
+                    const toolDuration = Date.now() - toolStartTime;
+                    // Emit tool invocation event if in workflow context
+                    if (ctx) {
+                        this.emitWorkflowEvent({
+                            type: 'toolInvocation',
+                            toolName: toolUse.name,
+                            input: toolUse.input,
+                            output: result,
+                            duration: toolDuration,
+                            node: ctx.workflowNode,
+                        });
+                    }
+                    // Call post-tool hooks
+                    await this.callHooks(effectiveHooks?.postToolUse, {
+                        toolName: toolUse.name,
+                        toolInput: toolUse.input,
+                        toolOutput: result,
+                        agentId: this.id,
+                        duration: toolDuration,
+                    });
+                    toolResults.push({
+                        type: 'tool_result',
+                        tool_use_id: toolUse.id,
+                        content: typeof result === 'string' ? result : JSON.stringify(result),
+                    });
+                }
+                // Add assistant message with tool uses
+                messages.push({ role: 'assistant', content: response.content });
+                // Add tool results
+                messages.push({ role: 'user', content: toolResults });
+                // Continue conversation
+                response = await this.callApi(messages, effectiveSystem, effectiveTools, effectiveModel, effectiveMaxTokens, effectiveTemperature, effectiveStop);
+                totalUsage = this.addUsage(totalUsage, response.usage);
+            }
+            // Extract text response
+            const textContent = response.content.find((block) => block.type === 'text');
+            if (!textContent) {
+                throw new Error('No text response received from API');
+            }
+            // Parse JSON from response
+            const jsonMatch = textContent.text.match(/\{[\s\S]*\}/);
+            if (!jsonMatch) {
+                throw new Error('No JSON object found in response');
+            }
+            const parsed = JSON.parse(jsonMatch[0]);
+            // Validate with schema
+            const validated = prompt.validateResponse(parsed);
+            // Call session end hooks
+            await this.callHooks(effectiveHooks?.sessionEnd, {
+                agentId: this.id,
+                agentName: this.name,
+                totalDuration: Date.now() - startTime,
+            });
+            const duration = Date.now() - startTime;
+            // Emit prompt end event if in workflow context
+            if (ctx) {
+                this.emitWorkflowEvent({
+                    type: 'agentPromptEnd',
+                    agentId: this.id,
+                    agentName: this.name,
+                    promptId: prompt.id,
+                    node: ctx.workflowNode,
+                    duration,
+                    tokenUsage: totalUsage,
+                });
+            }
+            const result = {
+                data: validated,
+                usage: totalUsage,
+                duration,
+                toolCalls: toolCallCount,
+            };
+            // Store in cache if enabled
+            if (cacheEnabled && cacheKey) {
+                await defaultCache.set(cacheKey, result, { prefix: this.id });
+            }
+            return result;
+        }
+        finally {
+            // Restore environment
+            this.restoreEnvironment(originalEnv);
+        }
+    }
+    /**
+     * Call the Anthropic API
+     */
+    async callApi(messages, system, tools, model, maxTokens, temperature, stop) {
+        const params = {
+            model,
+            max_tokens: maxTokens,
+            messages,
+        };
+        if (system) {
+            params.system = system;
+        }
+        if (tools && tools.length > 0) {
+            params.tools = tools.map((tool) => ({
+                name: tool.name,
+                description: tool.description,
+                input_schema: tool.input_schema,
+            }));
+        }
+        if (temperature !== undefined) {
+            params.temperature = temperature;
+        }
+        if (stop && stop.length > 0) {
+            params.stop_sequences = stop;
+        }
+        return this.client.messages.create(params);
+    }
+    /**
+     * Execute a tool (either direct or via MCP)
+     */
+    async executeTool(name, input) {
+        // First, check stored MCPHandler instances (they have registered executors)
+        for (const handler of this.mcpHandlers) {
+            if (handler.hasTool(name)) {
+                const result = await handler.executeTool(name, input);
+                if (result.is_error) {
+                    throw new Error(result.content);
+                }
+                return result.content;
+            }
+        }
+        // Fall back to main mcpHandler (for non-MCPHandler MCPServers)
+        if (this.mcpHandler.hasTool(name)) {
+            const result = await this.mcpHandler.executeTool(name, input);
+            if (result.is_error) {
+                throw new Error(result.content);
+            }
+            return result.content;
+        }
+        // Look for direct tool handler - this would be set by subclasses
+        throw new Error(`No handler found for tool '${name}'`);
+    }
+    /**
+     * Merge tools from config and MCP servers
+     */
+    mergeTools(configTools) {
+        const mcpTools = this.mcpHandler.getTools();
+        const directTools = configTools ?? [];
+        const allTools = [...directTools, ...mcpTools];
+        return allTools.length > 0 ? allTools : undefined;
+    }
+    /**
+     * Merge hooks from multiple sources
+     */
+    mergeHooks(promptHooks, overrideHooks, configHooks) {
+        return {
+            preToolUse: [
+                ...(configHooks?.preToolUse ?? []),
+                ...(overrideHooks?.preToolUse ?? []),
+                ...(promptHooks?.preToolUse ?? []),
+            ],
+            postToolUse: [
+                ...(configHooks?.postToolUse ?? []),
+                ...(overrideHooks?.postToolUse ?? []),
+                ...(promptHooks?.postToolUse ?? []),
+            ],
+            sessionStart: [
+                ...(configHooks?.sessionStart ?? []),
+                ...(overrideHooks?.sessionStart ?? []),
+                ...(promptHooks?.sessionStart ?? []),
+            ],
+            sessionEnd: [
+                ...(configHooks?.sessionEnd ?? []),
+                ...(overrideHooks?.sessionEnd ?? []),
+                ...(promptHooks?.sessionEnd ?? []),
+            ],
+        };
+    }
+    /**
+     * Call hooks of a specific type
+     */
+    async callHooks(hooks, context) {
+        if (!hooks)
+            return;
+        for (const hook of hooks) {
+            await hook(context);
+        }
+    }
+    /**
+     * Set up environment variables
+     */
+    setupEnvironment(env) {
+        if (!env)
+            return {};
+        const original = {};
+        for (const [key, value] of Object.entries(env)) {
+            original[key] = process.env[key];
+            process.env[key] = value;
+        }
+        return original;
+    }
+    /**
+     * Restore environment variables
+     */
+    restoreEnvironment(original) {
+        for (const [key, value] of Object.entries(original)) {
+            if (value === undefined) {
+                delete process.env[key];
+            }
+            else {
+                process.env[key] = value;
+            }
+        }
+    }
+    /**
+     * Add token usage from response
+     */
+    addUsage(total, usage) {
+        return {
+            input_tokens: total.input_tokens + usage.input_tokens,
+            output_tokens: total.output_tokens + usage.output_tokens,
+        };
+    }
+}
+//# sourceMappingURL=agent.js.map

package/dist/core/agent.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent.js","sourceRoot":"","sources":["../../src/core/agent.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,SAAS,MAAM,mBAAmB,CAAC;AAc1C,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC9C,OAAO,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAC5C,OAAO,EAAE,mBAAmB,EAAE,MAAM,cAAc,CAAC;AACnD,OAAO,EAAE,gBAAgB,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAsBnE;;GAEG;AACH,MAAM,OAAO,KAAK;IAChB,gDAAgD;IAChC,EAAE,CAAS;IAE3B,iBAAiB;IACD,IAAI,CAAS;IAE7B,2BAA2B;IACV,MAAM,CAAc;IAErC,gCAAgC;IACf,MAAM,CAAY;IAEnC,sCAAsC;IACrB,UAAU,CAAa;IAExC,0DAA0D;IACzC,WAAW,GAAiB,EAAE,CAAC;IAEhD,2BAA2B;IACV,KAAK,CAAS;IAE/B;;;OAGG;IACH,YAAY,SAAsB,EAAE;QAClC,IAAI,CAAC,EAAE,GAAG,UAAU,EAAE,CAAC;QACvB,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,IAAI,OAAO,CAAC;QACnC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,0BAA0B,CAAC;QAExD,0BAA0B;QAC1B,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,CAAC;YAC1B,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,iBAAiB;SACtC,CAAC,CAAC;QAEH,yBAAyB;QACzB,IAAI,CAAC,UAAU,GAAG,IAAI,UAAU,EAAE,CAAC;QAEnC,uBAAuB;QACvB,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;YAChB,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;gBAC9B,kEAAkE;gBAClE,gEAAgE;gBAChE,IAAI,GAAG,YAAY,UAAU,EAAE,CAAC;oBAC9B,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBAC7B,CAAC;gBACD,uDAAuD;gBACvD,IAAI,CAAC,UAAU,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC;YACtC,CAAC;QACH,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACI,KAAK,CAAC,MAAM,CACjB,MAAiB,EACjB,SAA2B;QAE3B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;QAC3D,OAAO,MAAM,CAAC,IAAI,CAAC;IACrB,CAAC;IAED;;;;;OAKG;IACI,KAAK,CAAC,kBAAkB,CAC7B,MAAiB,EACjB,SAA2B;QAE3B,OAAO,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;IAC/C,CAAC;IAED;;;;;OAKG;IACI,KAAK,CAAC,OAAO,CAClB,MAAiB,EACjB,SAA2B;QAE3B,wDAAwD;QACxD,MAAM,iBAAiB,GACrB,MAAM,CAAC,gBAAgB;YACvB,SAAS,EAAE,gBAAgB;YAC3B,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC;QAE/B,MAAM,YAAY,GAAG,iBAAiB;YACpC,CAAC,CAAC,qJAAqJ;YACvJ,CAAC,CAAC,EAAE,CAAC;QAEP,MAAM,kBAAkB,GAAoB;YAC1C,GAAG,SAAS;YACZ,MAAM,EACJ,YAAY;gBACZ,CAAC,MAAM,CAAC,cAAc,IAAI,SAAS,EAAE,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,EAAE,CAAC;SAC3E,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,kBAAkB,CAAC,CAAC;QACpE,OAAO,MAAM,CAAC,IAAI,CAAC;IACrB,CAAC;IAED;;OAEG;IACI,aAAa;QAClB,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;IAED;;OAEG;IACK,iBAAiB,CAAC,KAAoB;QAC5C,MAAM,GAAG,GAAG,mBAAmB,EAAE,CAAC;QAClC,IAAI,GAAG,EAAE,CAAC;YACR,GAAG,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QACvB,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,aAAa,CACzB,MAAiB,EACjB,SAA2B;QAE3B,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,IAAI,aAAa,GAAG,CAAC,CAAC;QACtB,IAAI,UAAU,GAAe,EAAE,YAAY,EAAE,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE,CAAC;QAEnE,2CAA2C;QAC3C,MAAM,GAAG,GAAG,mBAAmB,EAAE,CAAC;QAElC,mDAAmD;QACnD,MAAM,eAAe,GACnB,MAAM,CAAC,cAAc,IAAI,SAAS,EAAE,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC;QAEnE,MAAM,cAAc,GAAG,SAAS,EAAE,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC;QACtD,MAAM,kBAAkB,GAAG,SAAS,EAAE,SAAS,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,IAAI,IAAI,CAAC;QACjF,MAAM,oBAAoB,GACxB,SAAS,EAAE,WAAW,IAAI,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC;QAEpD,yBAAyB;QACzB,MAAM,YAAY,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW,IAAI,CAAC,SAAS,EAAE,YAAY,CAAC;QACzE,IAAI,QAA4B,CAAC;QAEjC,IAAI,YAAY,EAAE,CAAC;YACjB,MAAM,WAAW,GAAmB;gBAClC,IAAI,EAAE,MAAM,CAAC,gBAAgB,EAAE;gBAC/B,IAAI,EAAE,MAAM,CAAC,OAAO,EAAE;gBACtB,MAAM,EAAE,eAAe;gBACvB,KAAK,EAAE,cAAc;gBACrB,WAAW,EAAE,oBAAoB;gBACjC,SAAS,EAAE,kBAAkB;gBAC7B,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK;gBACxB,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;gBACtB,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM;gBAC1B,cAAc,EAAE,MAAM,CAAC,iBAAiB,EAAE;aAC3C,CAAC;YACF,QAAQ,GAAG,gBAAgB,CAAC,WAAW,CAAC,CAAC;YAEzC,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAgC,CAAC;YAC/E,IAAI,MAAM,EAAE,CAAC;gBACX,uBAAuB;gBACvB,IAAI,GAAG,EAAE,CAAC;oBACR,IAAI,CAAC,iBAAiB,CAAC;wBACrB,IAAI,EAAE,UAAU;wBAChB,GAAG,EAAE,QAAQ;wBACb,IAAI,EAAE,GAAG,CAAC,YAAY;qBACvB,CAAC,CAAC;gBACL,CAAC;gBACD,OAAO,MAAM,CAAC;YAChB,CAAC;YAED,wBAAwB;YACxB,IAAI,GAAG,EAAE,CAAC;gBACR,IAAI,CAAC,iBAAiB,CAAC;oBACrB,IAAI,EAAE,WAAW;oBACjB,GAAG,EAAE,QAAQ;oBACb,IAAI,EAAE,GAAG,CAAC,YAAY;iBACvB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,iDAAiD;QACjD,IAAI,GAAG,EAAE,CAAC;YACR,IAAI,CAAC,iBAAiB,CAAC;gBACrB,IAAI,EAAE,kBAAkB;gBACxB,OAAO,EAAE,IAAI,CAAC,EAAE;gBAChB,SAAS,EAAE,IAAI,CAAC,IAAI;gBACpB,QAAQ,EAAE,MAAM,CAAC,EAAE;gBACnB,IAAI,EAAE,GAAG,CAAC,YAAY;aACvB,CAAC,CAAC;QACL,CAAC;QAED,MAAM,cAAc,GAAG,IAAI,CAAC,UAAU,CACpC,MAAM,CAAC,aAAa,IAAI,SAAS,EAAE,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,KAAK,CAC9D,CAAC;QAEF,MAAM,cAAc,GAAG,IAAI,CAAC,UAAU,CACpC,MAAM,CAAC,aAAa,EACpB,SAAS,EAAE,KAAK,EAChB,IAAI,CAAC,MAAM,CAAC,KAAK,CAClB,CAAC;QAEF,MAAM,aAAa,GAAG,SAAS,EAAE,IAAI,CAAC;QAEtC,+BAA+B;QAC/B,MAAM,WAAW,GAAG,IAAI,CAAC,gBAAgB,CAAC,SAAS,EAAE,GAAG,IAAI,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAE7E,IAAI,CAAC;YACH,2BAA2B;YAC3B,MAAM,IAAI,CAAC,SAAS,CAAC,cAAc,EAAE,YAAY,EAAE;gBACjD,OAAO,EAAE,IAAI,CAAC,EAAE;gBAChB,SAAS,EAAE,IAAI,CAAC,IAAI;aACE,CAAC,CAAC;YAE1B,yBAAyB;YACzB,MAAM,QAAQ,GAAc;gBAC1B,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,gBAAgB,EAAE,EAAE;aACrD,CAAC;YAEF,4BAA4B;YAC5B,IAAI,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAC/B,QAAQ,EACR,eAAe,EACf,cAAc,EACd,cAAc,EACd,kBAAkB,EAClB,oBAAoB,EACpB,aAAa,CACd,CAAC;YAEF,UAAU,GAAG,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC;YAEvD,uBAAuB;YACvB,OAAO,QAAQ,CAAC,WAAW,KAAK,UAAU,EAAE,CAAC;gBAC3C,MAAM,aAAa,GAAG,QAAQ,CAAC,OAAO,CAAC,MAAM,CAC3C,CAAC,KAAK,EAAmC,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,UAAU,CACtE,CAAC;gBAEF,MAAM,WAAW,GAAqC,EAAE,CAAC;gBAEzD,KAAK,MAAM,OAAO,IAAI,aAAa,EAAE,CAAC;oBACpC,aAAa,EAAE,CAAC;oBAEhB,sBAAsB;oBACtB,MAAM,IAAI,CAAC,SAAS,CAAC,cAAc,EAAE,UAAU,EAAE;wBAC/C,QAAQ,EAAE,OAAO,CAAC,IAAI;wBACtB,SAAS,EAAE,OAAO,CAAC,KAAK;wBACxB,OAAO,EAAE,IAAI,CAAC,EAAE;qBACI,CAAC,CAAC;oBAExB,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;oBAEjC,eAAe;oBACf,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;oBAEnE,MAAM,YAAY,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,aAAa,CAAC;oBAEhD,oDAAoD;oBACpD,IAAI,GAAG,EAAE,CAAC;wBACR,IAAI,CAAC,iBAAiB,CAAC;4BACrB,IAAI,EAAE,gBAAgB;4BACtB,QAAQ,EAAE,OAAO,CAAC,IAAI;4BACtB,KAAK,EAAE,OAAO,CAAC,KAAK;4BACpB,MAAM,EAAE,MAAM;4BACd,QAAQ,EAAE,YAAY;4BACtB,IAAI,EAAE,GAAG,CAAC,YAAY;yBACvB,CAAC,CAAC;oBACL,CAAC;oBAED,uBAAuB;oBACvB,MAAM,IAAI,CAAC,SAAS,CAAC,cAAc,EAAE,WAAW,EAAE;wBAChD,QAAQ,EAAE,OAAO,CAAC,IAAI;wBACtB,SAAS,EAAE,OAAO,CAAC,KAAK;wBACxB,UAAU,EAAE,MAAM;wBAClB,OAAO,EAAE,IAAI,CAAC,EAAE;wBAChB,QAAQ,EAAE,YAAY;qBACD,CAAC,CAAC;oBAEzB,WAAW,CAAC,IAAI,CAAC;wBACf,IAAI,EAAE,aAAa;wBACnB,WAAW,EAAE,OAAO,CAAC,EAAE;wBACvB,OAAO,EACL,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;qBAC/D,CAAC,CAAC;gBACL,CAAC;gBAED,uCAAuC;gBACvC,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,QAAQ,CAAC,OAAO,EAAE,CAAC,CAAC;gBAEhE,mBAAmB;gBACnB,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,CAAC;gBAEtD,wBAAwB;gBACxB,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAC3B,QAAQ,EACR,eAAe,EACf,cAAc,EACd,cAAc,EACd,kBAAkB,EAClB,oBAAoB,EACpB,aAAa,CACd,CAAC;gBAEF,UAAU,GAAG,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC;YACzD,CAAC;YAED,wBAAwB;YACxB,MAAM,WAAW,GAAG,QAAQ,CAAC,OAAO,CAAC,IAAI,CACvC,CAAC,KAAK,EAAgC,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,MAAM,CAC/D,CAAC;YAEF,IAAI,CAAC,WAAW,EAAE,CAAC;gBACjB,MAAM,IAAI,KAAK,CAAC,oCAAoC,CAAC,CAAC;YACxD,CAAC;YAED,2BAA2B;YAC3B,MAAM,SAAS,GAAG,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;YACxD,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;YACtD,CAAC;YAED,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;YAExC,uBAAuB;YACvB,MAAM,SAAS,GAAG,MAAM,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;YAElD,yBAAyB;YACzB,MAAM,IAAI,CAAC,SAAS,CAAC,cAAc,EAAE,UAAU,EAAE;gBAC/C,OAAO,EAAE,IAAI,CAAC,EAAE;gBAChB,SAAS,EAAE,IAAI,CAAC,IAAI;gBACpB,aAAa,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;aACjB,CAAC,CAAC;YAExB,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAExC,+CAA+C;YAC/C,IAAI,GAAG,EAAE,CAAC;gBACR,IAAI,CAAC,iBAAiB,CAAC;oBACrB,IAAI,EAAE,gBAAgB;oBACtB,OAAO,EAAE,IAAI,CAAC,EAAE;oBAChB,SAAS,EAAE,IAAI,CAAC,IAAI;oBACpB,QAAQ,EAAE,MAAM,CAAC,EAAE;oBACnB,IAAI,EAAE,GAAG,CAAC,YAAY;oBACtB,QAAQ;oBACR,UAAU,EAAE,UAAU;iBACvB,CAAC,CAAC;YACL,CAAC;YAED,MAAM,MAAM,GAAoB;gBAC9B,IAAI,EAAE,SAAS;gBACf,KAAK,EAAE,UAAU;gBACjB,QAAQ;gBACR,SAAS,EAAE,aAAa;aACzB,CAAC;YAEF,4BAA4B;YAC5B,IAAI,YAAY,IAAI,QAAQ,EAAE,CAAC;gBAC7B,MAAM,YAAY,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;YAChE,CAAC;YAED,OAAO,MAAM,CAAC;QAChB,CAAC;gBAAS,CAAC;YACT,sBAAsB;YACtB,IAAI,CAAC,kBAAkB,CAAC,WAAW,CAAC,CAAC;QACvC,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,OAAO,CACnB,QAAmB,EACnB,MAA0B,EAC1B,KAAyB,EACzB,KAAa,EACb,SAAiB,EACjB,WAA+B,EAC/B,IAA0B;QAE1B,MAAM,MAAM,GAAkC;YAC5C,KAAK;YACL,UAAU,EAAE,SAAS;YACrB,QAAQ;SACT,CAAC;QAEF,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,CAAC,MAAM,GAAG,MAAM,CAAC;QACzB,CAAC;QAED,IAAI,KAAK,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,MAAM,CAAC,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBAClC,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,WAAW,EAAE,IAAI,CAAC,WAAW;gBAC7B,YAAY,EAAE,IAAI,CAAC,YAA8C;aAClE,CAAC,CAAC,CAAC;QACN,CAAC;QAED,IAAI,WAAW,KAAK,SAAS,EAAE,CAAC;YAC9B,MAAM,CAAC,WAAW,GAAG,WAAW,CAAC;QACnC,CAAC;QAED,IAAI,IAAI,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5B,MAAM,CAAC,cAAc,GAAG,IAAI,CAAC;QAC/B,CAAC;QAED,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IAC7C,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,WAAW,CAAC,IAAY,EAAE,KAAc;QACpD,4EAA4E;QAC5E,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACvC,IAAI,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC1B,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,WAAW,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;gBACtD,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;oBACpB,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC,OAAiB,CAAC,CAAC;gBAC5C,CAAC;gBACD,OAAO,MAAM,CAAC,OAAO,CAAC;YACxB,CAAC;QACH,CAAC;QAED,+DAA+D;QAC/D,IAAI,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;YAClC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YAC9D,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;gBACpB,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC,OAAiB,CAAC,CAAC;YAC5C,CAAC;YACD,OAAO,MAAM,CAAC,OAAO,CAAC;QACxB,CAAC;QAED,iEAAiE;QACjE,MAAM,IAAI,KAAK,CAAC,8BAA8B,IAAI,GAAG,CAAC,CAAC;IACzD,CAAC;IAED;;OAEG;IACK,UAAU,CAAC,WAAoB;QACrC,MAAM,QAAQ,GAAG,IAAI,CAAC,UAAU,CAAC,QAAQ,EAAE,CAAC;QAC5C,MAAM,WAAW,GAAG,WAAW,IAAI,EAAE,CAAC;QAEtC,MAAM,QAAQ,GAAG,CAAC,GAAG,WAAW,EAAE,GAAG,QAAQ,CAAC,CAAC;QAC/C,OAAO,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC;IACpD,CAAC;IAED;;OAEG;IACK,UAAU,CAChB,WAAwB,EACxB,aAA0B,EAC1B,WAAwB;QAExB,OAAO;YACL,UAAU,EAAE;gBACV,GAAG,CAAC,WAAW,EAAE,UAAU,IAAI,EAAE,CAAC;gBAClC,GAAG,CAAC,aAAa,EAAE,UAAU,IAAI,EAAE,CAAC;gBACpC,GAAG,CAAC,WAAW,EAAE,UAAU,IAAI,EAAE,CAAC;aACnC;YACD,WAAW,EAAE;gBACX,GAAG,CAAC,WAAW,EAAE,WAAW,IAAI,EAAE,CAAC;gBACnC,GAAG,CAAC,aAAa,EAAE,WAAW,IAAI,EAAE,CAAC;gBACrC,GAAG,CAAC,WAAW,EAAE,WAAW,IAAI,EAAE,CAAC;aACpC;YACD,YAAY,EAAE;gBACZ,GAAG,CAAC,WAAW,EAAE,YAAY,IAAI,EAAE,CAAC;gBACpC,GAAG,CAAC,aAAa,EAAE,YAAY,IAAI,EAAE,CAAC;gBACtC,GAAG,CAAC,WAAW,EAAE,YAAY,IAAI,EAAE,CAAC;aACrC;YACD,UAAU,EAAE;gBACV,GAAG,CAAC,WAAW,EAAE,UAAU,IAAI,EAAE,CAAC;gBAClC,GAAG,CAAC,aAAa,EAAE,UAAU,IAAI,EAAE,CAAC;gBACpC,GAAG,CAAC,WAAW,EAAE,UAAU,IAAI,EAAE,CAAC;aACnC;SACF,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,SAAS,CACrB,KAA2D,EAC3D,OAAU;QAEV,IAAI,CAAC,KAAK;YAAE,OAAO;QACnB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,IAAI,CAAC,OAAO,CAAC,CAAC;QACtB,CAAC;IACH,CAAC;IAED;;OAEG;IACK,gBAAgB,CACtB,GAA4B;QAE5B,IAAI,CAAC,GAAG;YAAE,OAAO,EAAE,CAAC;QAEpB,MAAM,QAAQ,GAAuC,EAAE,CAAC;QACxD,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;YAC/C,QAAQ,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YACjC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;QAC3B,CAAC;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED;;OAEG;IACK,kBAAkB,CAAC,QAA4C;QACrE,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;YACpD,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;gBACxB,OAAO,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAC1B,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;YAC3B,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACK,QAAQ,CAAC,KAAiB,EAAE,KAAsB;QACxD,OAAO;YACL,YAAY,EAAE,KAAK,CAAC,YAAY,GAAG,KAAK,CAAC,YAAY;YACrD,aAAa,EAAE,KAAK,CAAC,aAAa,GAAG,KAAK,CAAC,aAAa;SACzD,CAAC;IACJ,CAAC;CACF"}

package/{src/core/context.ts → dist/core/context.d.ts}

@@ -4,116 +4,65 @@
  * Uses Node.js AsyncLocalStorage to automatically propagate context
  * through async call chains without manual passing.
  */
-
 import { AsyncLocalStorage } from 'node:async_hooks';
 import type { WorkflowNode, WorkflowEvent } from '../types/index.js';
-
 /**
  * Context available during agent/prompt execution
  */
 export interface AgentExecutionContext {
-  /** Current workflow node in the hierarchy */
-  workflowNode: WorkflowNode;
-
-  /** Function to emit events to the workflow tree */
-  emitEvent: (event: WorkflowEvent) => void;
-
-  /** Workflow ID for tracking */
-  workflowId: string;
-
-  /** Parent workflow ID if nested */
-  parentWorkflowId?: string;
+    /** Current workflow node in the hierarchy */
+    workflowNode: WorkflowNode;
+    /** Function to emit events to the workflow tree */
+    emitEvent: (event: WorkflowEvent) => void;
+    /** Workflow ID for tracking */
+    workflowId: string;
+    /** Parent workflow ID if nested */
+    parentWorkflowId?: string;
 }
-
 /**
  * Global AsyncLocalStorage instance for execution context
  * Single shared instance per application (recommended pattern)
  */
-const executionContext = new AsyncLocalStorage<AgentExecutionContext>();
-
+declare const executionContext: AsyncLocalStorage<AgentExecutionContext>;
 /**
  * Get the current execution context
  * @returns Current context or undefined if not in a workflow
  */
-export function getExecutionContext(): AgentExecutionContext | undefined {
-  return executionContext.getStore();
-}
-
+export declare function getExecutionContext(): AgentExecutionContext | undefined;
 /**
  * Get the current execution context, throwing if not available
  * @param operation Name of the operation requiring context
  * @returns Current context
  * @throws Error if not within a workflow context
  */
-export function requireExecutionContext(
-  operation: string
-): AgentExecutionContext {
-  const context = executionContext.getStore();
-  if (!context) {
-    throw new Error(
-      `${operation} called outside of workflow context. ` +
-        `Agent/Prompt operations must be executed within a workflow step.`
-    );
-  }
-  return context;
-}
-
+export declare function requireExecutionContext(operation: string): AgentExecutionContext;
 /**
  * Run a function within an execution context
  * @param context Context to establish
  * @param fn Function to execute
  * @returns Result of the function
  */
-export async function runInContext<T>(
-  context: AgentExecutionContext,
-  fn: () => Promise<T>
-): Promise<T> {
-  return executionContext.run(context, fn);
-}
-
+export declare function runInContext<T>(context: AgentExecutionContext, fn: () => Promise<T>): Promise<T>;
 /**
  * Run a synchronous function within an execution context
  * @param context Context to establish
  * @param fn Function to execute
  * @returns Result of the function
  */
-export function runInContextSync<T>(
-  context: AgentExecutionContext,
-  fn: () => T
-): T {
-  return executionContext.run(context, fn);
-}
-
+export declare function runInContextSync<T>(context: AgentExecutionContext, fn: () => T): T;
 /**
  * Check if currently within an execution context
  * @returns true if context is available
  */
-export function hasExecutionContext(): boolean {
-  return executionContext.getStore() !== undefined;
-}
-
+export declare function hasExecutionContext(): boolean;
 /**
  * Create a child context with updated node
  * @param childNode The new workflow node for the child context
  * @returns New context with child node
  */
-export function createChildContext(
-  childNode: WorkflowNode
-): AgentExecutionContext | undefined {
-  const parent = getExecutionContext();
-  if (!parent) {
-    return undefined;
-  }
-
-  return {
-    workflowNode: childNode,
-    emitEvent: parent.emitEvent,
-    workflowId: parent.workflowId,
-    parentWorkflowId: parent.parentWorkflowId,
-  };
-}
-
+export declare function createChildContext(childNode: WorkflowNode): AgentExecutionContext | undefined;
 /**
  * Export the raw AsyncLocalStorage for advanced use cases
  */
 export { executionContext as agentExecutionStorage };
+//# sourceMappingURL=context.d.ts.map

package/dist/core/context.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"context.d.ts","sourceRoot":"","sources":["../../src/core/context.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AACrD,OAAO,KAAK,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAErE;;GAEG;AACH,MAAM,WAAW,qBAAqB;IACpC,6CAA6C;IAC7C,YAAY,EAAE,YAAY,CAAC;IAE3B,mDAAmD;IACnD,SAAS,EAAE,CAAC,KAAK,EAAE,aAAa,KAAK,IAAI,CAAC;IAE1C,+BAA+B;IAC/B,UAAU,EAAE,MAAM,CAAC;IAEnB,mCAAmC;IACnC,gBAAgB,CAAC,EAAE,MAAM,CAAC;CAC3B;AAED;;;GAGG;AACH,QAAA,MAAM,gBAAgB,0CAAiD,CAAC;AAExE;;;GAGG;AACH,wBAAgB,mBAAmB,IAAI,qBAAqB,GAAG,SAAS,CAEvE;AAED;;;;;GAKG;AACH,wBAAgB,uBAAuB,CACrC,SAAS,EAAE,MAAM,GAChB,qBAAqB,CASvB;AAED;;;;;GAKG;AACH,wBAAsB,YAAY,CAAC,CAAC,EAClC,OAAO,EAAE,qBAAqB,EAC9B,EAAE,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,GACnB,OAAO,CAAC,CAAC,CAAC,CAEZ;AAED;;;;;GAKG;AACH,wBAAgB,gBAAgB,CAAC,CAAC,EAChC,OAAO,EAAE,qBAAqB,EAC9B,EAAE,EAAE,MAAM,CAAC,GACV,CAAC,CAEH;AAED;;;GAGG;AACH,wBAAgB,mBAAmB,IAAI,OAAO,CAE7C;AAED;;;;GAIG;AACH,wBAAgB,kBAAkB,CAChC,SAAS,EAAE,YAAY,GACtB,qBAAqB,GAAG,SAAS,CAYnC;AAED;;GAEG;AACH,OAAO,EAAE,gBAAgB,IAAI,qBAAqB,EAAE,CAAC"}

package/dist/core/context.js

@@ -0,0 +1,80 @@
+/**
+ * AgentExecutionContext - Provides zero-plumbing context propagation
+ *
+ * Uses Node.js AsyncLocalStorage to automatically propagate context
+ * through async call chains without manual passing.
+ */
+import { AsyncLocalStorage } from 'node:async_hooks';
+/**
+ * Global AsyncLocalStorage instance for execution context
+ * Single shared instance per application (recommended pattern)
+ */
+const executionContext = new AsyncLocalStorage();
+/**
+ * Get the current execution context
+ * @returns Current context or undefined if not in a workflow
+ */
+export function getExecutionContext() {
+    return executionContext.getStore();
+}
+/**
+ * Get the current execution context, throwing if not available
+ * @param operation Name of the operation requiring context
+ * @returns Current context
+ * @throws Error if not within a workflow context
+ */
+export function requireExecutionContext(operation) {
+    const context = executionContext.getStore();
+    if (!context) {
+        throw new Error(`${operation} called outside of workflow context. ` +
+            `Agent/Prompt operations must be executed within a workflow step.`);
+    }
+    return context;
+}
+/**
+ * Run a function within an execution context
+ * @param context Context to establish
+ * @param fn Function to execute
+ * @returns Result of the function
+ */
+export async function runInContext(context, fn) {
+    return executionContext.run(context, fn);
+}
+/**
+ * Run a synchronous function within an execution context
+ * @param context Context to establish
+ * @param fn Function to execute
+ * @returns Result of the function
+ */
+export function runInContextSync(context, fn) {
+    return executionContext.run(context, fn);
+}
+/**
+ * Check if currently within an execution context
+ * @returns true if context is available
+ */
+export function hasExecutionContext() {
+    return executionContext.getStore() !== undefined;
+}
+/**
+ * Create a child context with updated node
+ * @param childNode The new workflow node for the child context
+ * @returns New context with child node
+ */
+export function createChildContext(childNode) {
+    const parent = getExecutionContext();
+    if (!parent) {
+        return undefined;
+    }
+    return {
+        workflowNode: childNode,
+        emitEvent: parent.emitEvent,
+        workflowId: parent.workflowId,
+        parentWorkflowId: parent.parentWorkflowId,
+    };
+}
+/**
+ * Export the raw AsyncLocalStorage for advanced use cases
+ */
+export { executionContext as agentExecutionStorage };
+//# sourceMappingURL=context.js.map

package/dist/core/context.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"context.js","sourceRoot":"","sources":["../../src/core/context.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAoBrD;;;GAGG;AACH,MAAM,gBAAgB,GAAG,IAAI,iBAAiB,EAAyB,CAAC;AAExE;;;GAGG;AACH,MAAM,UAAU,mBAAmB;IACjC,OAAO,gBAAgB,CAAC,QAAQ,EAAE,CAAC;AACrC,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,uBAAuB,CACrC,SAAiB;IAEjB,MAAM,OAAO,GAAG,gBAAgB,CAAC,QAAQ,EAAE,CAAC;IAC5C,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,MAAM,IAAI,KAAK,CACb,GAAG,SAAS,uCAAuC;YACjD,kEAAkE,CACrE,CAAC;IACJ,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED;;;;;GAKG;AACH,MAAM,CAAC,KAAK,UAAU,YAAY,CAChC,OAA8B,EAC9B,EAAoB;IAEpB,OAAO,gBAAgB,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;AAC3C,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,gBAAgB,CAC9B,OAA8B,EAC9B,EAAW;IAEX,OAAO,gBAAgB,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;AAC3C,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,mBAAmB;IACjC,OAAO,gBAAgB,CAAC,QAAQ,EAAE,KAAK,SAAS,CAAC;AACnD,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,kBAAkB,CAChC,SAAuB;IAEvB,MAAM,MAAM,GAAG,mBAAmB,EAAE,CAAC;IACrC,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,OAAO;QACL,YAAY,EAAE,SAAS;QACvB,SAAS,EAAE,MAAM,CAAC,SAAS;QAC3B,UAAU,EAAE,MAAM,CAAC,UAAU;QAC7B,gBAAgB,EAAE,MAAM,CAAC,gBAAgB;KAC1C,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,OAAO,EAAE,gBAAgB,IAAI,qBAAqB,EAAE,CAAC"}