@polka-codes/core 0.9.69 → 0.9.71

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,21 @@ declare const ruleSchema: z.ZodUnion<readonly [z.ZodString, z.ZodObject<{
 export { ruleSchema }
 export { ruleSchema as ruleSchema_alias_1 }
 
+declare interface StepFn {
+    <T>(name: string, fn: () => Promise<T>): Promise<T>;
+    <T>(name: string, options: StepOptions, fn: () => Promise<T>): 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 +1532,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

@@ -1753,7 +1753,7 @@ var agentWorkflow = async (input, { step, tools, logger }) => {
   for (let i = 0; i < maxToolRoundTrips; i++) {
     messages.push(...nextMessage);
     await event(`start-round-${i}`, { kind: "StartRequest" /* StartRequest */, userMessage: nextMessage });
-    const assistantMessage = await step(`agent-round-${i}`, async () => {
+    const assistantMessage = await step(`agent-round-${i}`, { retry: 2 }, async () => {
       return await tools.generateText({
         messages,
         tools: toolSet,
@@ -2050,23 +2050,44 @@ var silentLogger = {
 };
 function createContext(tools, stepFn, logger = silentLogger) {
   if (!stepFn) {
-    stepFn = async (_name, fn) => fn();
+    stepFn = async (_name, arg2, arg3) => {
+      const fn = typeof arg2 === "function" ? arg2 : arg3;
+      return fn();
+    };
   }
   return { step: stepFn, logger, tools };
 }
 var makeStepFn = () => {
   const results = /* @__PURE__ */ new Map();
   const callStack = [];
-  return async (name, fn) => {
+  return async (name, arg2, arg3) => {
+    let fn;
+    let options;
+    if (typeof arg2 === "function") {
+      fn = arg2;
+      options = arg3;
+    } else {
+      options = arg2;
+      fn = arg3;
+    }
     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 ?? 0;
+      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.71",
   "license": "AGPL-3.0",
   "author": "github@polka.codes",
   "type": "module",