@polka-codes/core 0.9.61 → 0.9.62

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

@@ -237,6 +237,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.";
+        readonly parameters: z.ZodObject<{
+            query: z.ZodString;
+        }, z.core.$strip>;
+    }, WebProvider>;
+    name: "search";
+    description: "Search the web for information.";
+    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 +272,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 +302,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 +344,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 +364,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 +601,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, FilesystemProvider>;
 
-export declare const handler_alias_14: ToolHandler<typeof toolInfo_alias_14, MemoryProvider>;
+export declare const handler_alias_15: ToolHandler<typeof toolInfo_alias_15, MemoryProvider>;
 
-export declare const handler_alias_15: ToolHandler<typeof toolInfo_alias_15, TodoProvider>;
+export declare const handler_alias_16: ToolHandler<typeof toolInfo_alias_16, TodoProvider>;
 
-export declare const handler_alias_16: ToolHandler<typeof toolInfo_alias_16, FilesystemProvider>;
+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 +837,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>;
@@ -1113,6 +1134,14 @@ export declare const toolInfo_alias_12: {
 };
 
 export declare const toolInfo_alias_13: {
+    readonly name: "search";
+    readonly description: "Search the web for information.";
+    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 +1151,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 +1165,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 +1185,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 +1465,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

@@ -30,6 +30,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

@@ -702,6 +702,9 @@ var MockProvider = class {
   async askFollowupQuestion(_question, _options) {
     return "mock answer";
   }
+  async search(_query) {
+    return "mock search result";
+  }
   async listMemoryTopics() {
     return ["default"];
   }
@@ -1241,19 +1244,53 @@ 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.",
+  parameters: z16.object({
+    query: z16.string().describe("The query to search for")
+  })
+};
+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 +1308,7 @@ var toolInfo14 = {
     ]
   })
 };
-var handler14 = async (provider, args) => {
+var handler15 = async (provider, args) => {
   if (!provider.searchFiles) {
     return {
       type: "Error" /* Error */,
@@ -1281,7 +1318,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 +1355,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 +1388,7 @@ var toolInfo15 = {
     }
   })
 };
-var handler15 = async (provider, args) => {
+var handler16 = async (provider, args) => {
   if (!provider.updateMemory) {
     return {
       type: "Error" /* Error */,
@@ -1361,7 +1398,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 +1428,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 +1448,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 +1459,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 +1496,7 @@ export default App;
     ]
   })
 };
-var handler17 = async (provider, args) => {
+var handler18 = async (provider, args) => {
   if (!provider.writeFile) {
     return {
       type: "Error" /* Error */,
@@ -1469,7 +1506,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 +1529,8 @@ var handler17 = async (provider, args) => {
   };
 };
 var writeToFile_default = {
-  ...toolInfo17,
-  handler: handler17
+  ...toolInfo18,
+  handler: handler18
 };
 
 // src/UsageMeter.ts
@@ -1646,7 +1683,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 +1701,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 +1733,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 +1771,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;
       }
@@ -2018,6 +2059,7 @@ export {
   replaceInFile_default as replaceInFile,
   replaceInFile as replaceInFileHelper,
   responsePrompts,
+  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.62",
   "license": "AGPL-3.0",
   "author": "github@polka.codes",
   "type": "module",