@polka-codes/core 0.9.69 → 0.9.70

package/dist/_tsup-dts-rollup.d.ts

@@ -153,7 +153,7 @@ declare const configSchema: z.ZodOptional<z.ZodNullable<z.ZodObject<{
 export { configSchema }
 export { configSchema as configSchema_alias_1 }
 
-declare function createContext<TTools extends ToolRegistry>(tools: WorkflowTools<TTools>, stepFn?: <T>(name: string, fn: () => Promise<T>) => Promise<T>, logger?: Logger): WorkflowContext<TTools>;
+declare function createContext<TTools extends ToolRegistry>(tools: WorkflowTools<TTools>, stepFn?: StepFn, logger?: Logger): WorkflowContext<TTools>;
 export { createContext }
 export { createContext as createContext_alias_1 }
 export { createContext as createContext_alias_2 }
@@ -821,7 +821,7 @@ export { Logger }
 export { Logger as Logger_alias_1 }
 export { Logger as Logger_alias_2 }
 
-declare const makeStepFn: () => (<T>(name: string, fn: () => Promise<T>) => Promise<T>);
+declare const makeStepFn: () => StepFn;
 export { makeStepFn }
 export { makeStepFn as makeStepFn_alias_1 }
 export { makeStepFn as makeStepFn_alias_2 }
@@ -941,6 +941,18 @@ declare const ruleSchema: z.ZodUnion<readonly [z.ZodString, z.ZodObject<{
 export { ruleSchema }
 export { ruleSchema as ruleSchema_alias_1 }
 
+declare type StepFn = <T>(name: string, fn: () => Promise<T>, options?: StepOptions) => Promise<T>;
+export { StepFn }
+export { StepFn as StepFn_alias_1 }
+export { StepFn as StepFn_alias_2 }
+
+declare type StepOptions = {
+    retry?: number;
+};
+export { StepOptions }
+export { StepOptions as StepOptions_alias_1 }
+export { StepOptions as StepOptions_alias_2 }
+
 /**
  * Union type of all possible task events
  */
@@ -1517,7 +1529,7 @@ export { WebProvider as WebProvider_alias_1 }
 export { WebProvider as WebProvider_alias_2 }
 
 declare type WorkflowContext<TTools extends ToolRegistry> = {
-    step: <T>(name: string, fn: () => Promise<T>) => Promise<T>;
+    step: StepFn;
     logger: Logger;
     tools: WorkflowTools<TTools>;
 };

package/dist/index.d.ts

@@ -87,6 +87,8 @@ export { Logger } from './_tsup-dts-rollup.js';
 export { ToolSignature } from './_tsup-dts-rollup.js';
 export { ToolRegistry } from './_tsup-dts-rollup.js';
 export { WorkflowTools } from './_tsup-dts-rollup.js';
+export { StepOptions } from './_tsup-dts-rollup.js';
+export { StepFn } from './_tsup-dts-rollup.js';
 export { WorkflowContext } from './_tsup-dts-rollup.js';
 export { WorkflowFn } from './_tsup-dts-rollup.js';
 export { makeStepFn } from './_tsup-dts-rollup.js';

package/dist/index.js

@@ -2050,23 +2050,32 @@ var silentLogger = {
 };
 function createContext(tools, stepFn, logger = silentLogger) {
   if (!stepFn) {
-    stepFn = async (_name, fn) => fn();
+    stepFn = async (_name, fn, _options) => fn();
   }
   return { step: stepFn, logger, tools };
 }
 var makeStepFn = () => {
   const results = /* @__PURE__ */ new Map();
   const callStack = [];
-  return async (name, fn) => {
+  return async (name, fn, options) => {
     callStack.push(name);
     const key = callStack.join(">");
     try {
       if (results.has(key)) {
         return results.get(key);
       }
-      const result = await fn();
-      results.set(key, result);
-      return result;
+      const maxRetryCount = options?.retry ?? 1;
+      let lastError;
+      for (let retryCount = 0; retryCount <= maxRetryCount; retryCount++) {
+        try {
+          const result = await fn();
+          results.set(key, result);
+          return result;
+        } catch (error) {
+          lastError = error;
+        }
+      }
+      throw lastError;
     } finally {
       callStack.pop();
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@polka-codes/core",
-  "version": "0.9.69",
+  "version": "0.9.70",
   "license": "AGPL-3.0",
   "author": "github@polka.codes",
   "type": "module",