@polka-codes/core 0.9.61 → 0.9.63

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

@@ -92,6 +92,10 @@ declare type Config = z.infer<typeof configSchema>;
 export { Config }
 export { Config as Config_alias_1 }
 
+declare type ConfigRule = z.infer<typeof ruleSchema>;
+export { ConfigRule }
+export { ConfigRule as ConfigRule_alias_1 }
+
 declare const configSchema: z.ZodObject<{
     prices: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodRecord<z.ZodString, z.ZodObject<{
         inputPrice: z.ZodOptional<z.ZodNumber>;
@@ -125,7 +129,25 @@ declare const configSchema: z.ZodObject<{
         parameters: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>;
         budget: z.ZodOptional<z.ZodNumber>;
     }, z.core.$strip>>>;
-    rules: z.ZodOptional<z.ZodUnion<[z.ZodOptional<z.ZodArray<z.ZodString>>, z.ZodString]>>;
+    tools: z.ZodOptional<z.ZodObject<{
+        search: z.ZodOptional<z.ZodUnion<[z.ZodObject<{
+            provider: z.ZodOptional<z.ZodString>;
+            model: z.ZodOptional<z.ZodString>;
+            parameters: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>;
+            budget: z.ZodOptional<z.ZodNumber>;
+        }, z.core.$strip>, z.ZodBoolean]>>;
+    }, z.core.$strip>>;
+    rules: z.ZodOptional<z.ZodUnion<[z.ZodOptional<z.ZodArray<z.ZodUnion<readonly [z.ZodString, z.ZodObject<{
+        path: z.ZodString;
+    }, z.core.$strict>, z.ZodObject<{
+        url: z.ZodString;
+    }, z.core.$strict>, z.ZodObject<{
+        repo: z.ZodString;
+        path: z.ZodString;
+        tag: z.ZodOptional<z.ZodString>;
+        commit: z.ZodOptional<z.ZodString>;
+        branch: z.ZodOptional<z.ZodString>;
+    }, z.core.$strict>]>>>, z.ZodString]>>;
     excludeFiles: z.ZodOptional<z.ZodArray<z.ZodString>>;
 }, z.core.$strict>;
 export { configSchema }
@@ -237,6 +259,24 @@ export { _default_13 as replaceInFile }
 export { _default_13 as replaceInFile_alias_1 }
 
 declare const _default_14: {
+    handler: ToolHandler<{
+        readonly name: "search";
+        readonly description: "Search the web for information using Google Search. Use this tool to find current information, facts, news, documentation, or research that is not available in your training data. Returns comprehensive search results with relevant content extracted from the web.";
+        readonly parameters: z.ZodObject<{
+            query: z.ZodString;
+        }, z.core.$strip>;
+    }, WebProvider>;
+    name: "search";
+    description: "Search the web for information using Google Search. Use this tool to find current information, facts, news, documentation, or research that is not available in your training data. Returns comprehensive search results with relevant content extracted from the web.";
+    parameters: z.ZodObject<{
+        query: z.ZodString;
+    }, z.core.$strip>;
+};
+export { _default_14 as default_alias_13 }
+export { _default_14 as search }
+export { _default_14 as search_alias_1 }
+
+declare const _default_15: {
     handler: ToolHandler<{
         readonly name: "searchFiles";
         readonly description: "Request to perform a regex search across files in a specified directory, outputting context-rich results that include surrounding lines. This tool searches for patterns or specific content across multiple files, displaying each match with encapsulating context.";
@@ -254,11 +294,11 @@ declare const _default_14: {
         filePattern: z.ZodOptional<z.ZodString>;
     }, z.core.$strip>;
 };
-export { _default_14 as default_alias_13 }
-export { _default_14 as searchFiles }
-export { _default_14 as searchFiles_alias_1 }
+export { _default_15 as default_alias_14 }
+export { _default_15 as searchFiles }
+export { _default_15 as searchFiles_alias_1 }
 
-declare const _default_15: {
+declare const _default_16: {
     handler: ToolHandler<{
         readonly name: "updateMemory";
         readonly description: "Appends, replaces, or removes content from a memory topic. Use \"append\" to add to existing content, \"replace\" to overwrite entirely, or \"remove\" to delete a topic. Memory persists across tool calls within a workflow.";
@@ -284,11 +324,11 @@ declare const _default_15: {
         content: z.ZodOptional<z.ZodNullable<z.ZodString>>;
     }, z.core.$strip>;
 };
-export { _default_15 as default_alias_14 }
-export { _default_15 as updateMemory }
-export { _default_15 as updateMemory_alias_1 }
+export { _default_16 as default_alias_15 }
+export { _default_16 as updateMemory }
+export { _default_16 as updateMemory_alias_1 }
 
-declare const _default_16: {
+declare const _default_17: {
     handler: ToolHandler<{
         readonly name: "updateTodoItem";
         readonly description: "Add or update a to-do item.";
@@ -326,11 +366,11 @@ declare const _default_16: {
     }>>>;
     }, $strip>;
 };
-export { _default_16 as default_alias_15 }
-export { _default_16 as updateTodoItem }
-export { _default_16 as updateTodoItem_alias_1 }
+export { _default_17 as default_alias_16 }
+export { _default_17 as updateTodoItem }
+export { _default_17 as updateTodoItem_alias_1 }
 
-declare const _default_17: {
+declare const _default_18: {
     handler: ToolHandler<{
         readonly name: "writeToFile";
         readonly description: "Request to write content to a file at the specified path. If the file exists, it will be overwritten with the provided content. If the file doesn't exist, it will be created. This tool will automatically create any directories needed to write the file. Ensure that the output content does not include incorrect escaped character patterns such as `&lt;`, `&gt;`, or `&amp;`. Also ensure there is no unwanted CDATA tags in the content.";
@@ -346,9 +386,9 @@ declare const _default_17: {
         content: z.ZodString;
     }, z.core.$strip>;
 };
-export { _default_17 as default_alias_16 }
-export { _default_17 as writeToFile }
-export { _default_17 as writeToFile_alias_1 }
+export { _default_18 as default_alias_17 }
+export { _default_18 as writeToFile }
+export { _default_18 as writeToFile_alias_1 }
 
 declare const _default_2: {
     handler: ToolHandler<{
@@ -583,13 +623,15 @@ export declare const handler_alias_11: ToolHandler<typeof toolInfo_alias_11, Fil
 
 export declare const handler_alias_12: ToolHandler<typeof toolInfo_alias_12, FilesystemProvider>;
 
-export declare const handler_alias_13: ToolHandler<typeof toolInfo_alias_13, FilesystemProvider>;
+export declare const handler_alias_13: ToolHandler<typeof toolInfo_alias_13, WebProvider>;
 
-export declare const handler_alias_14: ToolHandler<typeof toolInfo_alias_14, MemoryProvider>;
+export declare const handler_alias_14: ToolHandler<typeof toolInfo_alias_14, FilesystemProvider>;
 
-export declare const handler_alias_15: ToolHandler<typeof toolInfo_alias_15, TodoProvider>;
+export declare const handler_alias_15: ToolHandler<typeof toolInfo_alias_15, MemoryProvider>;
 
-export declare const handler_alias_16: ToolHandler<typeof toolInfo_alias_16, FilesystemProvider>;
+export declare const handler_alias_16: ToolHandler<typeof toolInfo_alias_16, TodoProvider>;
+
+export declare const handler_alias_17: ToolHandler<typeof toolInfo_alias_17, FilesystemProvider>;
 
 export declare const handler_alias_2: ToolHandler<typeof toolInfo_alias_2, WebProvider>;
 
@@ -817,6 +859,7 @@ declare class MockProvider implements ToolProvider {
         summary?: string;
     }>;
     askFollowupQuestion(_question: string, _options?: string[]): Promise<string>;
+    search(_query: string): Promise<string>;
     listMemoryTopics(): Promise<string[]>;
     readMemory(_topic?: string): Promise<string | undefined>;
     updateMemory(_operation: 'append' | 'replace' | 'remove', _topic?: string, _content?: string): Promise<void>;
@@ -850,6 +893,15 @@ export { ParseOutputResult }
 export { ParseOutputResult as ParseOutputResult_alias_1 }
 export { ParseOutputResult as ParseOutputResult_alias_2 }
 
+declare const providerModelSchema: z.ZodObject<{
+    provider: z.ZodOptional<z.ZodString>;
+    model: z.ZodOptional<z.ZodString>;
+    parameters: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>;
+    budget: z.ZodOptional<z.ZodNumber>;
+}, z.core.$strip>;
+export { providerModelSchema }
+export { providerModelSchema as providerModelSchema_alias_1 }
+
 declare const replaceInFile_2: (fileContent: string, diff: string) => ReplaceResult;
 export { replaceInFile_2 as replaceInFileHelper }
 export { replaceInFile_2 as replaceInFile_alias_2 }
@@ -875,6 +927,20 @@ export { responsePrompts }
 export { responsePrompts as responsePrompts_alias_1 }
 export { responsePrompts as responsePrompts_alias_2 }
 
+declare const ruleSchema: z.ZodUnion<readonly [z.ZodString, z.ZodObject<{
+    path: z.ZodString;
+}, z.core.$strict>, z.ZodObject<{
+    url: z.ZodString;
+}, z.core.$strict>, z.ZodObject<{
+    repo: z.ZodString;
+    path: z.ZodString;
+    tag: z.ZodOptional<z.ZodString>;
+    commit: z.ZodOptional<z.ZodString>;
+    branch: z.ZodOptional<z.ZodString>;
+}, z.core.$strict>]>;
+export { ruleSchema }
+export { ruleSchema as ruleSchema_alias_1 }
+
 /**
  * Union type of all possible task events
  */
@@ -1113,6 +1179,14 @@ export declare const toolInfo_alias_12: {
 };
 
 export declare const toolInfo_alias_13: {
+    readonly name: "search";
+    readonly description: "Search the web for information using Google Search. Use this tool to find current information, facts, news, documentation, or research that is not available in your training data. Returns comprehensive search results with relevant content extracted from the web.";
+    readonly parameters: z.ZodObject<{
+        query: z.ZodString;
+    }, z.core.$strip>;
+};
+
+export declare const toolInfo_alias_14: {
     readonly name: "searchFiles";
     readonly description: "Request to perform a regex search across files in a specified directory, outputting context-rich results that include surrounding lines. This tool searches for patterns or specific content across multiple files, displaying each match with encapsulating context.";
     readonly parameters: z.ZodObject<{
@@ -1122,7 +1196,7 @@ export declare const toolInfo_alias_13: {
     }, z.core.$strip>;
 };
 
-export declare const toolInfo_alias_14: {
+export declare const toolInfo_alias_15: {
     readonly name: "updateMemory";
     readonly description: "Appends, replaces, or removes content from a memory topic. Use \"append\" to add to existing content, \"replace\" to overwrite entirely, or \"remove\" to delete a topic. Memory persists across tool calls within a workflow.";
     readonly parameters: z.ZodObject<{
@@ -1136,7 +1210,7 @@ export declare const toolInfo_alias_14: {
     }, z.core.$strip>;
 };
 
-export declare const toolInfo_alias_15: {
+export declare const toolInfo_alias_16: {
     readonly name: "updateTodoItem";
     readonly description: "Add or update a to-do item.";
     readonly parameters: ZodObject<    {
@@ -1156,7 +1230,7 @@ export declare const toolInfo_alias_15: {
     }, $strip>;
 };
 
-export declare const toolInfo_alias_16: {
+export declare const toolInfo_alias_17: {
     readonly name: "writeToFile";
     readonly description: "Request to write content to a file at the specified path. If the file exists, it will be overwritten with the provided content. If the file doesn't exist, it will be created. This tool will automatically create any directories needed to write the file. Ensure that the output content does not include incorrect escaped character patterns such as `&lt;`, `&gt;`, or `&amp;`. Also ensure there is no unwanted CDATA tags in the content.";
     readonly parameters: z.ZodObject<{
@@ -1436,6 +1510,7 @@ export { UsageMeter as UsageMeter_alias_1 }
 
 declare type WebProvider = {
     fetchUrl?: (url: string) => Promise<string>;
+    search?: (query: string) => Promise<string>;
 };
 export { WebProvider }
 export { WebProvider as WebProvider_alias_1 }

package/dist/index.d.ts

@@ -3,6 +3,9 @@ export { ParseOutputResult } from './_tsup-dts-rollup.js';
 export { parseJsonFromMarkdown } from './_tsup-dts-rollup.js';
 export { responsePrompts } from './_tsup-dts-rollup.js';
 export { computeRateLimitBackoffSeconds } from './_tsup-dts-rollup.js';
+export { providerModelSchema_alias_1 as providerModelSchema } from './_tsup-dts-rollup.js';
+export { ruleSchema_alias_1 as ruleSchema } from './_tsup-dts-rollup.js';
+export { ConfigRule_alias_1 as ConfigRule } from './_tsup-dts-rollup.js';
 export { configSchema_alias_1 as configSchema } from './_tsup-dts-rollup.js';
 export { Config_alias_1 as Config } from './_tsup-dts-rollup.js';
 export { ToolParameterValue } from './_tsup-dts-rollup.js';
@@ -30,6 +33,7 @@ export { readMemory } from './_tsup-dts-rollup.js';
 export { removeFile } from './_tsup-dts-rollup.js';
 export { renameFile } from './_tsup-dts-rollup.js';
 export { replaceInFile } from './_tsup-dts-rollup.js';
+export { search } from './_tsup-dts-rollup.js';
 export { searchFiles } from './_tsup-dts-rollup.js';
 export { updateMemory } from './_tsup-dts-rollup.js';
 export { updateTodoItem } from './_tsup-dts-rollup.js';

package/dist/index.js

@@ -133,6 +133,18 @@ var providerModelSchema = z.object({
   parameters: z.record(z.string(), z.any()).optional(),
   budget: z.number().positive().optional()
 });
+var ruleSchema = z.union([
+  z.string(),
+  z.object({ path: z.string() }).strict(),
+  z.object({ url: z.string() }).strict(),
+  z.object({
+    repo: z.string(),
+    path: z.string(),
+    tag: z.string().optional(),
+    commit: z.string().optional(),
+    branch: z.string().optional()
+  }).strict()
+]);
 var configSchema = z.object({
   prices: z.record(
     z.string(),
@@ -177,7 +189,10 @@ var configSchema = z.object({
     )
   ).optional(),
   commands: z.record(z.string(), providerModelSchema).optional(),
-  rules: z.array(z.string()).optional().or(z.string()).optional(),
+  tools: z.object({
+    search: providerModelSchema.or(z.boolean()).optional()
+  }).optional(),
+  rules: z.array(ruleSchema).optional().or(z.string()).optional(),
   excludeFiles: z.array(z.string()).optional()
 }).strict();
 
@@ -702,6 +717,9 @@ var MockProvider = class {
   async askFollowupQuestion(_question, _options) {
     return "mock answer";
   }
+  async search(_query) {
+    return "mock search result";
+  }
   async listMemoryTopics() {
     return ["default"];
   }
@@ -1241,19 +1259,74 @@ var replaceInFile_default = {
   handler: handler13
 };
 
-// src/tools/searchFiles.ts
+// src/tools/search.ts
 import { z as z16 } from "zod";
 var toolInfo14 = {
+  name: "search",
+  description: "Search the web for information using Google Search. Use this tool to find current information, facts, news, documentation, or research that is not available in your training data. Returns comprehensive search results with relevant content extracted from the web.",
+  parameters: z16.object({
+    query: z16.string().describe("The query to search for").meta({ usageValue: "Your search query here" })
+  }).meta({
+    examples: [
+      {
+        description: "Search for current events or news",
+        input: {
+          query: "latest developments in AI language models 2024"
+        }
+      },
+      {
+        description: "Look up technical documentation",
+        input: {
+          query: "TypeScript advanced type system features"
+        }
+      },
+      {
+        description: "Research specific information",
+        input: {
+          query: "Node.js performance optimization best practices"
+        }
+      }
+    ]
+  })
+};
+var handler14 = async (provider, args) => {
+  const { query } = toolInfo14.parameters.parse(args);
+  if (!provider.search) {
+    return {
+      type: "Error" /* Error */,
+      message: {
+        type: "text",
+        value: "This tool requires a web provider to be installed."
+      }
+    };
+  }
+  const result = await provider.search(query);
+  return {
+    type: "Reply" /* Reply */,
+    message: {
+      type: "text",
+      value: result
+    }
+  };
+};
+var search_default = {
+  ...toolInfo14,
+  handler: handler14
+};
+
+// src/tools/searchFiles.ts
+import { z as z17 } from "zod";
+var toolInfo15 = {
   name: "searchFiles",
   description: "Request to perform a regex search across files in a specified directory, outputting context-rich results that include surrounding lines. This tool searches for patterns or specific content across multiple files, displaying each match with encapsulating context.",
-  parameters: z16.object({
-    path: z16.string().describe(
+  parameters: z17.object({
+    path: z17.string().describe(
       "The path of the directory to search in (relative to the current working directory). This directory will be recursively searched."
     ).meta({ usageValue: "Directory path here" }),
-    regex: z16.string().describe("The regular expression pattern to search for. Uses Rust regex syntax.").meta({
+    regex: z17.string().describe("The regular expression pattern to search for. Uses Rust regex syntax.").meta({
       usageValue: "Your regex pattern here"
     }),
-    filePattern: z16.string().optional().describe(
+    filePattern: z17.string().optional().describe(
       'Comma-separated glob pattern to filter files (e.g., "*.ts" for TypeScript files or "*.ts,*.js" for both TypeScript and JavaScript files). If not provided, it will search all files (*).'
     ).meta({
       usageValue: "file pattern here (optional)"
@@ -1271,7 +1344,7 @@ var toolInfo14 = {
     ]
   })
 };
-var handler14 = async (provider, args) => {
+var handler15 = async (provider, args) => {
   if (!provider.searchFiles) {
     return {
       type: "Error" /* Error */,
@@ -1281,7 +1354,7 @@ var handler14 = async (provider, args) => {
       }
     };
   }
-  const parsed = toolInfo14.parameters.safeParse(args);
+  const parsed = toolInfo15.parameters.safeParse(args);
   if (!parsed.success) {
     return {
       type: "Error" /* Error */,
@@ -1318,19 +1391,19 @@ ${files.join("\n")}
   }
 };
 var searchFiles_default = {
-  ...toolInfo14,
-  handler: handler14
+  ...toolInfo15,
+  handler: handler15
 };
 
 // src/tools/updateMemory.ts
-import { z as z17 } from "zod";
-var toolInfo15 = {
+import { z as z18 } from "zod";
+var toolInfo16 = {
   name: "updateMemory",
   description: 'Appends, replaces, or removes content from a memory topic. Use "append" to add to existing content, "replace" to overwrite entirely, or "remove" to delete a topic. Memory persists across tool calls within a workflow.',
-  parameters: z17.object({
-    operation: z17.enum(["append", "replace", "remove"]).describe("The operation to perform."),
-    topic: z17.string().nullish().describe('The topic to update in memory. Defaults to ":default:".'),
-    content: z17.string().nullish().describe("The content for append or replace operations. Must be omitted for remove operation.")
+  parameters: z18.object({
+    operation: z18.enum(["append", "replace", "remove"]).describe("The operation to perform."),
+    topic: z18.string().nullish().describe('The topic to update in memory. Defaults to ":default:".'),
+    content: z18.string().nullish().describe("The content for append or replace operations. Must be omitted for remove operation.")
   }).superRefine((data, ctx) => {
     if (data.operation === "append" || data.operation === "replace") {
       if (data.content === void 0) {
@@ -1351,7 +1424,7 @@ var toolInfo15 = {
     }
   })
 };
-var handler15 = async (provider, args) => {
+var handler16 = async (provider, args) => {
   if (!provider.updateMemory) {
     return {
       type: "Error" /* Error */,
@@ -1361,7 +1434,7 @@ var handler15 = async (provider, args) => {
       }
     };
   }
-  const params = toolInfo15.parameters.parse(args);
+  const params = toolInfo16.parameters.parse(args);
   await provider.updateMemory(params.operation, params.topic ?? void 0, params.content ?? void 0);
   switch (params.operation) {
     case "append":
@@ -1391,17 +1464,17 @@ var handler15 = async (provider, args) => {
   }
 };
 var updateMemory_default = {
-  ...toolInfo15,
-  handler: handler15
+  ...toolInfo16,
+  handler: handler16
 };
 
 // src/tools/updateTodoItem.ts
-var toolInfo16 = {
+var toolInfo17 = {
   name: "updateTodoItem",
   description: "Add or update a to-do item.",
   parameters: UpdateTodoItemInputSchema
 };
-var handler16 = async (provider, args) => {
+var handler17 = async (provider, args) => {
   if (!provider.updateTodoItem) {
     return {
       type: "Error" /* Error */,
@@ -1411,7 +1484,7 @@ var handler16 = async (provider, args) => {
       }
     };
   }
-  const input = toolInfo16.parameters.parse(args);
+  const input = toolInfo17.parameters.parse(args);
   const result = await provider.updateTodoItem(input);
   return {
     type: "Reply" /* Reply */,
@@ -1422,18 +1495,18 @@ var handler16 = async (provider, args) => {
   };
 };
 var updateTodoItem_default = {
-  ...toolInfo16,
-  handler: handler16
+  ...toolInfo17,
+  handler: handler17
 };
 
 // src/tools/writeToFile.ts
-import { z as z18 } from "zod";
-var toolInfo17 = {
+import { z as z19 } from "zod";
+var toolInfo18 = {
   name: "writeToFile",
   description: "Request to write content to a file at the specified path. If the file exists, it will be overwritten with the provided content. If the file doesn't exist, it will be created. This tool will automatically create any directories needed to write the file. Ensure that the output content does not include incorrect escaped character patterns such as `<`, `>`, or `&`. Also ensure there is no unwanted CDATA tags in the content.",
-  parameters: z18.object({
-    path: z18.string().describe("The path of the file to write to").meta({ usageValue: "File path here" }),
-    content: z18.string().describe(
+  parameters: z19.object({
+    path: z19.string().describe("The path of the file to write to").meta({ usageValue: "File path here" }),
+    content: z19.string().describe(
       "The content to write to the file. ALWAYS provide the COMPLETE intended content of the file, without any truncation or omissions. You MUST include ALL parts of the file, even if they haven't been modified."
     ).meta({ usageValue: "Your file content here" })
   }).meta({
@@ -1459,7 +1532,7 @@ export default App;
     ]
   })
 };
-var handler17 = async (provider, args) => {
+var handler18 = async (provider, args) => {
   if (!provider.writeFile) {
     return {
       type: "Error" /* Error */,
@@ -1469,7 +1542,7 @@ var handler17 = async (provider, args) => {
       }
     };
   }
-  const parsed = toolInfo17.parameters.safeParse(args);
+  const parsed = toolInfo18.parameters.safeParse(args);
   if (!parsed.success) {
     return {
       type: "Error" /* Error */,
@@ -1492,8 +1565,8 @@ var handler17 = async (provider, args) => {
   };
 };
 var writeToFile_default = {
-  ...toolInfo17,
-  handler: handler17
+  ...toolInfo18,
+  handler: handler18
 };
 
 // src/UsageMeter.ts
@@ -1646,7 +1719,7 @@ var UsageMeter = class {
 
 // src/workflow/agent.workflow.ts
 import { jsonSchema } from "ai";
-import { toJSONSchema, z as z19 } from "zod";
+import { toJSONSchema, z as z20 } from "zod";
 
 // src/workflow/types.ts
 var TaskEventKind = /* @__PURE__ */ ((TaskEventKind2) => {
@@ -1664,13 +1737,13 @@ var TaskEventKind = /* @__PURE__ */ ((TaskEventKind2) => {
 })(TaskEventKind || {});
 
 // src/workflow/agent.workflow.ts
-var agentWorkflow = async (input, { step, tools }) => {
+var agentWorkflow = async (input, { step, tools, logger }) => {
   const event = (name, event2) => step(name, () => tools.taskEvent(event2));
-  const { tools: toolInfo18, maxToolRoundTrips = 200 } = input;
+  const { tools: toolInfo19, maxToolRoundTrips = 200 } = input;
   const messages = "systemPrompt" in input ? [{ role: "system", content: input.systemPrompt }] : input.messages;
   await event("start-task", { kind: "StartTask" /* StartTask */, systemPrompt: "systemPrompt" in input ? input.systemPrompt : "" });
   const toolSet = {};
-  for (const tool of toolInfo18) {
+  for (const tool of toolInfo19) {
     toolSet[tool.name] = {
       description: tool.description,
       inputSchema: jsonSchema(toJSONSchema(tool.parameters))
@@ -1696,7 +1769,11 @@ var agentWorkflow = async (input, { step, tools }) => {
       if (msg.content) {
         for (const part of msg.content) {
           if (part.type === "tool-call") {
-            toolCalls.push(part);
+            if (toolSet[part.toolName]) {
+              toolCalls.push(part);
+            } else {
+              logger.warn("Tool not found. Skipping.", part);
+            }
           }
         }
       }
@@ -1730,7 +1807,7 @@ var agentWorkflow = async (input, { step, tools }) => {
       }
       const validated = input.outputSchema.safeParse(parsed.data);
       if (!validated.success) {
-        const errorMessage = `Output validation failed. Error: ${z19.prettifyError(validated.error)}. Please correct the output.`;
+        const errorMessage = `Output validation failed. Error: ${z20.prettifyError(validated.error)}. Please correct the output.`;
         nextMessage = [{ role: "user", content: errorMessage }];
         continue;
       }
@@ -2010,6 +2087,7 @@ export {
   listTodoItems_default as listTodoItems,
   makeStepFn,
   parseJsonFromMarkdown,
+  providerModelSchema,
   readBinaryFile_default as readBinaryFile,
   readFile_default as readFile,
   readMemory_default as readMemory,
@@ -2018,6 +2096,8 @@ export {
   replaceInFile_default as replaceInFile,
   replaceInFile as replaceInFileHelper,
   responsePrompts,
+  ruleSchema,
+  search_default as search,
   searchFiles_default as searchFiles,
   toJsonModelMessage,
   updateMemory_default as updateMemory,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@polka-codes/core",
-  "version": "0.9.61",
+  "version": "0.9.63",
   "license": "AGPL-3.0",
   "author": "github@polka.codes",
   "type": "module",
@@ -19,7 +19,7 @@
   "dependencies": {
     "@ai-sdk/provider": "^2.0.0",
     "@ai-sdk/provider-utils": "^3.0.15",
-    "ai": "^5.0.86",
+    "ai": "^5.0.97",
     "yaml": "^2.8.1",
     "zod": "^4.1.12"
   }