agentgui 1.0.940 → 1.0.942

package/site/app/js/backend.js

@@ -27,14 +27,17 @@ function withToken(url) {
   return url + (url.includes('?') ? '&' : '?') + 'token=' + encodeURIComponent(tok);
 }
 
+function lsGet(k) { try { return localStorage.getItem(k); } catch { return null; } }
+function lsSet(k, v) { try { localStorage.setItem(k, v); } catch {} }
+
 export function getBackend() {
   const u = new URL(location.href);
   const fromQs = u.searchParams.get('backend');
-  if (fromQs) { localStorage.setItem(KEY, fromQs); return fromQs; }
-  return localStorage.getItem(KEY) || DEFAULT_BACKEND;
+  if (fromQs) { lsSet(KEY, fromQs); return fromQs; }
+  return lsGet(KEY) || DEFAULT_BACKEND;
 }
 
-export function setBackend(url) { localStorage.setItem(KEY, url); }
+export function setBackend(url) { lsSet(KEY, url); }
 
 export async function probeBackend(base) {
   try {
@@ -262,25 +265,31 @@ export async function* streamChat(base, { model, messages, signal, agentId, resu
   const sessionId = started?.sessionId;
   if (!sessionId) { yield { type: 'error', error: 'no sessionId from server' }; return; }
 
+  const finish = () => { done = true; if (resolveWait) { resolveWait(); resolveWait = null; } };
+
   const unsub = addSessionListener(sessionId, (ev) => {
     if (ev.type === 'streaming_progress') {
       const block = ev.block;
       if (block?.type === 'text' && block.text) push({ type: 'text', text: block.text });
       else if (block?.type === 'tool_use') push({ type: 'tool', block });
-      else if (block?.type === 'tool_result') push({ type: 'tool', block });
+      else if (block?.type === 'tool_result') push({ type: 'tool_result', block });
       else if (block?.type === 'result') push({ type: 'result', block });
     } else if (ev.type === 'streaming_complete') {
-      done = true;
-      if (resolveWait) { resolveWait(); resolveWait = null; }
+      finish();
     } else if (ev.type === 'streaming_error') {
       errored = ev.error || 'streaming error';
-      done = true;
-      if (resolveWait) { resolveWait(); resolveWait = null; }
+      finish();
     }
   });
 
-  // Wire AbortSignal to chat.cancel.
-  const onAbort = () => { wsCall(base, 'chat.cancel', { sessionId }).catch(() => {}); };
+  // If the websocket drops mid-stream, streaming_complete will never arrive —
+  // surface an error and end the iterator instead of hanging forever.
+  const onWs = (s) => { if ((s === 'closed' || s === 'error') && !done) { errored = errored || 'connection lost during stream'; finish(); } };
+  const unsubWs = onWsStatus ? onWsStatus(onWs) : null;
+
+  // Wire AbortSignal to chat.cancel — and end the iterator immediately so the
+  // caller's busy state clears even if the server never emits a final event.
+  const onAbort = () => { wsCall(base, 'chat.cancel', { sessionId }).catch(() => {}); finish(); };
   if (signal) {
     if (signal.aborted) onAbort();
     else signal.addEventListener('abort', onAbort, { once: true });
@@ -297,6 +306,7 @@ export async function* streamChat(base, { model, messages, signal, agentId, resu
     if (errored) yield { type: 'error', error: errored };
   } finally {
     unsub();
+    if (typeof unsubWs === 'function') unsubWs();
     if (signal) signal.removeEventListener?.('abort', onAbort);
   }
 }