@agntk/core 0.3.3 → 1.0.0

package/dist/usage-limits.d.ts

@@ -1,45 +1,17 @@
-/**
- * @agntk/core - Usage Limits
- *
- * Token and request caps for agents. Prevents runaway loops from
- * burning API budget by enforcing maxRequests, maxInputTokens,
- * maxOutputTokens, and maxTotalTokens.
- */
 import type { ToolSet, StepResult } from 'ai';
-/** Usage limits for an agent run. All fields are optional — only set limits are enforced. */
 export interface UsageLimits {
-    /** Maximum number of LLM round-trips (steps). */
     maxRequests?: number;
-    /** Maximum cumulative input tokens across all steps. */
     maxInputTokens?: number;
-    /** Maximum cumulative output tokens across all steps. */
     maxOutputTokens?: number;
-    /** Maximum cumulative total tokens (input + output) across all steps. */
     maxTotalTokens?: number;
 }
-/** Which limit was exceeded. */
 export type UsageLimitType = 'maxRequests' | 'maxInputTokens' | 'maxOutputTokens' | 'maxTotalTokens';
-/** Current usage snapshot at the time a limit was exceeded. */
 export interface UsageSnapshot {
     requests: number;
     inputTokens: number;
     outputTokens: number;
     totalTokens: number;
 }
-/**
- * Thrown when an agent exceeds a configured usage limit.
- *
- * @example
- * ```typescript
- * try {
- *   const result = await agent.stream({ prompt: 'do something' });
- * } catch (error) {
- *   if (error instanceof UsageLimitExceeded) {
- *     console.log(`Hit ${error.limitType}: ${error.currentValue}/${error.limitValue}`);
- *   }
- * }
- * ```
- */
 export declare class UsageLimitExceeded extends Error {
     readonly name = "UsageLimitExceeded";
     readonly limitType: UsageLimitType;
@@ -48,15 +20,6 @@ export declare class UsageLimitExceeded extends Error {
     readonly usage: UsageSnapshot;
     constructor(limitType: UsageLimitType, limitValue: number, currentValue: number, usage: UsageSnapshot);
 }
-/**
- * Create a StopCondition that checks usage limits after each step.
- *
- * Compatible with AI SDK's `stopWhen` array — combine with `stepCountIs()`.
- *
- * When a limit is exceeded, the condition throws `UsageLimitExceeded`
- * rather than returning `true`, because returning `true` would silently
- * stop the agent. Throwing gives the caller a clear signal with details.
- */
 export declare function usageLimitStop<TOOLS extends ToolSet>(limits: UsageLimits): (options: {
     steps: Array<StepResult<TOOLS>>;
 }) => boolean;

package/dist/usage-limits.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"usage-limits.d.ts","sourceRoot":"","sources":["../src/usage-limits.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,MAAM,IAAI,CAAC;AAM9C,6FAA6F;AAC7F,MAAM,WAAW,WAAW;IAC1B,iDAAiD;IACjD,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,wDAAwD;IACxD,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,yDAAyD;IACzD,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,yEAAyE;IACzE,cAAc,CAAC,EAAE,MAAM,CAAC;CACzB;AAED,gCAAgC;AAChC,MAAM,MAAM,cAAc,GAAG,aAAa,GAAG,gBAAgB,GAAG,iBAAiB,GAAG,gBAAgB,CAAC;AAErG,+DAA+D;AAC/D,MAAM,WAAW,aAAa;IAC5B,QAAQ,EAAE,MAAM,CAAC;IACjB,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC;CACrB;AAMD;;;;;;;;;;;;;GAaG;AACH,qBAAa,kBAAmB,SAAQ,KAAK;IAC3C,SAAkB,IAAI,wBAAwB;IAC9C,QAAQ,CAAC,SAAS,EAAE,cAAc,CAAC;IACnC,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;IAC5B,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;IAC9B,QAAQ,CAAC,KAAK,EAAE,aAAa,CAAC;gBAG5B,SAAS,EAAE,cAAc,EACzB,UAAU,EAAE,MAAM,EAClB,YAAY,EAAE,MAAM,EACpB,KAAK,EAAE,aAAa;CAUvB;AAMD;;;;;;;;GAQG;AACH,wBAAgB,cAAc,CAAC,KAAK,SAAS,OAAO,EAClD,MAAM,EAAE,WAAW,GAClB,CAAC,OAAO,EAAE;IAAE,KAAK,EAAE,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CAAA;CAAE,KAAK,OAAO,CAsB3D"}
+{"version":3,"file":"usage-limits.d.ts","sourceRoot":"","sources":["../src/usage-limits.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,MAAM,IAAI,CAAC;AAE9C,MAAM,WAAW,WAAW;IAC1B,WAAW,CAAC,EAAE,MAAM,CAAC;IAErB,cAAc,CAAC,EAAE,MAAM,CAAC;IAExB,eAAe,CAAC,EAAE,MAAM,CAAC;IAEzB,cAAc,CAAC,EAAE,MAAM,CAAC;CACzB;AAED,MAAM,MAAM,cAAc,GACtB,aAAa,GACb,gBAAgB,GAChB,iBAAiB,GACjB,gBAAgB,CAAC;AAErB,MAAM,WAAW,aAAa;IAC5B,QAAQ,EAAE,MAAM,CAAC;IACjB,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC;CACrB;AAED,qBAAa,kBAAmB,SAAQ,KAAK;IAC3C,SAAkB,IAAI,wBAAwB;IAC9C,QAAQ,CAAC,SAAS,EAAE,cAAc,CAAC;IACnC,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;IAC5B,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;IAC9B,QAAQ,CAAC,KAAK,EAAE,aAAa,CAAC;gBAG5B,SAAS,EAAE,cAAc,EACzB,UAAU,EAAE,MAAM,EAClB,YAAY,EAAE,MAAM,EACpB,KAAK,EAAE,aAAa;CAQvB;AAED,wBAAgB,cAAc,CAAC,KAAK,SAAS,OAAO,EAClD,MAAM,EAAE,WAAW,GAClB,CAAC,OAAO,EAAE;IAAE,KAAK,EAAE,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CAAA;CAAE,KAAK,OAAO,CAqC3D"}

package/dist/usage-limits.js

@@ -1,27 +1,3 @@
-/**
- * @agntk/core - Usage Limits
- *
- * Token and request caps for agents. Prevents runaway loops from
- * burning API budget by enforcing maxRequests, maxInputTokens,
- * maxOutputTokens, and maxTotalTokens.
- */
-// ============================================================================
-// Error
-// ============================================================================
-/**
- * Thrown when an agent exceeds a configured usage limit.
- *
- * @example
- * ```typescript
- * try {
- *   const result = await agent.stream({ prompt: 'do something' });
- * } catch (error) {
- *   if (error instanceof UsageLimitExceeded) {
- *     console.log(`Hit ${error.limitType}: ${error.currentValue}/${error.limitValue}`);
- *   }
- * }
- * ```
- */
 export class UsageLimitExceeded extends Error {
     name = 'UsageLimitExceeded';
     limitType;
@@ -36,18 +12,6 @@ export class UsageLimitExceeded extends Error {
         this.usage = usage;
     }
 }
-// ============================================================================
-// Stop Condition Factory
-// ============================================================================
-/**
- * Create a StopCondition that checks usage limits after each step.
- *
- * Compatible with AI SDK's `stopWhen` array — combine with `stepCountIs()`.
- *
- * When a limit is exceeded, the condition throws `UsageLimitExceeded`
- * rather than returning `true`, because returning `true` would silently
- * stop the agent. Throwing gives the caller a clear signal with details.
- */
 export function usageLimitStop(limits) {
     return ({ steps }) => {
         const usage = computeUsage(steps);
@@ -63,12 +27,9 @@ export function usageLimitStop(limits) {
         if (limits.maxTotalTokens !== undefined && usage.totalTokens > limits.maxTotalTokens) {
             throw new UsageLimitExceeded('maxTotalTokens', limits.maxTotalTokens, usage.totalTokens, usage);
         }
-        return false; // Don't stop — limits not reached
+        return false;
     };
 }
-// ============================================================================
-// Helpers
-// ============================================================================
 function computeUsage(steps) {
     let inputTokens = 0;
     let outputTokens = 0;

package/dist/usage-limits.js.map

@@ -1 +1 @@
-{"version":3,"file":"usage-limits.js","sourceRoot":"","sources":["../src/usage-limits.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AA+BH,+EAA+E;AAC/E,QAAQ;AACR,+EAA+E;AAE/E;;;;;;;;;;;;;GAaG;AACH,MAAM,OAAO,kBAAmB,SAAQ,KAAK;IACzB,IAAI,GAAG,oBAAoB,CAAC;IACrC,SAAS,CAAiB;IAC1B,UAAU,CAAS;IACnB,YAAY,CAAS;IACrB,KAAK,CAAgB;IAE9B,YACE,SAAyB,EACzB,UAAkB,EAClB,YAAoB,EACpB,KAAoB;QAEpB,KAAK,CACH,yBAAyB,SAAS,MAAM,YAAY,YAAY,UAAU,GAAG,CAC9E,CAAC;QACF,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;QACjC,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;CACF;AAED,+EAA+E;AAC/E,yBAAyB;AACzB,+EAA+E;AAE/E;;;;;;;;GAQG;AACH,MAAM,UAAU,cAAc,CAC5B,MAAmB;IAEnB,OAAO,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE;QACnB,MAAM,KAAK,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC;QAElC,IAAI,MAAM,CAAC,WAAW,KAAK,SAAS,IAAI,KAAK,CAAC,QAAQ,GAAG,MAAM,CAAC,WAAW,EAAE,CAAC;YAC5E,MAAM,IAAI,kBAAkB,CAAC,aAAa,EAAE,MAAM,CAAC,WAAW,EAAE,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QACzF,CAAC;QAED,IAAI,MAAM,CAAC,cAAc,KAAK,SAAS,IAAI,KAAK,CAAC,WAAW,GAAG,MAAM,CAAC,cAAc,EAAE,CAAC;YACrF,MAAM,IAAI,kBAAkB,CAAC,gBAAgB,EAAE,MAAM,CAAC,cAAc,EAAE,KAAK,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC;QAClG,CAAC;QAED,IAAI,MAAM,CAAC,eAAe,KAAK,SAAS,IAAI,KAAK,CAAC,YAAY,GAAG,MAAM,CAAC,eAAe,EAAE,CAAC;YACxF,MAAM,IAAI,kBAAkB,CAAC,iBAAiB,EAAE,MAAM,CAAC,eAAe,EAAE,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC;QACrG,CAAC;QAED,IAAI,MAAM,CAAC,cAAc,KAAK,SAAS,IAAI,KAAK,CAAC,WAAW,GAAG,MAAM,CAAC,cAAc,EAAE,CAAC;YACrF,MAAM,IAAI,kBAAkB,CAAC,gBAAgB,EAAE,MAAM,CAAC,cAAc,EAAE,KAAK,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC;QAClG,CAAC;QAED,OAAO,KAAK,CAAC,CAAC,kCAAkC;IAClD,CAAC,CAAC;AACJ,CAAC;AAED,+EAA+E;AAC/E,UAAU;AACV,+EAA+E;AAE/E,SAAS,YAAY,CAAwB,KAA+B;IAC1E,IAAI,WAAW,GAAG,CAAC,CAAC;IACpB,IAAI,YAAY,GAAG,CAAC,CAAC;IACrB,IAAI,WAAW,GAAG,CAAC,CAAC;IAEpB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,WAAW,IAAI,IAAI,CAAC,KAAK,EAAE,WAAW,IAAI,CAAC,CAAC;QAC5C,YAAY,IAAI,IAAI,CAAC,KAAK,EAAE,YAAY,IAAI,CAAC,CAAC;QAC9C,WAAW,IAAI,IAAI,CAAC,KAAK,EAAE,WAAW,IAAI,CAAC,CAAC;IAC9C,CAAC;IAED,OAAO;QACL,QAAQ,EAAE,KAAK,CAAC,MAAM;QACtB,WAAW;QACX,YAAY;QACZ,WAAW;KACZ,CAAC;AACJ,CAAC"}
+{"version":3,"file":"usage-limits.js","sourceRoot":"","sources":["../src/usage-limits.ts"],"names":[],"mappings":"AAyBA,MAAM,OAAO,kBAAmB,SAAQ,KAAK;IACzB,IAAI,GAAG,oBAAoB,CAAC;IACrC,SAAS,CAAiB;IAC1B,UAAU,CAAS;IACnB,YAAY,CAAS;IACrB,KAAK,CAAgB;IAE9B,YACE,SAAyB,EACzB,UAAkB,EAClB,YAAoB,EACpB,KAAoB;QAEpB,KAAK,CAAC,yBAAyB,SAAS,MAAM,YAAY,YAAY,UAAU,GAAG,CAAC,CAAC;QACrF,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;QACjC,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;CACF;AAED,MAAM,UAAU,cAAc,CAC5B,MAAmB;IAEnB,OAAO,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE;QACnB,MAAM,KAAK,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC;QAElC,IAAI,MAAM,CAAC,WAAW,KAAK,SAAS,IAAI,KAAK,CAAC,QAAQ,GAAG,MAAM,CAAC,WAAW,EAAE,CAAC;YAC5E,MAAM,IAAI,kBAAkB,CAAC,aAAa,EAAE,MAAM,CAAC,WAAW,EAAE,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QACzF,CAAC;QAED,IAAI,MAAM,CAAC,cAAc,KAAK,SAAS,IAAI,KAAK,CAAC,WAAW,GAAG,MAAM,CAAC,cAAc,EAAE,CAAC;YACrF,MAAM,IAAI,kBAAkB,CAC1B,gBAAgB,EAChB,MAAM,CAAC,cAAc,EACrB,KAAK,CAAC,WAAW,EACjB,KAAK,CACN,CAAC;QACJ,CAAC;QAED,IAAI,MAAM,CAAC,eAAe,KAAK,SAAS,IAAI,KAAK,CAAC,YAAY,GAAG,MAAM,CAAC,eAAe,EAAE,CAAC;YACxF,MAAM,IAAI,kBAAkB,CAC1B,iBAAiB,EACjB,MAAM,CAAC,eAAe,EACtB,KAAK,CAAC,YAAY,EAClB,KAAK,CACN,CAAC;QACJ,CAAC;QAED,IAAI,MAAM,CAAC,cAAc,KAAK,SAAS,IAAI,KAAK,CAAC,WAAW,GAAG,MAAM,CAAC,cAAc,EAAE,CAAC;YACrF,MAAM,IAAI,kBAAkB,CAC1B,gBAAgB,EAChB,MAAM,CAAC,cAAc,EACrB,KAAK,CAAC,WAAW,EACjB,KAAK,CACN,CAAC;QACJ,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAwB,KAA+B;IAC1E,IAAI,WAAW,GAAG,CAAC,CAAC;IACpB,IAAI,YAAY,GAAG,CAAC,CAAC;IACrB,IAAI,WAAW,GAAG,CAAC,CAAC;IAEpB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,WAAW,IAAI,IAAI,CAAC,KAAK,EAAE,WAAW,IAAI,CAAC,CAAC;QAC5C,YAAY,IAAI,IAAI,CAAC,KAAK,EAAE,YAAY,IAAI,CAAC,CAAC;QAC9C,WAAW,IAAI,IAAI,CAAC,KAAK,EAAE,WAAW,IAAI,CAAC,CAAC;IAC9C,CAAC;IAED,OAAO;QACL,QAAQ,EAAE,KAAK,CAAC,MAAM;QACtB,WAAW;QACX,YAAY;QACZ,WAAW;KACZ,CAAC;AACJ,CAAC"}

package/dist/workflow/durable-tool.d.ts

@@ -1,103 +1,19 @@
-/**
- * @agntk/core - Durable Tool Wrapper
- *
- * Wraps tools with durability directives for crash recovery.
- * Uses the Workflow DevKit's "use step" directive to checkpoint tool executions.
- * Each tool step gets a descriptive name for observability in the workflow inspector.
- *
- * @see https://useworkflow.dev
- */
 import type { Tool } from 'ai';
 type ToolSet = Record<string, Tool>;
 export interface DurabilityConfig {
-    /** Enable durability for this tool. Default: true */
     enabled?: boolean;
-    /** Whether this tool can run independently in parallel. Default: false */
     independent?: boolean;
-    /** Number of retry attempts on failure. Default: 3 */
     retryCount?: number;
-    /** Timeout duration string (e.g., "30s", "5m"). Default: "5m" */
     timeout?: string;
-    /** Custom step name for the workflow inspector. Default: "tool-exec-{toolName}" */
     stepName?: string;
 }
-/**
- * Wrap a single tool with durability directives.
- *
- * When wrapped, the tool's execute function uses the "use step" directive
- * to create a checkpoint. If the process crashes, the workflow runtime
- * resumes from the last checkpoint.
- *
- * Each wrapped tool gets a descriptive step name for the workflow inspector:
- * - Default: "tool-exec-{toolName}"
- * - Custom via `config.stepName`
- *
- * @param tool - The tool to wrap
- * @param config - Durability configuration
- * @param toolName - Name of the tool (used for step naming)
- * @returns Wrapped tool with durability enabled
- *
- * @example
- * ```typescript
- * const durableReadFile = wrapToolAsDurableStep(readFileTool, {
- *   retryCount: 3,
- *   timeout: '30s',
- * }, 'read_file');
- * ```
- */
 export declare function wrapToolAsDurableStep(tool: Tool, config?: DurabilityConfig, toolName?: string): Tool;
-/**
- * Wrap all tools in a set with durability directives.
- * Each tool gets a step name based on its key in the record.
- *
- * @param tools - Record of tools to wrap
- * @param config - Durability configuration applied to all tools
- * @returns Record of wrapped tools
- *
- * @example
- * ```typescript
- * const durableTools = wrapToolsAsDurable(myTools, {
- *   retryCount: 3,
- *   timeout: '5m',
- * });
- * ```
- */
 export declare function wrapToolsAsDurable(tools: ToolSet, config?: DurabilityConfig): ToolSet;
-/**
- * Wrap only specified tools with durability, leaving others unchanged.
- *
- * @param tools - All tools
- * @param toolNames - Names of tools to wrap
- * @param config - Durability configuration
- * @returns Tools with selective durability applied
- */
 export declare function wrapSelectedToolsAsDurable(tools: ToolSet, toolNames: string[], config?: DurabilityConfig): ToolSet;
-/**
- * Wrap a tool as an independent durable step.
- *
- * Independent steps can be executed in parallel with other independent steps.
- * The workflow runtime optimizes by running these concurrently.
- *
- * @param tool - The tool to wrap
- * @param toolName - Name of the tool (used for step naming)
- * @returns Wrapped tool marked as independent
- */
 export declare function wrapToolAsIndependentStep(tool: Tool, toolName?: string): Tool;
-/**
- * Symbol for storing durability metadata on tools.
- */
 export declare const DURABILITY_CONFIG: unique symbol;
-/**
- * Get durability configuration from a tool (if set).
- */
 export declare function getDurabilityConfig(tool: Tool): DurabilityConfig | undefined;
-/**
- * Set durability configuration on a tool.
- */
 export declare function setDurabilityConfig(tool: Tool, config: DurabilityConfig): Tool;
-/**
- * Get the step name assigned to a durable tool.
- */
 export declare function getStepName(tool: Tool): string | undefined;
 export {};
 //# sourceMappingURL=durable-tool.d.ts.map

package/dist/workflow/durable-tool.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"durable-tool.d.ts","sourceRoot":"","sources":["../../src/workflow/durable-tool.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,IAAI,CAAC;AAS/B,KAAK,OAAO,GAAG,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;AAEpC,MAAM,WAAW,gBAAgB;IAC/B,qDAAqD;IACrD,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,0EAA0E;IAC1E,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,sDAAsD;IACtD,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,iEAAiE;IACjE,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,mFAAmF;IACnF,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAMD;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,wBAAgB,qBAAqB,CACnC,IAAI,EAAE,IAAI,EACV,MAAM,GAAE,gBAAqB,EAC7B,QAAQ,CAAC,EAAE,MAAM,GAChB,IAAI,CAgDN;AAMD;;;;;;;;;;;;;;;GAeG;AACH,wBAAgB,kBAAkB,CAChC,KAAK,EAAE,OAAO,EACd,MAAM,GAAE,gBAAqB,GAC5B,OAAO,CAST;AAMD;;;;;;;GAOG;AACH,wBAAgB,0BAA0B,CACxC,KAAK,EAAE,OAAO,EACd,SAAS,EAAE,MAAM,EAAE,EACnB,MAAM,GAAE,gBAAqB,GAC5B,OAAO,CAaT;AAMD;;;;;;;;;GASG;AACH,wBAAgB,yBAAyB,CAAC,IAAI,EAAE,IAAI,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,IAAI,CAE7E;AAMD;;GAEG;AACH,eAAO,MAAM,iBAAiB,eAA6B,CAAC;AAE5D;;GAEG;AACH,wBAAgB,mBAAmB,CAAC,IAAI,EAAE,IAAI,GAAG,gBAAgB,GAAG,SAAS,CAG5E;AAED;;GAEG;AACH,wBAAgB,mBAAmB,CAAC,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,gBAAgB,GAAG,IAAI,CAI9E;AAED;;GAEG;AACH,wBAAgB,WAAW,CAAC,IAAI,EAAE,IAAI,GAAG,MAAM,GAAG,SAAS,CAE1D"}
+{"version":3,"file":"durable-tool.d.ts","sourceRoot":"","sources":["../../src/workflow/durable-tool.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,IAAI,CAAC;AAK/B,KAAK,OAAO,GAAG,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;AAEpC,MAAM,WAAW,gBAAgB;IAC/B,OAAO,CAAC,EAAE,OAAO,CAAC;IAElB,WAAW,CAAC,EAAE,OAAO,CAAC;IAEtB,UAAU,CAAC,EAAE,MAAM,CAAC;IAEpB,OAAO,CAAC,EAAE,MAAM,CAAC;IAEjB,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED,wBAAgB,qBAAqB,CACnC,IAAI,EAAE,IAAI,EACV,MAAM,GAAE,gBAAqB,EAC7B,QAAQ,CAAC,EAAE,MAAM,GAChB,IAAI,CA4CN;AAED,wBAAgB,kBAAkB,CAAC,KAAK,EAAE,OAAO,EAAE,MAAM,GAAE,gBAAqB,GAAG,OAAO,CAMzF;AAED,wBAAgB,0BAA0B,CACxC,KAAK,EAAE,OAAO,EACd,SAAS,EAAE,MAAM,EAAE,EACnB,MAAM,GAAE,gBAAqB,GAC5B,OAAO,CAaT;AAED,wBAAgB,yBAAyB,CAAC,IAAI,EAAE,IAAI,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,IAAI,CAE7E;AAED,eAAO,MAAM,iBAAiB,eAA6B,CAAC;AAE5D,wBAAgB,mBAAmB,CAAC,IAAI,EAAE,IAAI,GAAG,gBAAgB,GAAG,SAAS,CAG5E;AAED,wBAAgB,mBAAmB,CAAC,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,gBAAgB,GAAG,IAAI,CAI9E;AAED,wBAAgB,WAAW,CAAC,IAAI,EAAE,IAAI,GAAG,MAAM,GAAG,SAAS,CAE1D"}

package/dist/workflow/durable-tool.js

@@ -1,41 +1,5 @@
-/**
- * @agntk/core - Durable Tool Wrapper
- *
- * Wraps tools with durability directives for crash recovery.
- * Uses the Workflow DevKit's "use step" directive to checkpoint tool executions.
- * Each tool step gets a descriptive name for observability in the workflow inspector.
- *
- * @see https://useworkflow.dev
- */
 import { createLogger } from '@agntk/logger';
 const log = createLogger('@agntk/core:workflow:tool');
-// ============================================================================
-// Single Tool Wrapper
-// ============================================================================
-/**
- * Wrap a single tool with durability directives.
- *
- * When wrapped, the tool's execute function uses the "use step" directive
- * to create a checkpoint. If the process crashes, the workflow runtime
- * resumes from the last checkpoint.
- *
- * Each wrapped tool gets a descriptive step name for the workflow inspector:
- * - Default: "tool-exec-{toolName}"
- * - Custom via `config.stepName`
- *
- * @param tool - The tool to wrap
- * @param config - Durability configuration
- * @param toolName - Name of the tool (used for step naming)
- * @returns Wrapped tool with durability enabled
- *
- * @example
- * ```typescript
- * const durableReadFile = wrapToolAsDurableStep(readFileTool, {
- *   retryCount: 3,
- *   timeout: '30s',
- * }, 'read_file');
- * ```
- */
 export function wrapToolAsDurableStep(tool, config = {}, toolName) {
     const { enabled = true } = config;
     if (!enabled) {
@@ -51,10 +15,7 @@ export function wrapToolAsDurableStep(tool, config = {}, toolName) {
         ...tool,
         // eslint-disable-next-line @typescript-eslint/no-explicit-any
         execute: async (input, options) => {
-            // The "use step" directive tells the workflow runtime to checkpoint here.
-            // When the workflow resumes after a crash, it skips to this point
-            // and returns the cached result instead of re-executing.
-            "use step";
+            'use step';
             log.debug(`Executing durable step: ${stepName}`, {
                 stepName,
                 retryCount: config.retryCount,
@@ -69,52 +30,17 @@ export function wrapToolAsDurableStep(tool, config = {}, toolName) {
                 log.error(`Durable step failed: ${stepName}`, {
                     error: error instanceof Error ? error.message : String(error),
                 });
-                // Re-throw to let the workflow runtime handle retries
                 throw error;
             }
         },
     };
-    // Store step name and config as metadata
     setDurabilityConfig(wrappedTool, { ...config, stepName });
     return wrappedTool;
 }
-// ============================================================================
-// Batch Tool Wrapper
-// ============================================================================
-/**
- * Wrap all tools in a set with durability directives.
- * Each tool gets a step name based on its key in the record.
- *
- * @param tools - Record of tools to wrap
- * @param config - Durability configuration applied to all tools
- * @returns Record of wrapped tools
- *
- * @example
- * ```typescript
- * const durableTools = wrapToolsAsDurable(myTools, {
- *   retryCount: 3,
- *   timeout: '5m',
- * });
- * ```
- */
 export function wrapToolsAsDurable(tools, config = {}) {
     log.debug('Wrapping all tools as durable', { count: Object.keys(tools).length });
-    return Object.fromEntries(Object.entries(tools).map(([name, tool]) => [
-        name,
-        wrapToolAsDurableStep(tool, config, name),
-    ]));
+    return Object.fromEntries(Object.entries(tools).map(([name, tool]) => [name, wrapToolAsDurableStep(tool, config, name)]));
 }
-// ============================================================================
-// Selective Wrapping
-// ============================================================================
-/**
- * Wrap only specified tools with durability, leaving others unchanged.
- *
- * @param tools - All tools
- * @param toolNames - Names of tools to wrap
- * @param config - Durability configuration
- * @returns Tools with selective durability applied
- */
 export function wrapSelectedToolsAsDurable(tools, toolNames, config = {}) {
     const nameSet = new Set(toolNames);
     log.debug('Selectively wrapping tools as durable', {
@@ -126,47 +52,19 @@ export function wrapSelectedToolsAsDurable(tools, toolNames, config = {}) {
         nameSet.has(name) ? wrapToolAsDurableStep(tool, config, name) : tool,
     ]));
 }
-// ============================================================================
-// Independent Step Wrapper
-// ============================================================================
-/**
- * Wrap a tool as an independent durable step.
- *
- * Independent steps can be executed in parallel with other independent steps.
- * The workflow runtime optimizes by running these concurrently.
- *
- * @param tool - The tool to wrap
- * @param toolName - Name of the tool (used for step naming)
- * @returns Wrapped tool marked as independent
- */
 export function wrapToolAsIndependentStep(tool, toolName) {
     return wrapToolAsDurableStep(tool, { independent: true }, toolName);
 }
-// ============================================================================
-// Durability Metadata
-// ============================================================================
-/**
- * Symbol for storing durability metadata on tools.
- */
 export const DURABILITY_CONFIG = Symbol('durabilityConfig');
-/**
- * Get durability configuration from a tool (if set).
- */
 export function getDurabilityConfig(tool) {
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     return tool[DURABILITY_CONFIG];
 }
-/**
- * Set durability configuration on a tool.
- */
 export function setDurabilityConfig(tool, config) {
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     tool[DURABILITY_CONFIG] = config;
     return tool;
 }
-/**
- * Get the step name assigned to a durable tool.
- */
 export function getStepName(tool) {
     return getDurabilityConfig(tool)?.stepName;
 }

package/dist/workflow/durable-tool.js.map

@@ -1 +1 @@
-{"version":3,"file":"durable-tool.js","sourceRoot":"","sources":["../../src/workflow/durable-tool.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAGH,OAAO,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAE7C,MAAM,GAAG,GAAG,YAAY,CAAC,2BAA2B,CAAC,CAAC;AAqBtD,+EAA+E;AAC/E,sBAAsB;AACtB,+EAA+E;AAE/E;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,MAAM,UAAU,qBAAqB,CACnC,IAAU,EACV,SAA2B,EAAE,EAC7B,QAAiB;IAEjB,MAAM,EAAE,OAAO,GAAG,IAAI,EAAE,GAAG,MAAM,CAAC;IAElC,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,IAAI,CAAC;IACd,CAAC;IAED,8DAA8D;IAC9D,MAAM,eAAe,GAAI,IAAY,CAAC,OAAO,CAAC;IAC9C,IAAI,CAAC,eAAe,EAAE,CAAC;QACrB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,aAAa,QAAQ,EAAE,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC;IAEvF,MAAM,WAAW,GAAG;QAClB,GAAG,IAAI;QACP,8DAA8D;QAC9D,OAAO,EAAE,KAAK,EAAE,KAAU,EAAE,OAAY,EAAE,EAAE;YAC1C,0EAA0E;YAC1E,kEAAkE;YAClE,yDAAyD;YACzD,UAAU,CAAC;YAEX,GAAG,CAAC,KAAK,CAAC,2BAA2B,QAAQ,EAAE,EAAE;gBAC/C,QAAQ;gBACR,UAAU,EAAE,MAAM,CAAC,UAAU;gBAC7B,OAAO,EAAE,MAAM,CAAC,OAAO;aACxB,CAAC,CAAC;YAEH,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,eAAe,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;gBACrD,GAAG,CAAC,KAAK,CAAC,2BAA2B,QAAQ,EAAE,CAAC,CAAC;gBACjD,OAAO,MAAM,CAAC;YAChB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,GAAG,CAAC,KAAK,CAAC,wBAAwB,QAAQ,EAAE,EAAE;oBAC5C,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;iBAC9D,CAAC,CAAC;gBACH,sDAAsD;gBACtD,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC;KACM,CAAC;IAEV,yCAAyC;IACzC,mBAAmB,CAAC,WAAW,EAAE,EAAE,GAAG,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAC;IAE1D,OAAO,WAAW,CAAC;AACrB,CAAC;AAED,+EAA+E;AAC/E,qBAAqB;AACrB,+EAA+E;AAE/E;;;;;;;;;;;;;;;GAeG;AACH,MAAM,UAAU,kBAAkB,CAChC,KAAc,EACd,SAA2B,EAAE;IAE7B,GAAG,CAAC,KAAK,CAAC,+BAA+B,EAAE,EAAE,KAAK,EAAE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;IAEjF,OAAO,MAAM,CAAC,WAAW,CACvB,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC;QAC1C,IAAI;QACJ,qBAAqB,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC;KAC1C,CAAC,CACH,CAAC;AACJ,CAAC;AAED,+EAA+E;AAC/E,qBAAqB;AACrB,+EAA+E;AAE/E;;;;;;;GAOG;AACH,MAAM,UAAU,0BAA0B,CACxC,KAAc,EACd,SAAmB,EACnB,SAA2B,EAAE;IAE7B,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,SAAS,CAAC,CAAC;IACnC,GAAG,CAAC,KAAK,CAAC,uCAAuC,EAAE;QACjD,KAAK,EAAE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM;QAChC,QAAQ,EAAE,SAAS;KACpB,CAAC,CAAC;IAEH,OAAO,MAAM,CAAC,WAAW,CACvB,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC;QAC1C,IAAI;QACJ,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,qBAAqB,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI;KACrE,CAAC,CACH,CAAC;AACJ,CAAC;AAED,+EAA+E;AAC/E,2BAA2B;AAC3B,+EAA+E;AAE/E;;;;;;;;;GASG;AACH,MAAM,UAAU,yBAAyB,CAAC,IAAU,EAAE,QAAiB;IACrE,OAAO,qBAAqB,CAAC,IAAI,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE,EAAE,QAAQ,CAAC,CAAC;AACtE,CAAC;AAED,+EAA+E;AAC/E,sBAAsB;AACtB,+EAA+E;AAE/E;;GAEG;AACH,MAAM,CAAC,MAAM,iBAAiB,GAAG,MAAM,CAAC,kBAAkB,CAAC,CAAC;AAE5D;;GAEG;AACH,MAAM,UAAU,mBAAmB,CAAC,IAAU;IAC5C,8DAA8D;IAC9D,OAAQ,IAAY,CAAC,iBAAiB,CAAC,CAAC;AAC1C,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,mBAAmB,CAAC,IAAU,EAAE,MAAwB;IACtE,8DAA8D;IAC7D,IAAY,CAAC,iBAAiB,CAAC,GAAG,MAAM,CAAC;IAC1C,OAAO,IAAI,CAAC;AACd,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,WAAW,CAAC,IAAU;IACpC,OAAO,mBAAmB,CAAC,IAAI,CAAC,EAAE,QAAQ,CAAC;AAC7C,CAAC"}
+{"version":3,"file":"durable-tool.js","sourceRoot":"","sources":["../../src/workflow/durable-tool.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAE7C,MAAM,GAAG,GAAG,YAAY,CAAC,2BAA2B,CAAC,CAAC;AAgBtD,MAAM,UAAU,qBAAqB,CACnC,IAAU,EACV,SAA2B,EAAE,EAC7B,QAAiB;IAEjB,MAAM,EAAE,OAAO,GAAG,IAAI,EAAE,GAAG,MAAM,CAAC;IAElC,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,IAAI,CAAC;IACd,CAAC;IAED,8DAA8D;IAC9D,MAAM,eAAe,GAAI,IAAY,CAAC,OAAO,CAAC;IAC9C,IAAI,CAAC,eAAe,EAAE,CAAC;QACrB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,aAAa,QAAQ,EAAE,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC;IAEvF,MAAM,WAAW,GAAG;QAClB,GAAG,IAAI;QACP,8DAA8D;QAC9D,OAAO,EAAE,KAAK,EAAE,KAAU,EAAE,OAAY,EAAE,EAAE;YAC1C,UAAU,CAAC;YAEX,GAAG,CAAC,KAAK,CAAC,2BAA2B,QAAQ,EAAE,EAAE;gBAC/C,QAAQ;gBACR,UAAU,EAAE,MAAM,CAAC,UAAU;gBAC7B,OAAO,EAAE,MAAM,CAAC,OAAO;aACxB,CAAC,CAAC;YAEH,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,eAAe,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;gBACrD,GAAG,CAAC,KAAK,CAAC,2BAA2B,QAAQ,EAAE,CAAC,CAAC;gBACjD,OAAO,MAAM,CAAC;YAChB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,GAAG,CAAC,KAAK,CAAC,wBAAwB,QAAQ,EAAE,EAAE;oBAC5C,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;iBAC9D,CAAC,CAAC;gBAEH,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC;KACM,CAAC;IAEV,mBAAmB,CAAC,WAAW,EAAE,EAAE,GAAG,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAC;IAE1D,OAAO,WAAW,CAAC;AACrB,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,KAAc,EAAE,SAA2B,EAAE;IAC9E,GAAG,CAAC,KAAK,CAAC,+BAA+B,EAAE,EAAE,KAAK,EAAE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;IAEjF,OAAO,MAAM,CAAC,WAAW,CACvB,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,qBAAqB,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC,CAC/F,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,0BAA0B,CACxC,KAAc,EACd,SAAmB,EACnB,SAA2B,EAAE;IAE7B,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,SAAS,CAAC,CAAC;IACnC,GAAG,CAAC,KAAK,CAAC,uCAAuC,EAAE;QACjD,KAAK,EAAE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM;QAChC,QAAQ,EAAE,SAAS;KACpB,CAAC,CAAC;IAEH,OAAO,MAAM,CAAC,WAAW,CACvB,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC;QAC1C,IAAI;QACJ,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,qBAAqB,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI;KACrE,CAAC,CACH,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,yBAAyB,CAAC,IAAU,EAAE,QAAiB;IACrE,OAAO,qBAAqB,CAAC,IAAI,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE,EAAE,QAAQ,CAAC,CAAC;AACtE,CAAC;AAED,MAAM,CAAC,MAAM,iBAAiB,GAAG,MAAM,CAAC,kBAAkB,CAAC,CAAC;AAE5D,MAAM,UAAU,mBAAmB,CAAC,IAAU;IAC5C,8DAA8D;IAC9D,OAAQ,IAAY,CAAC,iBAAiB,CAAC,CAAC;AAC1C,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,IAAU,EAAE,MAAwB;IACtE,8DAA8D;IAC7D,IAAY,CAAC,iBAAiB,CAAC,GAAG,MAAM,CAAC;IAC1C,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,UAAU,WAAW,CAAC,IAAU;IACpC,OAAO,mBAAmB,CAAC,IAAI,CAAC,EAAE,QAAQ,CAAC;AAC7C,CAAC"}