@polka-codes/core 0.9.70 → 0.9.71

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

@@ -941,7 +941,10 @@ 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>;
+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 }

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,21 +2050,33 @@ var silentLogger = {
 };
 function createContext(tools, stepFn, logger = silentLogger) {
   if (!stepFn) {
-    stepFn = async (_name, fn, _options) => 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, options) => {
+  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 maxRetryCount = options?.retry ?? 1;
+      const maxRetryCount = options?.retry ?? 0;
       let lastError;
       for (let retryCount = 0; retryCount <= maxRetryCount; retryCount++) {
         try {

package/package.json

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