@codex-infinity/pi-infinity 0.63.3 → 0.64.1

package/dist/modes/interactive/interactive-mode.js

@@ -69,6 +69,8 @@ export class InteractiveMode {
         this.loadingAnimation = undefined;
         this.pendingWorkingMessage = undefined;
         this.defaultWorkingMessage = "Working...";
+        this.defaultHiddenThinkingLabel = "Thinking...";
+        this.hiddenThinkingLabel = this.defaultHiddenThinkingLabel;
         this.lastSigintTime = 0;
         this.lastEscapeTime = 0;
         this.changelogMarkdown = undefined;
@@ -1014,6 +1016,18 @@ export class InteractiveMode {
         this.footerDataProvider.setExtensionStatus(key, text);
         this.ui.requestRender();
     }
+    setHiddenThinkingLabel(label) {
+        this.hiddenThinkingLabel = label ?? this.defaultHiddenThinkingLabel;
+        for (const child of this.chatContainer.children) {
+            if (child instanceof AssistantMessageComponent) {
+                child.setHiddenThinkingLabel(this.hiddenThinkingLabel);
+            }
+        }
+        if (this.streamingComponent) {
+            this.streamingComponent.setHiddenThinkingLabel(this.hiddenThinkingLabel);
+        }
+        this.ui.requestRender();
+    }
     /**
      * Set an extension widget (string array or custom component).
      */
@@ -1085,6 +1099,7 @@ export class InteractiveMode {
         if (this.loadingAnimation) {
             this.loadingAnimation.setMessage(`${this.defaultWorkingMessage} (${keyText("app.interrupt")} to interrupt)`);
         }
+        this.setHiddenThinkingLabel();
     }
     // Maximum total widget lines to prevent viewport overflow
     static { this.MAX_WIDGET_LINES = 10; }
@@ -1214,6 +1229,7 @@ export class InteractiveMode {
                     this.pendingWorkingMessage = message;
                 }
             },
+            setHiddenThinkingLabel: (label) => this.setHiddenThinkingLabel(label),
             setWidget: (key, content, options) => this.setExtensionWidget(key, content, options),
             setFooter: (factory) => this.setExtensionFooter(factory),
             setHeader: (factory) => this.setExtensionHeader(factory),
@@ -1817,6 +1833,10 @@ export class InteractiveMode {
                 }
                 this.ui.requestRender();
                 break;
+            case "queue_update":
+                this.updatePendingMessagesDisplay();
+                this.ui.requestRender();
+                break;
             case "message_start":
                 if (event.message.role === "custom") {
                     this.addMessageToChat(event.message);
@@ -1828,7 +1848,7 @@ export class InteractiveMode {
                     this.ui.requestRender();
                 }
                 else if (event.message.role === "assistant") {
-                    this.streamingComponent = new AssistantMessageComponent(undefined, this.hideThinkingBlock, this.getMarkdownThemeWithSettings());
+                    this.streamingComponent = new AssistantMessageComponent(undefined, this.hideThinkingBlock, this.getMarkdownThemeWithSettings(), this.hiddenThinkingLabel);
                     this.streamingMessage = event.message;
                     this.chatContainer.addChild(this.streamingComponent);
                     this.streamingComponent.updateContent(this.streamingMessage);
@@ -2126,7 +2146,7 @@ export class InteractiveMode {
                 break;
             }
             case "assistant": {
-                const assistantComponent = new AssistantMessageComponent(message, this.hideThinkingBlock, this.getMarkdownThemeWithSettings());
+                const assistantComponent = new AssistantMessageComponent(message, this.hideThinkingBlock, this.getMarkdownThemeWithSettings(), this.hiddenThinkingLabel);
                 this.chatContainer.addChild(assistantComponent);
                 break;
             }
@@ -2722,7 +2742,7 @@ export class InteractiveMode {
                 },
                 onTransportChange: (transport) => {
                     this.settingsManager.setTransport(transport);
-                    this.session.agent.setTransport(transport);
+                    this.session.agent.transport = transport;
                 },
                 onThinkingLevelChange: (level) => {
                     this.session.setThinkingLevel(level);