@polka-codes/core 0.8.23 → 0.8.24

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

@@ -11,16 +11,16 @@ declare abstract class AgentBase {
     get parameters(): Readonly<any>;
     get messages(): Readonly<MessageParam[]>;
     setMessages(messages: Readonly<MessageParam[]>): void;
-    start(prompt: string): Promise<ExitReason>;
-    step(prompt: string): Promise<AssistantMessageContent[]>;
+    start(prompt: UserContent): Promise<ExitReason>;
+    step(prompt: UserContent): Promise<AssistantMessageContent[]>;
     handleStepResponse(response: AssistantMessageContent[]): Promise<{
         type: "reply";
-        message: string;
+        message: UserContent;
     } | {
         type: "exit";
         reason: ExitReason;
     }>;
-    continueTask(userMessage: string): Promise<ExitReason>;
+    continueTask(userMessage: UserContent): Promise<ExitReason>;
     protected abstract onBeforeInvokeTool(name: string, args: Record<string, string>): Promise<ToolResponse | undefined>;
     get model(): {
         provider: string;
@@ -1061,7 +1061,7 @@ export { _default_10 as updateKnowledge_alias_2 }
 declare const _default_11: {
     handler: ToolHandler<{
         readonly name: "write_to_file";
-        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;` and `&gt;`.";
+        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: [{
             readonly name: "path";
             readonly description: "The path of the file to write to";
@@ -1087,7 +1087,7 @@ declare const _default_11: {
     }, FilesystemProvider>;
     isAvailable: (provider: FilesystemProvider) => boolean;
     name: "write_to_file";
-    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;` and `&gt;`.";
+    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.";
     parameters: [{
         readonly name: "path";
         readonly description: "The path of the file to write to";
@@ -2111,6 +2111,8 @@ export declare const handler_alias_8: ToolHandler<typeof toolInfo_alias_8, Files
 
 export declare const handler_alias_9: ToolHandler<typeof toolInfo_alias_9, FilesystemProvider>;
 
+export declare type ImageBlockParam = Anthropic.Messages.ImageBlockParam;
+
 declare type Input = {
     commitMessages: string;
     commitDiff: string;
@@ -2613,7 +2615,7 @@ export { replaceInFile_2 as replaceInFile_alias_4 }
 declare const responsePrompts: {
     readonly errorInvokeTool: (tool: string, error: unknown) => string;
     readonly requireUseTool: "Error: No tool use detected. You MUST use a tool before proceeding.\ne.g. <tool_tool_name>tool_name</tool_tool_name>\n\nEnsure the opening and closing tags are correctly nested and closed, and that you are using the correct tool name.\nAvoid unnecessary text or symbols before or after the tool use.\nAvoid unnecessary escape characters or special characters.\n";
-    readonly toolResults: (tool: string, result: string) => string;
+    readonly toolResults: (tool: string, result: UserContent) => (TextBlockParam | ImageBlockParam)[];
     readonly commandResult: (command: string, exitCode: number, stdout: string, stderr: string) => string;
 };
 export { responsePrompts }
@@ -2726,7 +2728,7 @@ export { TaskEventKind as TaskEventKind_alias_2 }
  */
 declare interface TaskEventStartRequest extends TaskEventBase {
     kind: TaskEventKind.StartRequest;
-    userMessage: string;
+    userMessage: UserContent;
 }
 export { TaskEventStartRequest }
 export { TaskEventStartRequest as TaskEventStartRequest_alias_1 }
@@ -2809,6 +2811,8 @@ export { TaskEventUsageExceeded }
 export { TaskEventUsageExceeded as TaskEventUsageExceeded_alias_1 }
 export { TaskEventUsageExceeded as TaskEventUsageExceeded_alias_2 }
 
+export declare type TextBlockParam = Anthropic.Messages.TextBlockParam;
+
 declare interface TextContent {
     type: 'text';
     content: string;
@@ -3077,7 +3081,7 @@ export declare const toolInfo_alias_12: {
 
 export declare const toolInfo_alias_13: {
     readonly name: "write_to_file";
-    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;` and `&gt;`.";
+    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: [{
         readonly name: "path";
         readonly description: "The path of the file to write to";
@@ -3427,7 +3431,7 @@ export { ToolResponseInvalid as ToolResponseInvalid_alias_1 }
 declare type ToolResponseOrToolPause = {
     type: 'response';
     tool: string;
-    response: string;
+    response: UserContent;
 } | {
     type: 'pause';
     tool: string;
@@ -3446,7 +3450,7 @@ export { ToolResponsePause as ToolResponsePause_alias_1 }
 
 declare type ToolResponseReply = {
     type: ToolResponseType.Reply;
-    message: string;
+    message: UserContent;
 };
 export { ToolResponseReply }
 export { ToolResponseReply as ToolResponseReply_alias_1 }
@@ -3530,6 +3534,11 @@ export { UsageMeter }
 export { UsageMeter as UsageMeter_alias_1 }
 export { UsageMeter as UsageMeter_alias_2 }
 
+declare type UserContent = string | (TextBlockParam | ImageBlockParam)[];
+export { UserContent }
+export { UserContent as UserContent_alias_1 }
+export { UserContent as UserContent_alias_2 }
+
 declare type WebProvider = {
     fetchUrl?: (url: string) => Promise<string>;
 };

package/dist/index.d.ts

@@ -4,6 +4,7 @@ export { defaultModels } from './_tsup-dts-rollup.js';
 export { createService } from './_tsup-dts-rollup.js';
 export { UsageMeter } from './_tsup-dts-rollup.js';
 export { AiServiceBase } from './_tsup-dts-rollup.js';
+export { UserContent } from './_tsup-dts-rollup.js';
 export { MessageParam } from './_tsup-dts-rollup.js';
 export { AiServiceOptions } from './_tsup-dts-rollup.js';
 export { ModelInfo } from './_tsup-dts-rollup.js';

package/dist/index.js

@@ -2166,7 +2166,7 @@ var updateKnowledge_default = {
 // src/tools/writeToFile.ts
 var toolInfo11 = {
   name: "write_to_file",
-  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 `<` and `>`.",
+  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: [
     {
       name: "path",
@@ -2217,7 +2217,9 @@ var handler11 = async (provider, args) => {
     };
   }
   const path = getString(args, "path");
-  const content = getString(args, "content");
+  let content = getString(args, "content");
+  const trimmedContent = content.trim();
+  if (trimmedContent.startsWith("<![CDATA[") && content.endsWith("]]>")) content = trimmedContent.slice(9, -3);
   await provider.writeFile(path, content);
   return {
     type: "Reply" /* Reply */,
@@ -2773,12 +2775,27 @@ Ensure the opening and closing tags are correctly nested and closed, and that yo
 Avoid unnecessary text or symbols before or after the tool use.
 Avoid unnecessary escape characters or special characters.
 `,
-  toolResults: (tool, result) => `<tool_response>
-<tool_name>${tool}</tool_name>
-<tool_result>
-${result}
-</tool_result>
-</tool_response>`,
+  toolResults: (tool, result) => {
+    if (typeof result === "string") {
+      return [
+        {
+          type: "text",
+          text: `<tool_response name=${tool}>${result}</tool_response>`
+        }
+      ];
+    }
+    return [
+      {
+        type: "text",
+        text: `<tool_response name=${tool}>`
+      },
+      ...result,
+      {
+        type: "text",
+        text: "</tool_response>"
+      }
+    ];
+  },
   commandResult: (command, exitCode, stdout, stderr) => `<command>${command}</command>
 <command_exit_code>${exitCode}</command_exit_code>
 <command_stdout>
@@ -2960,23 +2977,26 @@ ${instance.prompt}`;
           await this.#callback({ kind: "ToolUse" /* ToolUse */, agent: this, tool: content.name });
           const toolResp = await this.#invokeTool(content.name, content.params);
           switch (toolResp.type) {
-            case "Reply" /* Reply */:
+            case "Reply" /* Reply */: {
               await this.#callback({ kind: "ToolReply" /* ToolReply */, agent: this, tool: content.name });
               toolResponses.push({ type: "response", tool: content.name, response: toolResp.message });
               break;
+            }
             case "Exit" /* Exit */:
               if (toolResponses.length > 0) {
                 break outer;
               }
               return { type: "exit", reason: toolResp };
-            case "Invalid" /* Invalid */:
+            case "Invalid" /* Invalid */: {
               await this.#callback({ kind: "ToolInvalid" /* ToolInvalid */, agent: this, tool: content.name });
               toolResponses.push({ type: "response", tool: content.name, response: toolResp.message });
               break outer;
-            case "Error" /* Error */:
+            }
+            case "Error" /* Error */: {
               await this.#callback({ kind: "ToolError" /* ToolError */, agent: this, tool: content.name });
               toolResponses.push({ type: "response", tool: content.name, response: toolResp.message });
               break outer;
+            }
             case "Interrupted" /* Interrupted */:
               await this.#callback({ kind: "ToolInterrupted" /* ToolInterrupted */, agent: this, tool: content.name });
               return { type: "exit", reason: toolResp };
@@ -3026,7 +3046,7 @@ ${instance.prompt}`;
     if (toolResponses.length === 0) {
       return { type: "reply", message: responsePrompts.requireUseTool };
     }
-    const finalResp = toolResponses.filter((resp) => resp.type === "response").map(({ tool, response: response2 }) => responsePrompts.toolResults(tool, response2)).join("\n\n");
+    const finalResp = toolResponses.filter((resp) => resp.type === "response").flatMap(({ tool, response: response2 }) => responsePrompts.toolResults(tool, response2));
     return { type: "reply", message: finalResp };
   }
   async #invokeTool(name, args) {

package/package.json

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