@polka-codes/core 0.9.68 → 0.9.70

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

@@ -88,7 +88,7 @@ declare function computeRateLimitBackoffSeconds(count: number, baseSeconds?: num
 export { computeRateLimitBackoffSeconds }
 export { computeRateLimitBackoffSeconds as computeRateLimitBackoffSeconds_alias_1 }
 
-declare type Config = z.infer<typeof configSchema>;
+declare type Config = NonNullable<z.infer<typeof configSchema>>;
 export { Config }
 export { Config as Config_alias_1 }
 
@@ -96,7 +96,7 @@ declare type ConfigRule = z.infer<typeof ruleSchema>;
 export { ConfigRule }
 export { ConfigRule as ConfigRule_alias_1 }
 
-declare const configSchema: z.ZodObject<{
+declare const configSchema: z.ZodOptional<z.ZodNullable<z.ZodObject<{
     prices: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodRecord<z.ZodString, z.ZodObject<{
         inputPrice: z.ZodOptional<z.ZodNumber>;
         outputPrice: z.ZodOptional<z.ZodNumber>;
@@ -149,11 +149,11 @@ declare const configSchema: z.ZodObject<{
         branch: z.ZodOptional<z.ZodString>;
     }, z.core.$strict>]>>>, z.ZodString]>>;
     excludeFiles: z.ZodOptional<z.ZodArray<z.ZodString>>;
-}, z.core.$strict>;
+}, z.core.$strict>>>;
 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
  */
@@ -1387,7 +1399,7 @@ export { ToolResponseResult as ToolResponseResult_alias_1 }
 
 declare type ToolResponseResultMedia = {
     type: 'media';
-    base64Data: string;
+    data: string;
     mediaType: string;
     url: string;
 };
@@ -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

@@ -194,7 +194,7 @@ var configSchema = z.object({
   }).optional(),
   rules: z.array(ruleSchema).optional().or(z.string()).optional(),
   excludeFiles: z.array(z.string()).optional()
-}).strict();
+}).strict().nullish();
 
 // src/tool.ts
 var ToolResponseType = /* @__PURE__ */ ((ToolResponseType2) => {
@@ -761,7 +761,7 @@ var handler8 = async (provider, args) => {
           {
             type: "media",
             url,
-            base64Data: filePart.base64Data,
+            data: filePart.base64Data,
             mediaType: filePart.mediaType
           }
         ]
@@ -1823,11 +1823,6 @@ var agentWorkflow = async (input, { step, tools, logger }) => {
             value: `Tool '${toolCall.toolName}' not found.`
           }
         });
-        toolResults.push({
-          toolCallId: toolCall.toolCallId,
-          toolName: toolCall.toolName,
-          output: `Error: Tool '${toolCall.toolName}' not found.`
-        });
         continue;
       }
       await event(`event-tool-use-${toolCall.toolName}-${toolCall.toolCallId}`, {
@@ -2055,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.68",
+  "version": "0.9.70",
   "license": "AGPL-3.0",
   "author": "github@polka.codes",
   "type": "module",
@@ -19,8 +19,8 @@
   "dependencies": {
     "@ai-sdk/provider": "^2.0.0",
     "@ai-sdk/provider-utils": "^3.0.18",
-    "ai": "^5.0.104",
-    "yaml": "^2.8.1",
+    "ai": "^5.0.106",
+    "yaml": "^2.8.2",
     "zod": "^4.1.13"
   }
 }