wave-agent-sdk 0.5.0 → 0.6.0

package/src/types/messaging.ts

@@ -26,7 +26,6 @@ export type MessageBlock =
   | ImageBlock
   | CommandOutputBlock
   | CompressBlock
-  | MemoryBlock
   | SubagentBlock
   | ReasoningBlock
   | FileHistoryBlock;
@@ -67,6 +66,7 @@ export interface ToolBlock {
   error?: string | Error;
   compactParams?: string; // Compact parameter display
   parametersChunk?: string; // Incremental parameter updates for streaming
+  isManuallyBackgrounded?: boolean; // Whether the tool was manually backgrounded by the user
 }
 
 export interface ImageBlock {
@@ -88,14 +88,6 @@ export interface CompressBlock {
   sessionId: string;
 }
 
-export interface MemoryBlock {
-  type: "memory";
-  content: string;
-  isSuccess: boolean;
-  memoryType?: "project" | "user"; // Memory type
-  storagePath?: string; // Storage path text
-}
-
 export interface SubagentBlock {
   type: "subagent";
   subagentId: string;

package/src/types/processes.ts

@@ -12,7 +12,7 @@ export type BackgroundTaskStatus =
   | "killed";
 export type BackgroundTaskType = "shell" | "subagent";
 
-export interface BackgroundTask {
+export interface BackgroundTaskBase {
   id: string;
   type: BackgroundTaskType;
   status: BackgroundTaskStatus;
@@ -24,13 +24,24 @@ export interface BackgroundTask {
   stderr: string;
   exitCode?: number;
   runtime?: number;
+  /**
+   * Optional callback to be executed when the task is stopped.
+   * This allows tasks to define their own cleanup/abortion logic.
+   */
+  onStop?: () => void | Promise<void>;
 }
 
-export interface BackgroundShell extends BackgroundTask {
+export interface BackgroundShell extends BackgroundTaskBase {
   type: "shell";
   process: ChildProcess;
 }
 
+export interface BackgroundSubagent extends BackgroundTaskBase {
+  type: "subagent";
+}
+
+export type BackgroundTask = BackgroundShell | BackgroundSubagent;
+
 export interface ForegroundTask {
   id: string;
   backgroundHandler: () => Promise<void>;
@@ -40,8 +51,3 @@ export interface IForegroundTaskManager {
   registerForegroundTask(task: ForegroundTask): void;
   unregisterForegroundTask(id: string): void;
 }
-
-export interface IForegroundTaskManager {
-  registerForegroundTask(task: ForegroundTask): void;
-  unregisterForegroundTask(id: string): void;
-}

package/src/types/tasks.ts

@@ -0,0 +1,13 @@
+export type TaskStatus = "pending" | "in_progress" | "completed" | "deleted";
+
+export interface Task {
+  id: string;
+  subject: string;
+  description: string;
+  status: TaskStatus;
+  activeForm?: string;
+  owner?: string;
+  blocks: string[];
+  blockedBy: string[];
+  metadata: Record<string, unknown>;
+}

package/src/types/tools.ts

@@ -8,7 +8,6 @@
 export interface AskUserQuestionOption {
   label: string;
   description?: string;
-  isRecommended?: boolean;
 }
 
 export interface AskUserQuestion {
@@ -20,6 +19,10 @@ export interface AskUserQuestion {
 
 export interface AskUserQuestionInput {
   questions: AskUserQuestion[];
+  answers?: Record<string, string>;
+  metadata?: {
+    source?: string;
+  };
 }
 
 /**

package/src/utils/builtinSubagents.ts

@@ -1,4 +1,9 @@
-import { GENERAL_PURPOSE_SYSTEM_PROMPT } from "../constants/prompts.js";
+import {
+  BASH_SUBAGENT_SYSTEM_PROMPT,
+  GENERAL_PURPOSE_SYSTEM_PROMPT,
+  PLAN_SUBAGENT_SYSTEM_PROMPT,
+} from "../constants/prompts.js";
+import { BASH_TOOL_NAME } from "../constants/tools.js";
 import type { SubagentConfiguration } from "./subagentParser.js";
 
 /**
@@ -7,12 +12,32 @@ import type { SubagentConfiguration } from "./subagentParser.js";
  */
 export function getBuiltinSubagents(): SubagentConfiguration[] {
   return [
+    createBashSubagent(),
     createExploreSubagent(),
     createGeneralPurposeSubagent(),
+    createPlanSubagent(),
     // Add more built-in subagents here as needed
   ];
 }
 
+/**
+ * Create the Bash built-in subagent configuration
+ * Specialized for executing bash commands and git operations
+ */
+function createBashSubagent(): SubagentConfiguration {
+  return {
+    name: "Bash",
+    description:
+      "Command execution specialist for running bash commands. Use this for git operations, command execution, and other terminal tasks.",
+    systemPrompt: BASH_SUBAGENT_SYSTEM_PROMPT,
+    tools: [BASH_TOOL_NAME],
+    model: "inherit",
+    filePath: "<builtin:Bash>",
+    scope: "builtin",
+    priority: 3,
+  };
+}
+
 /**
  * Create the General-Purpose built-in subagent configuration
  * Specialized for multi-step research and implementation tasks
@@ -87,3 +112,24 @@ Complete the user's search request efficiently and report your findings clearly.
     priority: 3, // Lowest priority - can be overridden by user configs
   };
 }
+
+/**
+ * Create the Plan built-in subagent configuration
+ * Specialized for designing implementation plans and exploring codebases in read-only mode
+ */
+function createPlanSubagent(): SubagentConfiguration {
+  // Define allowed tools for read-only operations
+  const allowedTools = ["Glob", "Grep", "Read", "Bash", "LS", "LSP"];
+
+  return {
+    name: "Plan",
+    description:
+      "Software architect agent for designing implementation plans. Use this when you need to plan the implementation strategy for a task. Returns step-by-step plans, identifies critical files, and considers architectural trade-offs.",
+    systemPrompt: PLAN_SUBAGENT_SYSTEM_PROMPT,
+    tools: allowedTools,
+    model: "inherit", // Uses parent agent's model
+    filePath: "<builtin:Plan>",
+    scope: "builtin",
+    priority: 3, // Lowest priority - can be overridden by user configs
+  };
+}

package/src/utils/cacheControlUtils.ts

@@ -316,14 +316,8 @@ export function transformMessagesForClaudeCache(
   // Find the latest interval message index (20th, 40th, 60th, etc.)
   const intervalMessageIndex = findIntervalMessageIndex(messages);
 
-  // Find last system message index
-  let lastSystemIndex = -1;
-  for (let i = messages.length - 1; i >= 0; i--) {
-    if (messages[i].role === "system") {
-      lastSystemIndex = i;
-      break;
-    }
-  }
+  // Find first system message index
+  const firstSystemIndex = messages.findIndex((m) => m.role === "system");
 
   const result = messages.map((message, index) => {
     // Validate message structure
@@ -332,14 +326,27 @@ export function transformMessagesForClaudeCache(
       return message; // Return as-is to avoid breaking the flow
     }
 
-    // Last system message: always cached (hardcoded)
-    if (message.role === "system" && index === lastSystemIndex) {
+    // First system message: always cached (hardcoded)
+    if (message.role === "system" && index === firstSystemIndex) {
+      const content =
+        (message.content as string | ChatCompletionContentPart[]) || "";
+      // If content is already an array, check if it already has cache control
+      if (Array.isArray(content)) {
+        const hasCacheControl = content.some(
+          (part) =>
+            part.type === "text" &&
+            (part as ClaudeChatCompletionContentPartText).cache_control,
+        );
+        if (hasCacheControl) {
+          return message;
+        }
+      }
+
+      const transformedContent = addCacheControlToContent(content, true);
+
       return {
         ...message,
-        content: addCacheControlToContent(
-          (message.content as string | ChatCompletionContentPart[]) || "",
-          true,
-        ),
+        content: transformedContent,
       } as ChatCompletionMessageParam;
     }
 
@@ -364,12 +371,13 @@ export function transformMessagesForClaudeCache(
         } as ChatCompletionMessageParam;
       } else {
         // For other message types without tool calls, cache the content
+        const content =
+          (message.content as string | ChatCompletionContentPart[]) || "";
+        const transformedContent = addCacheControlToContent(content, true);
+
         return {
           ...message,
-          content: addCacheControlToContent(
-            (message.content as string | ChatCompletionContentPart[]) || "",
-            true,
-          ),
+          content: transformedContent,
         } as ChatCompletionMessageParam;
       }
     }

package/src/utils/constants.ts

@@ -31,8 +31,3 @@ export const USER_MEMORY_FILE = path.join(DATA_DIRECTORY, "AGENTS.md");
  */
 export const DEFAULT_WAVE_MAX_INPUT_TOKENS = 96000; // Default token limit
 export const DEFAULT_WAVE_MAX_OUTPUT_TOKENS = 8192; // Default output token limit
-
-/**
- * Default number of messages to keep uncompressed
- */
-export const DEFAULT_KEEP_LAST_MESSAGES_COUNT = 7;

package/src/utils/convertMessagesForAPI.ts

@@ -57,8 +57,8 @@ export function convertMessagesForAPI(
       );
       if (compressBlock && compressBlock.type === "compress") {
         recentMessages.unshift({
-          role: "system",
-          content: `[Compressed Message Summary] ${compressBlock.content}`,
+          role: "assistant",
+          content: compressBlock.content,
         });
       }
       break;

package/src/utils/messageOperations.ts

@@ -1,6 +1,5 @@
 import type { Message, Usage } from "../types/index.js";
 import { MessageSource } from "../types/index.js";
-import { DEFAULT_KEEP_LAST_MESSAGES_COUNT } from "./constants.js";
 import type { SubagentConfiguration } from "./subagentParser.js";
 import { readFileSync } from "fs";
 import { extname } from "path";
@@ -40,6 +39,7 @@ export interface UpdateToolBlockParams {
   images?: Array<{ data: string; mediaType?: string }>;
   compactParams?: string;
   parametersChunk?: string; // Incremental parameter updates for streaming
+  isManuallyBackgrounded?: boolean;
 }
 
 // Agent specific interfaces (without messages parameter)
@@ -53,14 +53,6 @@ export interface AddErrorBlockParams {
   error: string;
 }
 
-export interface AddMemoryBlockParams {
-  messages: Message[];
-  content: string;
-  isSuccess: boolean;
-  memoryType?: "project" | "user";
-  storagePath?: string;
-}
-
 export interface AddCommandOutputParams {
   messages: Message[];
   command: string;
@@ -230,6 +222,7 @@ export const updateToolBlockInMessage = ({
   images,
   compactParams,
   parametersChunk,
+  isManuallyBackgrounded,
 }: UpdateToolBlockParams): Message[] => {
   const newMessages = [...messages];
   // Find the last assistant message
@@ -253,6 +246,8 @@ export const updateToolBlockInMessage = ({
             toolBlock.compactParams = compactParams;
           if (parametersChunk !== undefined)
             toolBlock.parametersChunk = parametersChunk;
+          if (isManuallyBackgrounded !== undefined)
+            toolBlock.isManuallyBackgrounded = isManuallyBackgrounded;
         }
       } else {
         // If existing block not found, create new one
@@ -270,6 +265,7 @@ export const updateToolBlockInMessage = ({
           stage: stage ?? "start",
           compactParams: compactParams,
           parametersChunk: parametersChunk,
+          isManuallyBackgrounded: isManuallyBackgrounded,
         });
       }
       break;
@@ -315,113 +311,6 @@ export const addErrorBlockToMessage = ({
   return newMessages;
 };
 
-// Add Memory Block as new assistant message
-export const addMemoryBlockToMessage = ({
-  messages,
-  content,
-  isSuccess,
-  memoryType,
-  storagePath,
-}: AddMemoryBlockParams): Message[] => {
-  const newMessages = [...messages];
-
-  // Create new assistant message containing MemoryBlock
-  const memoryMessage: Message = {
-    role: "assistant",
-    blocks: [
-      {
-        type: "memory",
-        content,
-        isSuccess,
-        memoryType,
-        storagePath,
-      },
-    ],
-  };
-
-  // Add to end of message list
-  newMessages.push(memoryMessage);
-  return newMessages;
-};
-
-/**
- * Count valid blocks from the end
- * Only text, image, and tool type blocks are counted
- * @param messages Message array
- * @param targetCount Number of valid blocks to count
- * @returns { messageIndex: number, blockCount: number } Message index and actual counted block count
- */
-export const countValidBlocksFromEnd = (
-  messages: Message[],
-  targetCount: number,
-): { messageIndex: number; blockCount: number } => {
-  let validBlockCount = 0;
-
-  // Iterate messages from end to beginning
-  for (let i = messages.length - 1; i >= 0; i--) {
-    const message = messages[i];
-
-    // Iterate through all blocks of current message
-    for (const block of message.blocks) {
-      // Only count valid block types
-      if (
-        block.type === "text" ||
-        block.type === "image" ||
-        block.type === "tool"
-      ) {
-        validBlockCount++;
-
-        // If target count reached, return current message index
-        if (validBlockCount >= targetCount) {
-          return { messageIndex: i, blockCount: validBlockCount };
-        }
-      }
-    }
-  }
-
-  // If target count not reached, return index 0
-  return { messageIndex: 0, blockCount: validBlockCount };
-};
-
-/**
- * Get messages to be compressed and insertion position
- * @param messages Message array
- * @param keepLastCount Keep the last few valid blocks uncompressed
- * @returns { messagesToCompress: Message[], insertIndex: number }
- */
-export const getMessagesToCompress = (
-  messages: Message[],
-  keepLastCount: number = DEFAULT_KEEP_LAST_MESSAGES_COUNT,
-): { messagesToCompress: Message[]; insertIndex: number } => {
-  // Calculate message position to keep from end to beginning
-  const { messageIndex } = countValidBlocksFromEnd(messages, keepLastCount);
-
-  // Find the last message containing compression block
-  let lastCompressIndex = -1;
-  for (let i = messages.length - 1; i >= 0; i--) {
-    const hasCompressBlock = messages[i].blocks.some(
-      (block) => block.type === "compress",
-    );
-    if (hasCompressBlock) {
-      lastCompressIndex = i;
-      break;
-    }
-  }
-
-  // Determine compression start position
-  // If compression block exists, start from compression block position (include compression block)
-  // If no compression block, start from beginning
-  const startIndex = lastCompressIndex >= 0 ? lastCompressIndex : 0;
-
-  // Messages to compress are all messages from start position to before calculated position
-  const messagesToCompress = messages.slice(startIndex, messageIndex);
-
-  // Change insertion position to negative number, indicating position from end
-  const insertIndex = messageIndex - messages.length;
-
-  return { messagesToCompress, insertIndex };
-};
-
 // Add command output block to message list
 export const addCommandOutputMessage = ({
   messages,

package/src/utils/nameGenerator.ts

@@ -71,8 +71,25 @@ const nouns = [
 /**
  * Generates a random English name (adjective-noun)
  */
-export function generateRandomName(): string {
-  const adj = adjectives[Math.floor(Math.random() * adjectives.length)];
-  const noun = nouns[Math.floor(Math.random() * nouns.length)];
+export function generateRandomName(seed?: string): string {
+  let adjIndex: number;
+  let nounIndex: number;
+
+  if (seed) {
+    // Simple hash function to derive indices from seed
+    let hash = 0;
+    for (let i = 0; i < seed.length; i++) {
+      hash = (hash << 5) - hash + seed.charCodeAt(i);
+      hash |= 0; // Convert to 32bit integer
+    }
+    adjIndex = Math.abs(hash) % adjectives.length;
+    nounIndex = Math.abs(hash >> 8) % nouns.length;
+  } else {
+    adjIndex = Math.floor(Math.random() * adjectives.length);
+    nounIndex = Math.floor(Math.random() * nouns.length);
+  }
+
+  const adj = adjectives[adjIndex];
+  const noun = nouns[nounIndex];
   return `${adj}-${noun}`;
 }

package/dist/tools/todoWriteTool.d.ts

@@ -1,6 +0,0 @@
-import type { ToolPlugin } from "./types.js";
-/**
- * TodoWrite tool for creating and managing structured task lists
- */
-export declare const todoWriteTool: ToolPlugin;
-//# sourceMappingURL=todoWriteTool.d.ts.map

package/dist/tools/todoWriteTool.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"todoWriteTool.d.ts","sourceRoot":"","sources":["../../src/tools/todoWriteTool.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAA2B,MAAM,YAAY,CAAC;AAStE;;GAEG;AACH,eAAO,MAAM,aAAa,EAAE,UAoP3B,CAAC"}