wave-agent-sdk 0.12.0 → 0.12.2

package/src/utils/messageOperations.ts

@@ -614,6 +614,32 @@ export const removeLastUserMessage = (messages: Message[]): Message[] => {
   return newMessages;
 };
 
+/**
+ * Efficiently clone a message for UI freezing.
+ * Clones the message object and its blocks array, but reuses string references.
+ */
+export function cloneMessage(message: Message): Message {
+  return {
+    ...message,
+    blocks: message.blocks.map((block) => {
+      const clonedBlock = { ...block };
+      // Deep clone arrays/objects within blocks if they exist
+      if (clonedBlock.type === "tool" && clonedBlock.images) {
+        clonedBlock.images = clonedBlock.images.map((img) => ({ ...img }));
+      } else if (clonedBlock.type === "image" && clonedBlock.imageUrls) {
+        clonedBlock.imageUrls = [...clonedBlock.imageUrls];
+      } else if (clonedBlock.type === "file_history" && clonedBlock.snapshots) {
+        clonedBlock.snapshots = clonedBlock.snapshots.map((s) => ({ ...s }));
+      }
+      return clonedBlock;
+    }) as Message["blocks"],
+    // Clone additionalFields if it exists
+    ...(message.additionalFields
+      ? { additionalFields: { ...message.additionalFields } }
+      : {}),
+  };
+}
+
 /**
  * Helper to format tool and token summary
  */
@@ -631,3 +657,34 @@ export function formatToolTokenSummary(
   summary += ")";
   return summary;
 }
+
+/**
+ * Extracts displayable text from a Message object, handling various block types.
+ * Returns the first available content block.
+ */
+export function getMessageContent(message: Message): string {
+  // Find first available content block
+  const textBlock = message.blocks.find((block) => block.type === "text");
+  if (textBlock && "content" in textBlock) {
+    return textBlock.content;
+  }
+
+  const slashBlock = message.blocks.find((block) => block.type === "slash");
+  if (slashBlock && "command" in slashBlock) {
+    return `/${slashBlock.command}${slashBlock.args ? ` ${slashBlock.args}` : ""}`;
+  }
+
+  const bangBlock = message.blocks.find((block) => block.type === "bang");
+  if (bangBlock && "command" in bangBlock) {
+    return `!${bangBlock.command}`;
+  }
+
+  const compressBlock = message.blocks.find(
+    (block) => block.type === "compress",
+  );
+  if (compressBlock && "content" in compressBlock) {
+    return compressBlock.content;
+  }
+
+  return "";
+}

package/src/utils/stringUtils.ts

@@ -91,3 +91,23 @@ export const stripAnsiColors = (text: string): string => {
 export function formatLineNumberPrefix(lineNumber: number): string {
   return `${lineNumber.toString().padStart(6)}\t`;
 }
+
+/**
+ * Efficiently get the last N lines of a string without splitting the whole string.
+ */
+export function getLastLines(text: string, count: number): string {
+  if (!text || count <= 0) return "";
+  let pos = text.length;
+  let found = 0;
+  while (pos > 0 && found < count) {
+    const nextNewline = text.lastIndexOf("\n", pos - 1);
+    if (nextNewline === -1) {
+      pos = 0;
+      break;
+    } else {
+      pos = nextNewline;
+      found++;
+    }
+  }
+  return text.substring(pos === 0 && found < count ? 0 : pos + 1);
+}