@design.estate/dees-catalog 6.7.1 → 6.8.1

package/ts_web/elements/00group-harness/dees-harness-message/dees-harness-message.ts

@@ -16,6 +16,7 @@ import type { HLJSApi } from 'highlight.js';
 import type {
   IHarnessAttachment,
   IHarnessMessage,
+  IHarnessMessageViewState,
   IHarnessUsage,
   THarnessMessageStreamDelta,
 } from '../interfaces.js';
@@ -73,6 +74,9 @@ export class DeesHarnessMessage extends DeesElement {
   @property({ attribute: false })
   accessor suppressEntry: boolean = false;
 
+  @property({ attribute: false })
+  accessor viewState: IHarnessMessageViewState | undefined = undefined;
+
   @state()
   accessor renderedMarkdownHtml: string = '';
 
@@ -446,6 +450,7 @@ export class DeesHarnessMessage extends DeesElement {
           .call=${message.toolCall}
           .toolRegistry=${this.toolRegistry}
           .allowSubtaskStreams=${this.allowSubtaskStreams}
+          .viewState=${this.viewState}
         ></dees-harness-tool-card>
       `;
     }
@@ -478,7 +483,10 @@ export class DeesHarnessMessage extends DeesElement {
         ${message.reasoning?.length
           ? html`
               <div class="reasoningStack">
-                <dees-harness-reasoning .reasoningParts=${message.reasoning}></dees-harness-reasoning>
+                <dees-harness-reasoning
+                  .reasoningParts=${message.reasoning}
+                  .viewState=${this.viewState}
+                ></dees-harness-reasoning>
               </div>
             `
           : ''}
@@ -488,6 +496,7 @@ export class DeesHarnessMessage extends DeesElement {
               .call=${message.toolCall}
               .toolRegistry=${this.toolRegistry}
               .allowSubtaskStreams=${this.allowSubtaskStreams}
+              .viewState=${this.viewState}
             ></dees-harness-tool-card>`
           : ''}
         ${this.renderText()}
@@ -566,6 +575,19 @@ export class DeesHarnessMessage extends DeesElement {
   public async updated(changedProperties: Map<string, any>): Promise<void> {
     super.updated(changedProperties);
     if (!this.markdownActive || !this.message.text) return;
+    if (
+      this.viewState?.markdownText === this.message.text
+      && this.viewState.markdownHtml
+      && this.lastParsedText !== this.message.text
+    ) {
+      this.lastParsedText = this.message.text;
+      this.renderedMarkdownHtml = this.viewState.markdownHtml;
+      if (!this.message.streaming) {
+        await this.updateComplete;
+        await this.highlightCodeFences(this.markdownParseToken);
+      }
+      return;
+    }
     if (this.message.text === this.lastParsedText) return;
     if (this.message.streaming) {
       if (!this.markdownWhileStreaming) return;
@@ -590,6 +612,10 @@ export class DeesHarnessMessage extends DeesElement {
     if (token !== this.markdownParseToken) return;
     this.lastParsedText = textAtRequest;
     this.renderedMarkdownHtml = htmlResult;
+    if (this.viewState) {
+      this.viewState.markdownText = textAtRequest;
+      this.viewState.markdownHtml = htmlResult;
+    }
     if (!this.message.streaming) {
       await this.updateComplete;
       await this.highlightCodeFences(token);
@@ -624,6 +650,7 @@ export class DeesHarnessMessage extends DeesElement {
   }
 
   public async disconnectedCallback(): Promise<void> {
+    this.markdownParseToken += 1;
     if (this.markdownThrottleTimeout) {
       clearTimeout(this.markdownThrottleTimeout);
       this.markdownThrottleTimeout = null;