🤖 fix: batch multiple queued messages with proper display

- Track per-message display text in MessageQueue
- Compaction commands show their rawCommand, normal messages show their text
- All queued messages are shown newline-separated
- Follow-up messages during compaction now visible in queue

_Generated with `mux`_

Author: ammar-agent

src/node/services/messageQueue.test.ts

@@ -35,7 +35,7 @@ describe("MessageQueue", () => {
       expect(queue.getDisplayText()).toBe("/compact -t 3000");
     });
 
-    it("should return rawCommand even with multiple messages if last has compaction metadata", () => {
+    it("should show all messages when compaction is added after normal message", () => {
       queue.add("First message");
 
       const metadata: MuxFrontendMetadata = {
@@ -51,8 +51,8 @@ describe("MessageQueue", () => {
 
       queue.add("Summarize this conversation...", options);
 
-      // Should use rawCommand from latest options
-      expect(queue.getDisplayText()).toBe("/compact");
+      // Should show all messages: first message + compaction rawCommand
+      expect(queue.getDisplayText()).toBe("First message\n/compact");
     });
 
     it("should return joined messages when metadata type is not compaction-request", () => {
@@ -116,6 +116,67 @@ describe("MessageQueue", () => {
     });
   });
 
+  describe("multi-message batching", () => {
+    it("should batch multiple follow-up messages", () => {
+      queue.add("First message");
+      queue.add("Second message");
+      queue.add("Third message");
+
+      expect(queue.getMessages()).toEqual(["First message", "Second message", "Third message"]);
+      expect(queue.getDisplayText()).toBe("First message\nSecond message\nThird message");
+    });
+
+    it("should batch follow-up message after compaction", () => {
+      const metadata: MuxFrontendMetadata = {
+        type: "compaction-request",
+        rawCommand: "/compact",
+        parsed: {},
+      };
+
+      queue.add("Summarize...", {
+        model: "claude-3-5-sonnet-20241022",
+        muxMetadata: metadata,
+      });
+      queue.add("And then do this follow-up task");
+
+      // Display shows slash command for compaction, then the follow-up
+      expect(queue.getDisplayText()).toBe("/compact\nAnd then do this follow-up task");
+      // Raw messages have the actual prompt
+      expect(queue.getMessages()).toEqual(["Summarize...", "And then do this follow-up task"]);
+    });
+
+    it("should produce combined message for API call", () => {
+      queue.add("First message", { model: "gpt-4" });
+      queue.add("Second message");
+
+      const { message, options } = queue.produceMessage();
+
+      // Messages are joined with newlines
+      expect(message).toBe("First message\nSecond message");
+      // Latest options are used
+      expect(options?.model).toBe("gpt-4");
+    });
+
+    it("should batch messages with mixed images", () => {
+      const image1 = { url: "data:image/png;base64,abc", mediaType: "image/png" };
+      const image2 = { url: "data:image/jpeg;base64,def", mediaType: "image/jpeg" };
+
+      queue.add("Message with image", { model: "gpt-4", imageParts: [image1] });
+      queue.add("Follow-up without image");
+      queue.add("Another with image", { model: "gpt-4", imageParts: [image2] });
+
+      expect(queue.getMessages()).toEqual([
+        "Message with image",
+        "Follow-up without image",
+        "Another with image",
+      ]);
+      expect(queue.getImageParts()).toEqual([image1, image2]);
+      expect(queue.getDisplayText()).toBe(
+        "Message with image\nFollow-up without image\nAnother with image"
+      );
+    });
+  });
+
   describe("getImageParts", () => {
     it("should return accumulated images from multiple messages", () => {
       const image1 = {

src/node/services/messageQueue.ts

@@ -17,11 +17,13 @@ function isCompactionMetadata(meta: unknown): meta is CompactionMetadata {
  *
  * Stores:
  * - Message texts (accumulated)
+ * - Display texts (per message, handles slash commands like /compact)
  * - Latest options (model, thinking level, etc. - overwrites on each add)
  * - Image parts (accumulated across all messages)
  */
 export class MessageQueue {
   private messages: string[] = [];
+  private displayTexts: string[] = []; // Per-message display text
   private latestOptions?: SendMessageOptions;
   private accumulatedImages: ImagePart[] = [];
 
@@ -42,6 +44,13 @@ export class MessageQueue {
     // Add text message if non-empty
     if (trimmedMessage.length > 0) {
       this.messages.push(trimmedMessage);
+
+      // Determine display text: use rawCommand for compaction, otherwise the message itself
+      const muxMetadata = options?.muxMetadata as unknown;
+      const displayText = isCompactionMetadata(muxMetadata)
+        ? muxMetadata.rawCommand
+        : trimmedMessage;
+      this.displayTexts.push(displayText);
     }
 
     if (options) {
@@ -63,18 +72,11 @@ export class MessageQueue {
 
   /**
    * Get display text for queued messages.
-   * Returns rawCommand if this is a compaction request, otherwise joined messages.
-   * Matches StreamingMessageAggregator behavior.
+   * Returns all display texts joined with newlines.
+   * Compaction requests show their rawCommand, normal messages show their text.
    */
   getDisplayText(): string {
-    // Check if we have compaction metadata (cast from z.any() schema type)
-    const cmuxMetadata = this.latestOptions?.muxMetadata as unknown;
-    if (isCompactionMetadata(cmuxMetadata)) {
-      return cmuxMetadata.rawCommand;
-    }
-
-    // Otherwise return joined messages
-    return this.messages.join("\n");
+    return this.displayTexts.join("\n");
   }
 
   /**
@@ -109,6 +111,7 @@ export class MessageQueue {
    */
   clear(): void {
     this.messages = [];
+    this.displayTexts = [];
     this.latestOptions = undefined;
     this.accumulatedImages = [];
   }