wave-agent-sdk 0.6.4 → 0.7.0

package/src/tools/editTool.ts

@@ -23,7 +23,15 @@ export const editTool: ToolPlugin = {
   name: EDIT_TOOL_NAME,
   formatCompactParams,
   prompt: () =>
-    `Performs exact string replacements in files. \n\nUsage:\n- You must use your \`${READ_TOOL_NAME}\` tool at least once in the conversation before editing. This tool will error if you attempt an edit without reading the file. \n- When editing text from read_file tool output, ensure you preserve the exact indentation (tabs/spaces) as it appears AFTER the line number prefix. The line number prefix format is: spaces + line number + tab. Everything after that tab is the actual file content to match. Never include any part of the line number prefix in the old_string or new_string.\n- ALWAYS prefer editing existing files in the codebase. NEVER write new files unless explicitly required.\n- Only use emojis if the user explicitly requests it. Avoid adding emojis to files unless asked.\n- The edit will FAIL if \`old_string\` is not unique in the file. Either provide a larger string with more surrounding context to make it unique or use \`replace_all\` to change every instance of \`old_string\`. \n- Use \`replace_all\` for replacing and renaming strings across the file. This parameter is useful if you want to rename a variable for instance.`,
+    `Performs exact string replacements in files.
+
+Usage:
+- You must use your \`${READ_TOOL_NAME}\` tool at least once in the conversation before editing. This tool will error if you attempt an edit without reading the file. 
+- When editing text from ${READ_TOOL_NAME} tool output, ensure you preserve the exact indentation (tabs/spaces) as it appears AFTER the line number prefix. The line number prefix format is: spaces + line number + tab. Everything after that tab is the actual file content to match. Never include any part of the line number prefix in the old_string or new_string.
+- ALWAYS prefer editing existing files in the codebase. NEVER write new files unless explicitly required.
+- Only use emojis if the user explicitly requests it. Avoid adding emojis to files unless asked.
+- The edit will FAIL if \`old_string\` is not unique in the file. Either provide a larger string with more surrounding context to make it unique or use \`replace_all\` to change every instance of \`old_string\`. 
+- Use \`replace_all\` for replacing and renaming strings across the file. This parameter is useful if you want to rename a variable for instance.`,
   config: {
     type: "function",
     function: {

package/src/tools/readTool.ts

@@ -3,6 +3,7 @@ import { extname } from "path";
 import { logger } from "../utils/globalLogger.js";
 import type { ToolPlugin, ToolResult, ToolContext } from "./types.js";
 import { resolvePath, getDisplayPath } from "../utils/path.js";
+import { formatLineNumberPrefix } from "../utils/stringUtils.js";
 import {
   isBinaryDocument,
   getBinaryDocumentError,
@@ -142,7 +143,6 @@ Usage:
 - Any lines longer than 2000 characters will be truncated
 - Results are returned using cat -n format, with line numbers starting at 1
 - This tool allows Agent to read images (eg PNG, JPG, etc). When reading an image file the contents are presented visually as Agent is a multimodal LLM.
-- This tool can read Jupyter notebooks (.ipynb files) and returns all cells with their outputs, combining code, text, and visualizations.
 - You have the capability to call multiple tools in a single response. It is always better to speculatively read multiple files as a batch that are potentially useful.
 - You will regularly be asked to read screenshots. If the user provides a path to a screenshot ALWAYS use this tool to view the file at the path. This tool will work with all temporary file paths like /var/folders/123/abc/T/TemporaryItems/NSIRD_screencaptureui_ZfB1tD/Screenshot.png
 - If you read a file that exists but has empty contents you will receive a system reminder warning in place of file contents.
@@ -294,7 +294,7 @@ Usage:
           // Truncate overly long lines
           const truncatedLine =
             line.length > 2000 ? line.substring(0, 2000) + "..." : line;
-          return `${lineNumber.toString().padStart(6)}\t${truncatedLine}`;
+          return `${formatLineNumberPrefix(lineNumber)}${truncatedLine}`;
         })
         .join("\n");
 

package/src/tools/skillTool.ts

@@ -1,87 +1,91 @@
-import type { ToolPlugin, ToolResult } from "./types.js";
-import type { SkillManager } from "../managers/skillManager.js";
+import type { ToolPlugin, ToolResult, ToolContext } from "./types.js";
 import { SKILL_TOOL_NAME } from "../constants/tools.js";
+import type { SkillMetadata } from "../types/skills.js";
 
 /**
- * Create a skill tool plugin that uses the provided SkillManager
- * Note: SkillManager should be initialized before calling this function
+ * Skill tool plugin for invoking Wave skills
  */
-export function createSkillTool(skillManager: SkillManager): ToolPlugin {
-  // Ensure SkillManager is initialized
-  skillManager.getAvailableSkills();
-
-  return {
-    name: SKILL_TOOL_NAME,
-    get config() {
-      const availableSkills = skillManager.getAvailableSkills();
-
-      const getToolDescription = (): string => {
-        if (availableSkills.length === 0) {
-          return "Invoke a Wave skill by name. Skills are user-defined automation templates that can be personal or project-specific. No skills are currently available.";
-        }
-
-        const skillList = availableSkills
-          .map(
-            (skill) =>
-              `• **${skill.name}** (${skill.type}): ${skill.description}`,
-          )
-          .join("\n");
-
-        return `Invoke a Wave skill by name. Skills are user-defined automation templates that can be personal or project-specific.\n\nAvailable skills:\n${skillList}`;
-      };
-
-      return {
-        type: "function" as const,
-        function: {
-          name: SKILL_TOOL_NAME,
-          description: getToolDescription(),
-          parameters: {
-            type: "object",
-            properties: {
-              skill_name: {
-                type: "string",
-                description: "Name of the skill to invoke",
-                enum: availableSkills.map((skill) => skill.name),
-              },
-            },
-            required: ["skill_name"],
+export const skillTool: ToolPlugin = {
+  name: SKILL_TOOL_NAME,
+  config: {
+    type: "function" as const,
+    function: {
+      name: SKILL_TOOL_NAME,
+      description:
+        "Invoke a Wave skill by name. Skills are user-defined automation templates that can be personal or project-specific.",
+      parameters: {
+        type: "object",
+        properties: {
+          skill_name: {
+            type: "string",
+            description: "Name of the skill to invoke",
           },
         },
-      };
+        required: ["skill_name"],
+      },
     },
-    execute: async (args: Record<string, unknown>): Promise<ToolResult> => {
-      try {
-        // Validate arguments
-        const skillName = args.skill_name as string;
-        if (!skillName || typeof skillName !== "string") {
-          return {
-            success: false,
-            content: "",
-            error: "skill_name parameter is required and must be a string",
-          };
-        }
+  },
+
+  prompt: (args?: { availableSkills?: SkillMetadata[] }) => {
+    const availableSkills = args?.availableSkills;
+    if (!availableSkills || availableSkills.length === 0) {
+      return "Invoke a Wave skill by name. Skills are user-defined automation templates that can be personal or project-specific. No skills are currently available.";
+    }
 
-        // Execute the skill
-        const result = await skillManager.executeSkill({
-          skill_name: skillName,
-        });
+    const skillList = availableSkills
+      .map(
+        (skill) => `• **${skill.name}** (${skill.type}): ${skill.description}`,
+      )
+      .join("\n");
 
+    return `Invoke a Wave skill by name. Skills are user-defined automation templates that can be personal or project-specific.\n\nAvailable skills:\n${skillList}`;
+  },
+
+  execute: async (
+    args: Record<string, unknown>,
+    context: ToolContext,
+  ): Promise<ToolResult> => {
+    try {
+      const skillManager = context.skillManager;
+      if (!skillManager) {
         return {
-          success: true,
-          content: result.content,
-          shortResult: `Invoked skill: ${skillName}`,
+          success: false,
+          content: "",
+          error: "Skill manager not available in tool context",
         };
-      } catch (error) {
+      }
+
+      // Validate arguments
+      const skillName = args.skill_name as string;
+      if (!skillName || typeof skillName !== "string") {
         return {
           success: false,
           content: "",
-          error: error instanceof Error ? error.message : String(error),
+          error: "skill_name parameter is required and must be a string",
         };
       }
-    },
-    formatCompactParams: (params: Record<string, unknown>) => {
-      const skillName = params.skill_name as string;
-      return skillName || "unknown-skill";
-    },
-  };
-}
+
+      // Execute the skill
+      const result = await skillManager.executeSkill({
+        skill_name: skillName,
+      });
+
+      return {
+        success: true,
+        content: result.content,
+        shortResult: `Invoked skill: ${skillName}`,
+      };
+    } catch (error) {
+      return {
+        success: false,
+        content: "",
+        error: error instanceof Error ? error.message : String(error),
+      };
+    }
+  },
+
+  formatCompactParams: (params: Record<string, unknown>) => {
+    const skillName = params.skill_name as string;
+    return skillName || "unknown-skill";
+  },
+};

package/src/tools/taskOutputTool.ts

@@ -190,7 +190,7 @@ export const taskOutputTool: ToolPlugin = {
   },
   formatCompactParams: (params: Record<string, unknown>) => {
     const taskId = params.task_id as string;
-    const block = params.block as boolean;
+    const block = (params.block as boolean | undefined) ?? true;
     return `${taskId}${block ? " (blocking)" : ""}`;
   },
 };