fgg

app/lib/runtime/enhanced-message-parser.ts

@@ -41,8 +41,8 @@ export class EnhancedStreamingMessageParser extends StreamingMessageParser {
       const enhancedInput = this._detectAndWrapCodeBlocks(messageId, input);
 
       if (enhancedInput !== input) {
-        // Reset and reparse with enhanced input
-        this.reset();
+        // Reset only this message's state and reparse with enhanced input
+        this.resetMessage(messageId);
         output = super.parse(messageId, enhancedInput);
       }
     }
@@ -524,4 +524,9 @@ ${content.trim()}
     this._processedCodeBlocks.clear();
     this._artifactCounter = 0;
   }
+
+  resetMessage(messageId: string) {
+    super.resetMessage(messageId);
+    this._processedCodeBlocks.delete(messageId);
+  }
 }

app/lib/runtime/message-parser.ts

@@ -335,6 +335,10 @@ export class StreamingMessageParser {
     this.#messages.clear();
   }
 
+  resetMessage(messageId: string) {
+    this.#messages.delete(messageId);
+  }
+
   #parseActionTag(input: string, actionOpenIndex: number, actionEndIndex: number) {
     const actionTag = input.slice(actionOpenIndex, actionEndIndex + 1);