@polka-codes/core 0.9.28 → 0.9.30

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

@@ -343,6 +343,14 @@ export { commandStepSpecHandler as commandStepSpecHandler_alias_1 }
 export { commandStepSpecHandler as commandStepSpecHandler_alias_2 }
 export { commandStepSpecHandler as commandStepSpecHandler_alias_3 }
 
+/**
+ * Utility to compute exponential backoff delays for rate-limit handling. generated by polka.codes
+ *
+ * The backoff starts at baseSeconds and doubles each time, capped at capSeconds.
+ * Example with base=2, cap=60: 2, 4, 8, 16, 32, 60, 60, ...
+ */
+export declare function computeRateLimitBackoffSeconds(count: number, baseSeconds?: number, capSeconds?: number): number;
+
 declare type Config = z.infer<typeof configSchema>;
 export { Config }
 export { Config as Config_alias_1 }
@@ -1985,6 +1993,7 @@ declare type WorkflowContext = {
     parameters: Record<string, any>;
     verbose?: number;
     agentCallback?: TaskEventCallback;
+    logger?: Console;
 };
 export { WorkflowContext }
 export { WorkflowContext as WorkflowContext_alias_1 }

package/dist/index.js

@@ -1433,6 +1433,21 @@ function toToolInfoV1(tool) {
   return v1Tool;
 }
 
+// src/Agent/backoff.ts
+function computeRateLimitBackoffSeconds(count, baseSeconds = 2, capSeconds = 60) {
+  if (!Number.isFinite(count) || count <= 0) {
+    count = 1;
+  }
+  if (!Number.isFinite(baseSeconds) || baseSeconds <= 0) {
+    baseSeconds = 2;
+  }
+  if (!Number.isFinite(capSeconds) || capSeconds <= 0) {
+    capSeconds = 60;
+  }
+  const delay = baseSeconds * 2 ** (count - 1);
+  return Math.min(delay, capSeconds);
+}
+
 // src/Agent/parseAssistantMessage.ts
 function parseNestedParameters(content, parameterPrefix, childrenParams) {
   const result = {};
@@ -2011,6 +2026,7 @@ ${instance.prompt}`;
     const retryCount = this.config.retryCount ?? 5;
     const requestTimeoutSeconds = this.config.requestTimeoutSeconds ?? 90;
     let respMessages = [];
+    let rateLimitErrorCount = 0;
     for (let i = 0; i < retryCount; i++) {
       if (this.#aborted) {
         break;
@@ -2026,7 +2042,7 @@ ${instance.prompt}`;
         }
         if (requestTimeoutSeconds > 0 && requestAbortController) {
           timeout = setTimeout(() => {
-            console.debug(
+            console.error(
               `
 Request timeout after ${requestTimeoutSeconds} seconds. Canceling current request attempt ${i + 1}/${retryCount}.`
             );
@@ -2071,6 +2087,7 @@ Request timeout after ${requestTimeoutSeconds} seconds. Canceling current reques
         });
         const resp = await stream.response;
         respMessages = resp.messages;
+        rateLimitErrorCount = 0;
         if (timeout) {
           clearTimeout(timeout);
           timeout = void 0;
@@ -2080,8 +2097,19 @@ Request timeout after ${requestTimeoutSeconds} seconds. Canceling current reques
           if (this.#aborted) {
             break;
           }
-          console.debug(`Request attempt ${i + 1} timed out, will retry`);
+          console.error(`Request attempt ${i + 1} timed out, will retry`);
+        } else if (error?.error?.error?.code === "rate_limit_exceeded" || error?.error?.code === "rate_limit_exceeded" || error?.code === "rate_limit_exceeded" || error?.status === 429 || error?.error?.status === 429) {
+          rateLimitErrorCount++;
+          const waitSeconds = computeRateLimitBackoffSeconds(rateLimitErrorCount);
+          console.error(`Rate limit exceeded. Waiting ${waitSeconds}s before retrying...`);
+          if (timeout) {
+            clearTimeout(timeout);
+            timeout = void 0;
+          }
+          await new Promise((resolve) => setTimeout(resolve, waitSeconds * 1e3));
+          console.error("Retrying request...");
         } else {
+          rateLimitErrorCount = 0;
           console.error("Error in stream:", error);
         }
       } finally {
@@ -2110,7 +2138,7 @@ Request timeout after ${requestTimeoutSeconds} seconds. Canceling current reques
         break;
       }
       if (i < retryCount - 1) {
-        console.debug(`
+        console.error(`
 Retrying request ${i + 2} of ${retryCount}`);
       }
     }
@@ -3760,8 +3788,9 @@ var makeAgentStepSpecHandler = (getModelFn) => ({
     return {
       ...step2,
       async run(input, context, resumedState) {
+        const logger = context.logger ?? console;
         if (context.verbose && context.verbose >= 1) {
-          console.log(`[agent-step] Running agent step '${step2.id}' with input:`, input);
+          logger.log(`[agent-step] Running agent step '${step2.id}' with input:`, input);
         }
         try {
           const model = await getModelFn(step2, context);
@@ -3783,9 +3812,9 @@ var makeAgentStepSpecHandler = (getModelFn) => ({
                 throw new Error(`Unknown agent: ${agentName}`);
               }
               if (context.verbose && context.verbose >= 1) {
-                console.log(`[agent-step] Using agent: ${agentName}`);
+                logger.log(`[agent-step] Using agent: ${agentName}`);
               }
-              const agentOptions = {
+              return new AgentClass({
                 ai: model,
                 os: parameters.os ?? "linux",
                 provider: context.provider,
@@ -3795,15 +3824,15 @@ var makeAgentStepSpecHandler = (getModelFn) => ({
                 usageMeter,
                 parameters: modelParameters,
                 scripts: parameters.scripts,
-                callback: context.agentCallback
-              };
-              return new AgentClass(agentOptions);
+                callback: context.agentCallback,
+                requireToolUse: false
+              });
             } else {
               if (!step2.systemPrompt) {
                 throw new Error("No system prompt specified for the agent step.");
               }
               if (context.verbose && context.verbose >= 1) {
-                console.log(`[agent-step] Using generic WorkflowAgent`);
+                logger.log(`[agent-step] Using generic WorkflowAgent`);
               }
               const systemPrompt = resolveTemplatedString(step2.systemPrompt, input);
               return new WorkflowAgent("agent", model, {
@@ -3818,7 +3847,7 @@ var makeAgentStepSpecHandler = (getModelFn) => ({
                 toolFormat,
                 parameters: modelParameters,
                 usageMeter,
-                requireToolUse: true
+                requireToolUse: false
               });
             }
           };
@@ -3841,11 +3870,11 @@ var makeAgentStepSpecHandler = (getModelFn) => ({
             }
           }
           if (context.verbose && context.verbose >= 1) {
-            console.log(`[agent-step] Starting agent with content:`, JSON.stringify(combinedContentParts, null, 2));
+            logger.log(`[agent-step] Starting agent with content:`, JSON.stringify(combinedContentParts, null, 2));
           }
           const exitReason = await agent.start(combinedContentParts);
           if (context.verbose && context.verbose >= 1) {
-            console.log(`[agent-step] Agent exited with reason:`, exitReason);
+            logger.log(`[agent-step] Agent exited with reason:`, exitReason);
           }
           const handleExitReason = (reason) => {
             switch (reason.type) {
@@ -4012,15 +4041,16 @@ var command = (id, ...command2) => ({
 
 // src/workflow/runStep.ts
 var runStep = async (step2, input, context, resumedState, allOutputs) => {
+  const logger = context.logger ?? console;
   if (context.verbose && context.verbose >= 1) {
-    console.log(`[workflow] running step: ${step2.id}`);
-    console.log(`[workflow] input: ${JSON.stringify(input, null, 2)}`);
+    logger.log(`[workflow] running step: ${step2.id}`);
+    logger.log(`[workflow] input: ${JSON.stringify(input, null, 2)}`);
   }
   try {
     const validatedInput = step2.inputSchema?.parse(input) ?? input;
     const result = await step2.run({ ...validatedInput, $: allOutputs }, context, resumedState);
     if (context.verbose && context.verbose >= 1) {
-      console.log(`[workflow] step result: ${step2.id}`, JSON.stringify(result, null, 2));
+      logger.log(`[workflow] step result: ${step2.id}`, JSON.stringify(result, null, 2));
     }
     if (result.type === "success") {
       const validatedOutput = step2.outputSchema?.parse(result.output) ?? result.output;
@@ -4032,7 +4062,7 @@ var runStep = async (step2, input, context, resumedState, allOutputs) => {
     return result;
   } catch (error) {
     if (context.verbose && context.verbose >= 1) {
-      console.error(`[workflow] step error: ${step2.id}`, error);
+      logger.error(`[workflow] step error: ${step2.id}`, error);
     }
     return { type: "error", error };
   }

package/package.json

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