wave-agent-sdk 0.11.7 → 0.12.0

package/src/utils/messageOperations.ts

@@ -10,7 +10,6 @@ import { logger } from "./globalLogger.js";
 export interface UserMessageParams {
   content: string;
   images?: Array<{ path: string; mimeType: string }>;
-  customCommandContent?: string;
   source?: MessageSource;
   isMeta?: boolean;
 }
@@ -21,6 +20,26 @@ export interface AddUserMessageParams extends UserMessageParams {
   id?: string;
 }
 
+export interface AddSlashParams {
+  messages: Message[];
+  command: string;
+  args?: string;
+  content?: string;
+  id?: string;
+}
+
+export interface UpdateSlashParams {
+  messages: Message[];
+  command: string;
+  messageId?: string;
+  args?: string;
+  content?: string;
+  result?: string;
+  stage?: "running" | "success" | "error" | "aborted";
+  error?: string;
+  shortResult?: string;
+}
+
 export interface UpdateToolBlockParams {
   messages: Message[];
   id: string;
@@ -124,18 +143,16 @@ export const addUserMessageToMessages = ({
   messages,
   content,
   images,
-  customCommandContent,
   source,
   id,
   isMeta,
 }: AddUserMessageParams): Message[] => {
   const blocks: Message["blocks"] = [];
 
-  // Create text block with optional customCommandContent and source
+  // Create text block with optional source
   const textBlock = {
     type: "text" as const,
     content,
-    ...(customCommandContent && { customCommandContent }),
     ...(source && { source }),
   };
   blocks.push(textBlock);
@@ -159,7 +176,96 @@ export const addUserMessageToMessages = ({
 };
 
 /**
- * Update a user message's content or customCommandContent by its ID.
+ * Add a slash command message to the conversation.
+ */
+export const addSlashMessageToMessages = ({
+  messages,
+  command,
+  args,
+  content,
+  id,
+}: AddSlashParams): Message[] => {
+  const slashMessage: Message = {
+    id: id || generateMessageId(),
+    role: "user",
+    blocks: [
+      {
+        type: "slash",
+        command,
+        args,
+        content,
+        stage: "running",
+      },
+    ],
+  };
+  return [...messages, slashMessage];
+};
+
+/**
+ * Update a slash block in a message.
+ */
+export const updateSlashBlockInMessage = ({
+  messages,
+  command,
+  messageId,
+  args,
+  content,
+  result,
+  stage,
+  error,
+  shortResult,
+}: UpdateSlashParams): Message[] => {
+  const newMessages = [...messages];
+
+  // If messageId is provided, target that specific message
+  if (messageId) {
+    const messageIndex = newMessages.findIndex((msg) => msg.id === messageId);
+    if (messageIndex !== -1) {
+      const slashBlockIndex = newMessages[messageIndex].blocks.findIndex(
+        (block) => block.type === "slash" && block.command === command,
+      );
+
+      if (slashBlockIndex !== -1) {
+        const slashBlock = newMessages[messageIndex].blocks[slashBlockIndex];
+        if (slashBlock.type === "slash") {
+          if (args !== undefined) slashBlock.args = args;
+          if (content !== undefined) slashBlock.content = content;
+          if (result !== undefined) slashBlock.result = result;
+          if (stage !== undefined) slashBlock.stage = stage;
+          if (error !== undefined) slashBlock.error = error;
+          if (shortResult !== undefined) slashBlock.shortResult = shortResult;
+        }
+      }
+    }
+    return newMessages;
+  }
+
+  // Find the last user message with a slash block for this command
+  for (let i = newMessages.length - 1; i >= 0; i--) {
+    const msg = newMessages[i];
+    if (msg.role === "user") {
+      const slashBlockIndex = msg.blocks.findIndex(
+        (block) => block.type === "slash" && block.command === command,
+      );
+      if (slashBlockIndex !== -1) {
+        const slashBlock = msg.blocks[slashBlockIndex];
+        if (slashBlock.type === "slash") {
+          if (args !== undefined) slashBlock.args = args;
+          if (content !== undefined) slashBlock.content = content;
+          if (result !== undefined) slashBlock.result = result;
+          if (stage !== undefined) slashBlock.stage = stage;
+          if (error !== undefined) slashBlock.error = error;
+          if (shortResult !== undefined) slashBlock.shortResult = shortResult;
+        }
+        break;
+      }
+    }
+  }
+  return newMessages;
+};
+
+/**
+ * Update a user message's content by its ID.
  */
 export const updateUserMessageInMessages = (
   messages: Message[],
@@ -173,9 +279,6 @@ export const updateUserMessageInMessages = (
           return {
             ...block,
             ...(params.content !== undefined && { content: params.content }),
-            ...(params.customCommandContent !== undefined && {
-              customCommandContent: params.customCommandContent,
-            }),
             ...(params.source !== undefined && { source: params.source }),
           };
         }