bloby-bot 0.21.4 → 0.21.6

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "bloby-bot",
-  "version": "0.21.4",
+  "version": "0.21.6",
   "releaseNotes": [
     "1. react router implemented",
     "2. new workspace design",

package/supervisor/chat/src/hooks/useChat.ts

@@ -37,8 +37,8 @@ export function useChat(ws: WsClient | null) {
   const [streamBuffer, setStreamBuffer] = useState('');
   const [tools, setTools] = useState<ToolActivity[]>([]);
   const loaded = useRef(false);
-  /** Count of user messages awaiting a bot response — keeps typing indicator alive across turns */
-  const pendingMessages = useRef(0);
+  /** Ref to current streamBuffer (avoids stale closures in callbacks) */
+  const streamBufferRef = useRef('');
 
   // Load current conversation from DB on mount
   useEffect(() => {
@@ -126,13 +126,18 @@ export function useChat(ws: WsClient | null) {
 
     const unsubs = [
       ws.on('bot:typing', () => {
+        console.log('[useChat] bot:typing → streaming=true');
         setStreaming(true);
         setTools([]);
       }),
       ws.on('bot:token', (data: { token: string }) => {
-        // Auto-detect new turn — if tokens arrive after a response cleared streaming
+        // Auto-detect new turn — tokens arriving means agent is working
         setStreaming(true);
-        setStreamBuffer((buf) => buf + data.token);
+        setStreamBuffer((buf) => {
+          const next = buf + data.token;
+          streamBufferRef.current = next;
+          return next;
+        });
       }),
       ws.on('bot:tool', (data: { name: string; input?: any; status?: string }) => {
         setTools((prev) => {
@@ -142,7 +147,10 @@ export function useChat(ws: WsClient | null) {
         });
       }),
       ws.on('bot:response', (data: { conversationId: string; messageId?: string; content: string }) => {
+        console.log('[useChat] bot:response — adding message bubble');
         setConversationId(data.conversationId);
+
+        // Always add as a new bubble
         setMessages((msgs) => [
           ...msgs,
           {
@@ -152,21 +160,23 @@ export function useChat(ws: WsClient | null) {
             timestamp: new Date().toISOString(),
           },
         ]);
+
         setStreamBuffer('');
+        streamBufferRef.current = '';
         setTools([]);
-        // Only stop streaming when all pending messages have been answered
-        pendingMessages.current = Math.max(0, pendingMessages.current - 1);
-        if (pendingMessages.current === 0) {
-          setStreaming(false);
-        }
+        // Don't clear streaming here — wait for bot:idle from the server
+      }),
+      ws.on('bot:idle', () => {
+        // Server confirmed agent is idle — safe to stop streaming
+        console.log('[useChat] bot:idle → streaming=false');
+        setStreaming(false);
       }),
       ws.on('bot:error', (data: { error: string }) => {
+        console.log('[useChat] bot:error');
         setStreamBuffer('');
+        streamBufferRef.current = '';
+        setStreaming(false);
         setTools([]);
-        pendingMessages.current = Math.max(0, pendingMessages.current - 1);
-        if (pendingMessages.current === 0) {
-          setStreaming(false);
-        }
         setMessages((msgs) => [
           ...msgs,
           {
@@ -201,8 +211,27 @@ export function useChat(ws: WsClient | null) {
         audioData: audioData ? (audioData.startsWith('data:') ? audioData : `data:audio/webm;base64,${audioData}`) : undefined,
         attachments: optimisticAttachments,
       };
-      setMessages((msgs) => [...msgs, userMsg]);
-      pendingMessages.current += 1;
+
+      // If bot is currently streaming, commit the partial response as a completed message
+      // so the user's new message appears BELOW the bot's in-progress text (chronological order)
+      const partialContent = streamBufferRef.current;
+      if (partialContent) {
+        console.log('[useChat] Committing partial stream buffer as message before user send');
+        setMessages((msgs) => [
+          ...msgs,
+          {
+            id: 'partial-' + Date.now(),
+            role: 'assistant' as const,
+            content: partialContent,
+            timestamp: new Date().toISOString(),
+          },
+          userMsg,
+        ]);
+        setStreamBuffer('');
+        streamBufferRef.current = '';
+      } else {
+        setMessages((msgs) => [...msgs, userMsg]);
+      }
 
       // Build WS payload
       const payload: any = { conversationId, content };
@@ -231,17 +260,17 @@ export function useChat(ws: WsClient | null) {
     ws.send('user:stop', { conversationId });
     setStreaming(false);
     setStreamBuffer('');
+    streamBufferRef.current = '';
     setTools([]);
-    pendingMessages.current = 0;
   }, [ws, conversationId]);
 
   const clearContext = useCallback(() => {
     setMessages([]);
     setConversationId(null);
     setStreamBuffer('');
+    streamBufferRef.current = '';
     setStreaming(false);
     setTools([]);
-    pendingMessages.current = 0;
     prevConvId.current = null;
     loaded.current = false;
     fetch('/api/context/clear', { method: 'POST' }).catch(() => {});

package/supervisor/index.ts

@@ -80,7 +80,7 @@ const SW_JS = `// Service worker — app-shell caching + push notifications
 //   JS/CSS modules                      → stale-while-revalidate
 //   API, WebSocket, Vite internals      → network-only (no cache)
 
-var CACHE = 'bloby-v5';
+var CACHE = 'bloby-v6';
 var HASHED_RE = new RegExp('/assets/.+-[a-zA-Z0-9]{6,}[.](js|css)$');
 
 // Precache the HTML shell on install so the cache is never empty.
@@ -1167,7 +1167,7 @@ ${!connected ? '<script>setTimeout(()=>location.reload(),4000)</script>' : ''}
                   waChunkBuf = '';
                 }
 
-                // Agent finished a turn — handle backend restart + state cleanup
+                // Agent finished a turn — handle backend restart + notify client
                 if (type === 'bot:turn-complete') {
                   log.info(`[orchestrator] ──── TURN COMPLETE ────`);
                   log.info(`[orchestrator] File tools used: ${eventData.usedFileTools}`);
@@ -1186,7 +1186,9 @@ ${!connected ? '<script>setTimeout(()=>location.reload(),4000)</script>' : ''}
                     pendingUpdate = false;
                     runDeferredUpdate();
                   }
-                  return; // don't forward to client
+                  // Tell the client the agent is idle — streaming can stop
+                  broadcastBloby('bot:idle', { conversationId: convId });
+                  return;
                 }
 
                 // Conversation ended (query loop exited)