wave-agent-sdk 0.12.10 → 0.13.0

package/src/utils/messageOperations.ts

@@ -9,6 +9,7 @@ import { logger } from "./globalLogger.js";
 // Base user message parameters interface
 export interface UserMessageParams {
   content: string;
+  customCommandContent?: string;
   images?: Array<{ path: string; mimeType: string }>;
   source?: MessageSource;
   isMeta?: boolean;
@@ -122,6 +123,7 @@ export const generateMessageId = (): string => `msg-${randomUUID()}`;
 export const addUserMessageToMessages = ({
   messages,
   content,
+  customCommandContent,
   images,
   source,
   id,
@@ -129,10 +131,11 @@ export const addUserMessageToMessages = ({
 }: AddUserMessageParams): Message[] => {
   const blocks: Message["blocks"] = [];
 
-  // Create text block with optional source
+  // Create text block with optional source and customCommandContent
   const textBlock = {
     type: "text" as const,
     content,
+    ...(customCommandContent && { customCommandContent }),
     ...(source && { source }),
   };
   blocks.push(textBlock);
@@ -170,6 +173,9 @@ export const updateUserMessageInMessages = (
           return {
             ...block,
             ...(params.content !== undefined && { content: params.content }),
+            ...(params.customCommandContent !== undefined && {
+              customCommandContent: params.customCommandContent,
+            }),
             ...(params.source !== undefined && { source: params.source }),
           };
         }
@@ -413,7 +419,7 @@ export const addBangMessage = ({
         type: "bang",
         command,
         output: "",
-        isRunning: true,
+        stage: "running" as const,
         exitCode: null,
       },
     ],
@@ -435,7 +441,9 @@ export const updateBangInMessage = ({
     if (msg.role === "user") {
       const commandBlock = msg.blocks.find(
         (block) =>
-          block.type === "bang" && block.command === command && block.isRunning,
+          block.type === "bang" &&
+          block.command === command &&
+          block.stage === "running",
       );
       if (commandBlock && commandBlock.type === "bang") {
         commandBlock.output = output.trim();
@@ -460,10 +468,12 @@ export const completeBangInMessage = ({
     if (msg.role === "user") {
       const commandBlock = msg.blocks.find(
         (block) =>
-          block.type === "bang" && block.command === command && block.isRunning,
+          block.type === "bang" &&
+          block.command === command &&
+          block.stage === "running",
       );
       if (commandBlock && commandBlock.type === "bang") {
-        commandBlock.isRunning = false;
+        commandBlock.stage = "end";
         commandBlock.exitCode = exitCode;
         if (output !== undefined) {
           commandBlock.output = output.trim();