wave-agent-sdk 0.13.1 → 0.13.3

package/src/utils/containerSetup.ts

@@ -1,6 +1,7 @@
 import { Container } from "./container.js";
 import { ForegroundTaskManager } from "../managers/foregroundTaskManager.js";
 import { BackgroundTaskManager } from "../managers/backgroundTaskManager.js";
+import { NotificationQueue } from "../managers/notificationQueue.js";
 import { TaskManager } from "../services/taskManager.js";
 import { MessageManager } from "../managers/messageManager.js";
 import { AIManager } from "../managers/aiManager.js";
@@ -18,6 +19,7 @@ import { CronManager } from "../managers/cronManager.js";
 import { MemoryRuleManager } from "../managers/MemoryRuleManager.js";
 import { ReversionManager } from "../managers/reversionManager.js";
 import { SubagentManager } from "../managers/subagentManager.js";
+import { ForkedAgentManager } from "../managers/forkedAgentManager.js";
 import { LiveConfigManager } from "../managers/liveConfigManager.js";
 import { ConfigurationService } from "../services/configurationService.js";
 import { ReversionService } from "../services/reversionService.js";
@@ -75,6 +77,9 @@ export function setupAgentContainer(
   const container = new Container();
   container.register("AgentOptions", options);
 
+  const notificationQueue = new NotificationQueue();
+  container.register("NotificationQueue", notificationQueue);
+
   const foregroundTaskManager = new ForegroundTaskManager(container);
   container.register("ForegroundTaskManager", foregroundTaskManager);
   container.register("ConfigurationService", configurationService);
@@ -283,6 +288,9 @@ export function setupAgentContainer(
   });
   container.register("SubagentManager", subagentManager);
 
+  const forkedAgentManager = new ForkedAgentManager(container);
+  container.register("ForkedAgentManager", forkedAgentManager);
+
   const aiManager = new AIManager(container, {
     callbacks: {
       ...callbacks,

package/src/utils/convertMessagesForAPI.ts

@@ -1,5 +1,6 @@
 import type { Message } from "../types/index.js";
 import { convertImageToBase64 } from "./messageOperations.js";
+import { taskNotificationToXml } from "./notificationXml.js";
 import { ChatCompletionMessageToolCall } from "openai/resources";
 import { stripAnsiColors } from "./stringUtils.js";
 import {
@@ -239,6 +240,14 @@ export function convertMessagesForAPI(
             text: `<local-command-stdout>\n${stripAnsiColors(block.result)}\n</local-command-stdout>`,
           });
         }
+
+        // If there is a task notification block, convert it back to XML
+        if (block.type === "task_notification") {
+          contentParts.push({
+            type: "text",
+            text: taskNotificationToXml(block),
+          });
+        }
       });
 
       // Only add user message if there is meaningful content

package/src/utils/messageOperations.ts

@@ -1,5 +1,10 @@
 import { randomUUID } from "crypto";
-import type { Message, Usage, ToolBlock } from "../types/index.js";
+import type {
+  Message,
+  Usage,
+  ToolBlock,
+  TaskNotificationBlock,
+} from "../types/index.js";
 import { MessageSource } from "../types/index.js";
 import { readFileSync } from "fs";
 import { extname } from "path";
@@ -135,6 +140,7 @@ export const addUserMessageToMessages = ({
   const textBlock = {
     type: "text" as const,
     content,
+    stage: "end" as const,
     ...(customCommandContent && { customCommandContent }),
     ...(source && { source }),
   };
@@ -584,3 +590,38 @@ export function getMessageContent(message: Message): string {
 
   return "";
 }
+
+export interface AddNotificationMessageParams {
+  messages: Message[];
+  taskId: string;
+  taskType: "shell" | "agent";
+  status: "completed" | "failed" | "killed";
+  summary: string;
+  outputFile?: string;
+}
+
+export const addNotificationMessageToMessages = ({
+  messages,
+  taskId,
+  taskType,
+  status,
+  summary,
+  outputFile,
+}: AddNotificationMessageParams): Message[] => {
+  const block: TaskNotificationBlock = {
+    type: "task_notification",
+    taskId,
+    taskType,
+    status,
+    summary,
+    ...(outputFile !== undefined && { outputFile }),
+  };
+
+  const notificationMessage: Message = {
+    id: generateMessageId(),
+    role: "user",
+    blocks: [block],
+  };
+
+  return [...messages, notificationMessage];
+};

package/src/utils/notificationXml.ts

@@ -0,0 +1,52 @@
+import type { TaskNotificationBlock } from "../types/messaging.js";
+
+export function taskNotificationToXml(block: TaskNotificationBlock): string {
+  let xml = `<task-notification>\n`;
+  xml += `<task-id>${block.taskId}</task-id>\n`;
+  xml += `<task-type>${block.taskType}</task-type>\n`;
+  if (block.outputFile) {
+    xml += `<output-file>${block.outputFile}</output-file>\n`;
+  }
+  xml += `<status>${block.status}</status>\n`;
+  xml += `<summary>${block.summary}</summary>\n`;
+  xml += `</task-notification>`;
+  return xml;
+}
+
+function extractTag(xml: string, tag: string): string | null {
+  const regex = new RegExp(`<${tag}>(.*?)</${tag}>`, "s");
+  const match = xml.match(regex);
+  return match ? match[1] : null;
+}
+
+export function parseTaskNotificationXml(
+  xml: string,
+): TaskNotificationBlock | null {
+  try {
+    const taskId = extractTag(xml, "task-id");
+    const taskType = extractTag(xml, "task-type") as "shell" | "agent" | null;
+    const status = extractTag(xml, "status") as
+      | "completed"
+      | "failed"
+      | "killed"
+      | null;
+    const summary = extractTag(xml, "summary");
+
+    if (!taskId || !taskType || !status || !summary) {
+      return null;
+    }
+
+    const outputFile = extractTag(xml, "output-file") || undefined;
+
+    return {
+      type: "task_notification",
+      taskId,
+      taskType,
+      status,
+      summary,
+      ...(outputFile && { outputFile }),
+    };
+  } catch {
+    return null;
+  }
+}

package/src/utils/pathEncoder.ts

@@ -27,7 +27,6 @@ export interface PathEncodingOptions {
   pathSeparatorReplacement?: string; // Default: '-'
   spaceReplacement?: string; // Default: '_'
   invalidCharReplacement?: string; // Default: '_'
-  preserveCase?: boolean; // Default: false (convert to lowercase)
   hashLength?: number; // Default: 8 characters
 }
 
@@ -65,7 +64,6 @@ export class PathEncoder {
       pathSeparatorReplacement: options.pathSeparatorReplacement ?? "-",
       spaceReplacement: options.spaceReplacement ?? "_",
       invalidCharReplacement: options.invalidCharReplacement ?? "_",
-      preserveCase: options.preserveCase ?? false,
       hashLength: options.hashLength ?? 8,
     };
     this.constraints = this.getFilesystemConstraints();
@@ -109,10 +107,7 @@ export class PathEncoder {
       this.options.invalidCharReplacement,
     );
 
-    // Convert to lowercase unless preserveCase is true
-    if (!this.options.preserveCase) {
-      encoded = encoded.toLowerCase();
-    }
+    // Case is preserved
 
     // Handle length limit with hash
     if (encoded.length > this.options.maxLength) {