mu-harness 0.23.0 → 0.24.0

package/esm/channels/ws/client.js

@@ -38,6 +38,7 @@ export async function connectHarness(opts) {
     const subSessions = new Map();
     const approvalListeners = new Set();
     const subAgentListeners = new Set();
+    const modelLoadingListeners = new Set();
     const forkWaiters = new Map();
     const subagentWaiters = new Map();
     const listWaiters = [];
@@ -144,6 +145,10 @@ export async function connectHarness(opts) {
                 features.vision = frame.vision;
                 features.audio = frame.audio;
                 return;
+            case 'model_loading':
+                for (const l of [...modelLoadingListeners])
+                    l(frame.model, frame.loading);
+                return;
             case 'models:listed': {
                 models = frame.models.map((m) => ({ id: m.id, ownedBy: m.ownedBy }));
                 modelSelected = frame.selected;
@@ -300,6 +305,10 @@ export async function connectHarness(opts) {
         saveThinking: opts.saveThinking ?? (() => { }),
         history: opts.history,
         features,
+        subscribeModelLoading: (listener) => {
+            modelLoadingListeners.add(listener);
+            return () => modelLoadingListeners.delete(listener);
+        },
         banner: opts.banner,
         minimal: opts.minimal,
         commands: () => commands.map((c) => ({ name: c.command.replace(/^\//, ''), description: c.description })),

package/esm/channels/ws/protocol.d.ts

@@ -168,6 +168,10 @@ export type WsOutbound = {
     type: 'capabilities';
     vision: boolean;
     audio: boolean;
+} | {
+    type: 'model_loading';
+    model: string;
+    loading: boolean;
 } | {
     type: 'agents';
     agents: WireAgent[];

package/esm/channels/ws/server.js

@@ -164,12 +164,19 @@ export function webSocketAdapter(opts) {
                     const frame = await modelsFrame();
                     if (frame)
                         push(frame);
-                    // Detect the new model's input modalities (loads the model) and re-advertise.
-                    const modalities = await harness.models.capabilities(msg.ref).catch(() => undefined);
-                    if (modalities) {
-                        caps.vision = modalities.vision;
-                        caps.audio = modalities.audio;
-                        push({ type: 'capabilities', vision: caps.vision, audio: caps.audio });
+                    // Detecting the new model's modalities loads it (a /props round-trip can be a
+                    // 10-30s cold start) — surface that as a loading state to every channel.
+                    push({ type: 'model_loading', model: msg.ref, loading: true });
+                    try {
+                        const modalities = await harness.models.capabilities(msg.ref).catch(() => undefined);
+                        if (modalities) {
+                            caps.vision = modalities.vision;
+                            caps.audio = modalities.audio;
+                            push({ type: 'capabilities', vision: caps.vision, audio: caps.audio });
+                        }
+                    }
+                    finally {
+                        push({ type: 'model_loading', model: msg.ref, loading: false });
                     }
                     return;
                 }

package/esm/tui/chat/ChatApp.d.ts

@@ -41,6 +41,8 @@ export interface ChatHost {
         append(text: string): void;
     };
     features?: ChatFeatures;
+    /** Subscribe to model load/unload state (cold-start). Optional — only WS-backed hosts emit it. */
+    subscribeModelLoading?(listener: (model: string, loading: boolean) => void): () => void;
     banner?: string;
     minimal?: boolean;
     commands?(): {
@@ -72,6 +74,7 @@ export declare class ChatApp {
     private unsubscribe;
     private unsubscribeTheme;
     private unsubscribeSubAgents;
+    private unsubscribeModelLoading;
     private readonly runUnsubs;
     private readonly activeRuns;
     private mentionAc;
@@ -120,6 +123,7 @@ export declare class ChatApp {
     private tryDispatch;
     private dispatchMention;
     private swapSession;
+    private onModelLoading;
     private handleEvent;
     private applyUsage;
     private onTurnComplete;

package/esm/tui/chat/ChatApp.js

@@ -99,6 +99,7 @@ export class ChatApp {
     unsubscribe;
     unsubscribeTheme;
     unsubscribeSubAgents;
+    unsubscribeModelLoading;
     runUnsubs = new Set();
     activeRuns = new Set();
     mentionAc;
@@ -194,6 +195,7 @@ export class ChatApp {
             this.tui.setToastForeground(this.theme().colors.text);
             this.tui.requestRender(true);
         });
+        this.unsubscribeModelLoading = this.host.subscribeModelLoading?.((model, loading) => this.onModelLoading(model, loading));
         this.bindSession();
         if (this.feature('subAgents')) {
             this.unsubscribeSubAgents = this.host.subAgents.subscribe((run) => this.onSubAgentRun(run));
@@ -219,6 +221,7 @@ export class ChatApp {
         this.unsubscribeTheme?.();
         this.unsubscribeSubAgents?.();
         this.unsubscribeApproval?.();
+        this.unsubscribeModelLoading?.();
         this.clearRuns();
         this.stopSpinner();
         this.clearError();
@@ -431,6 +434,21 @@ export class ChatApp {
         this.setStatus('ready');
         this.tui.requestRender(true);
     }
+    onModelLoading(model, loading) {
+        const name = model.split('/').pop() ?? model;
+        if (loading) {
+            this.status.busy = true;
+            this.setStatus(`loading ${name}…`);
+            this.startSpinner();
+        }
+        else if (!this.running) {
+            // Don't clobber an in-flight turn's status when the load finishes.
+            this.status.busy = false;
+            this.stopSpinner();
+            this.setStatus('ready');
+        }
+        this.tui.requestRender();
+    }
     handleEvent(event) {
         this.updateSpeaker();
         this.transcript.applyEvent(event);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "mu-harness",
-  "version": "0.23.0",
+  "version": "0.24.0",
   "description": "Agent harness: createHarness wires mu-core into a host — XDG paths, model registry, plugins, disk-loaded agents & skills, sub-agents, sessions (JSONL + SQLite catalog), slash commands, permission/approval hooks, an optional scheduler, and a composable TUI chat app",
   "license": "MIT",
   "main": "./script/index.js",
@@ -23,8 +23,8 @@
     "@swc/wasm-typescript": "^1.15.0",
     "cli-highlight": "^2.1.11",
     "croner": "^9.0.0",
-    "mu-core": "^0.23.0",
-    "mu-tui": "^0.23.0",
+    "mu-core": "^0.24.0",
+    "mu-tui": "^0.24.0",
     "ws": "^8.18.0"
   },
   "_generatedBy": "dnt@dev",

package/script/channels/ws/client.js

@@ -41,6 +41,7 @@ async function connectHarness(opts) {
     const subSessions = new Map();
     const approvalListeners = new Set();
     const subAgentListeners = new Set();
+    const modelLoadingListeners = new Set();
     const forkWaiters = new Map();
     const subagentWaiters = new Map();
     const listWaiters = [];
@@ -147,6 +148,10 @@ async function connectHarness(opts) {
                 features.vision = frame.vision;
                 features.audio = frame.audio;
                 return;
+            case 'model_loading':
+                for (const l of [...modelLoadingListeners])
+                    l(frame.model, frame.loading);
+                return;
             case 'models:listed': {
                 models = frame.models.map((m) => ({ id: m.id, ownedBy: m.ownedBy }));
                 modelSelected = frame.selected;
@@ -303,6 +308,10 @@ async function connectHarness(opts) {
         saveThinking: opts.saveThinking ?? (() => { }),
         history: opts.history,
         features,
+        subscribeModelLoading: (listener) => {
+            modelLoadingListeners.add(listener);
+            return () => modelLoadingListeners.delete(listener);
+        },
         banner: opts.banner,
         minimal: opts.minimal,
         commands: () => commands.map((c) => ({ name: c.command.replace(/^\//, ''), description: c.description })),

package/script/channels/ws/protocol.d.ts

@@ -168,6 +168,10 @@ export type WsOutbound = {
     type: 'capabilities';
     vision: boolean;
     audio: boolean;
+} | {
+    type: 'model_loading';
+    model: string;
+    loading: boolean;
 } | {
     type: 'agents';
     agents: WireAgent[];

package/script/channels/ws/server.js

@@ -167,12 +167,19 @@ function webSocketAdapter(opts) {
                     const frame = await modelsFrame();
                     if (frame)
                         push(frame);
-                    // Detect the new model's input modalities (loads the model) and re-advertise.
-                    const modalities = await harness.models.capabilities(msg.ref).catch(() => undefined);
-                    if (modalities) {
-                        caps.vision = modalities.vision;
-                        caps.audio = modalities.audio;
-                        push({ type: 'capabilities', vision: caps.vision, audio: caps.audio });
+                    // Detecting the new model's modalities loads it (a /props round-trip can be a
+                    // 10-30s cold start) — surface that as a loading state to every channel.
+                    push({ type: 'model_loading', model: msg.ref, loading: true });
+                    try {
+                        const modalities = await harness.models.capabilities(msg.ref).catch(() => undefined);
+                        if (modalities) {
+                            caps.vision = modalities.vision;
+                            caps.audio = modalities.audio;
+                            push({ type: 'capabilities', vision: caps.vision, audio: caps.audio });
+                        }
+                    }
+                    finally {
+                        push({ type: 'model_loading', model: msg.ref, loading: false });
                     }
                     return;
                 }

package/script/tui/chat/ChatApp.d.ts

@@ -41,6 +41,8 @@ export interface ChatHost {
         append(text: string): void;
     };
     features?: ChatFeatures;
+    /** Subscribe to model load/unload state (cold-start). Optional — only WS-backed hosts emit it. */
+    subscribeModelLoading?(listener: (model: string, loading: boolean) => void): () => void;
     banner?: string;
     minimal?: boolean;
     commands?(): {
@@ -72,6 +74,7 @@ export declare class ChatApp {
     private unsubscribe;
     private unsubscribeTheme;
     private unsubscribeSubAgents;
+    private unsubscribeModelLoading;
     private readonly runUnsubs;
     private readonly activeRuns;
     private mentionAc;
@@ -120,6 +123,7 @@ export declare class ChatApp {
     private tryDispatch;
     private dispatchMention;
     private swapSession;
+    private onModelLoading;
     private handleEvent;
     private applyUsage;
     private onTurnComplete;

package/script/tui/chat/ChatApp.js

@@ -102,6 +102,7 @@ class ChatApp {
     unsubscribe;
     unsubscribeTheme;
     unsubscribeSubAgents;
+    unsubscribeModelLoading;
     runUnsubs = new Set();
     activeRuns = new Set();
     mentionAc;
@@ -197,6 +198,7 @@ class ChatApp {
             this.tui.setToastForeground(this.theme().colors.text);
             this.tui.requestRender(true);
         });
+        this.unsubscribeModelLoading = this.host.subscribeModelLoading?.((model, loading) => this.onModelLoading(model, loading));
         this.bindSession();
         if (this.feature('subAgents')) {
             this.unsubscribeSubAgents = this.host.subAgents.subscribe((run) => this.onSubAgentRun(run));
@@ -222,6 +224,7 @@ class ChatApp {
         this.unsubscribeTheme?.();
         this.unsubscribeSubAgents?.();
         this.unsubscribeApproval?.();
+        this.unsubscribeModelLoading?.();
         this.clearRuns();
         this.stopSpinner();
         this.clearError();
@@ -434,6 +437,21 @@ class ChatApp {
         this.setStatus('ready');
         this.tui.requestRender(true);
     }
+    onModelLoading(model, loading) {
+        const name = model.split('/').pop() ?? model;
+        if (loading) {
+            this.status.busy = true;
+            this.setStatus(`loading ${name}…`);
+            this.startSpinner();
+        }
+        else if (!this.running) {
+            // Don't clobber an in-flight turn's status when the load finishes.
+            this.status.busy = false;
+            this.stopSpinner();
+            this.setStatus('ready');
+        }
+        this.tui.requestRender();
+    }
     handleEvent(event) {
         this.updateSpeaker();
         this.transcript.applyEvent(event);