@steerable/agent-ui 0.7.4 → 0.8.0

package/dist/hooks/useChatStream.js

@@ -48,6 +48,21 @@ function reducer(state, action) {
             next[idx] = patched;
             return { messages: next };
         }
+        case 'replace-content': {
+            // A retried attempt restarts the reply, so the accumulated text is no
+            // longer an append target: replacing it is the only way the live view
+            // can match what the backend will persist.
+            const idx = findLastAssistantIndex(state.messages);
+            if (idx === -1)
+                return state;
+            const patched = {
+                ...state.messages[idx],
+                content: action.content,
+            };
+            const next = state.messages.slice();
+            next[idx] = patched;
+            return { messages: next };
+        }
         case 'append-tool-call': {
             const idx = findLastAssistantIndex(state.messages);
             if (idx === -1)
@@ -202,6 +217,18 @@ export function useChatStream(options) {
                 return;
             }
             case 'agent':
+                // `stream_restart`: the backend discarded the attempt it was
+                // streaming and started the request again. It carries the
+                // authoritative replacement text rather than a length to cut, so
+                // the live view cannot drift from the persisted reply. Handled here
+                // because the text lives in this hook's state; still forwarded, as
+                // every other `agent` event is.
+                if (event.event === 'stream_restart') {
+                    const content = typeof event.payload?.content === 'string' ? event.payload.content : '';
+                    dispatch({ type: 'replace-content', content });
+                }
+                options.onUnknownEvent?.(event);
+                return;
             case 'orchestration':
             case 'loader-hint':
             case 'keepalive':

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@steerable/agent-ui",
-  "version": "0.7.4",
+  "version": "0.8.0",
   "description": "Steerable framework Tier 4 — headless React hooks (useAgentSession, useChatStream, useToolCallStatus) and Tailwind-themed components (ChatPanel, MessageList, OrchestrationPlanCard, ToolCallRenderer, SSEStreamView) for embedding Steerable agents in any React app.",
   "license": "Apache-2.0",
   "homepage": "https://steerableframework.com/storybook/",
@@ -57,7 +57,7 @@
     "README.md"
   ],
   "dependencies": {
-    "@steerable/agent-protocol": "0.7.4"
+    "@steerable/agent-protocol": "0.8.0"
   },
   "peerDependencies": {
     "react": ">=18.0.0",

package/src/hooks/useChatStream.test.ts

@@ -185,6 +185,33 @@ describe('useChatStream', () => {
     expect(onUnknownEvent.mock.calls[1]?.[0].type).toBe('agent');
   });
 
+  it('stream_restart replaces the streamed text instead of appending to it', async () => {
+    // A retry after partial output discards the abandoned attempt. The host
+    // has already handed those deltas over, so the only way the live view can
+    // match the persisted reply is to take the replacement text.
+    const { transport } = makeTransport([
+      [
+        { type: 'content', content: 'half a sen' },
+        { type: 'agent', event: 'stream_restart', payload: { content: '' } },
+        { type: 'content', content: 'complete reply' },
+        { type: 'done' },
+      ],
+    ]);
+    const onUnknownEvent = vi.fn();
+    const { result } = renderHook(() => useChatStream({ transport, onUnknownEvent }));
+
+    await act(async () => {
+      await result.current.sendUserMessage({ content: 'go' });
+    });
+
+    await waitFor(() => {
+      const last = result.current.messages.at(-1);
+      expect(last?.content).toBe('complete reply');
+    });
+    // Still forwarded: callers that track agent events keep seeing them.
+    expect(onUnknownEvent.mock.calls.some(([e]) => e.event === 'stream_restart')).toBe(true);
+  });
+
   it('emits an error overlay onto the assistant message when the stream throws', async () => {
     const transport: ChatStreamTransport = {
       stream: async () => {

package/src/hooks/useChatStream.ts

@@ -150,6 +150,7 @@ type Action =
   | { type: 'insert-before-last-assistant'; message: ChatMessage }
   | { type: 'patch-last-assistant'; patch: Partial<ChatMessage> }
   | { type: 'append-content'; delta: string }
+  | { type: 'replace-content'; content: string }
   | { type: 'append-tool-call'; call: ToolCall }
   | { type: 'attach-tool-result'; result: ToolResult }
   | { type: 'finalize-assistant' };
@@ -194,6 +195,20 @@ function reducer(state: State, action: Action): State {
       next[idx] = patched;
       return { messages: next };
     }
+    case 'replace-content': {
+      // A retried attempt restarts the reply, so the accumulated text is no
+      // longer an append target: replacing it is the only way the live view
+      // can match what the backend will persist.
+      const idx = findLastAssistantIndex(state.messages);
+      if (idx === -1) return state;
+      const patched: ChatMessage = {
+        ...state.messages[idx],
+        content: action.content,
+      } as ChatMessage;
+      const next = state.messages.slice();
+      next[idx] = patched;
+      return { messages: next };
+    }
     case 'append-tool-call': {
       const idx = findLastAssistantIndex(state.messages);
       if (idx === -1) return state;
@@ -354,6 +369,19 @@ export function useChatStream(
           return;
         }
         case 'agent':
+          // `stream_restart`: the backend discarded the attempt it was
+          // streaming and started the request again. It carries the
+          // authoritative replacement text rather than a length to cut, so
+          // the live view cannot drift from the persisted reply. Handled here
+          // because the text lives in this hook's state; still forwarded, as
+          // every other `agent` event is.
+          if (event.event === 'stream_restart') {
+            const content =
+              typeof event.payload?.content === 'string' ? event.payload.content : '';
+            dispatch({ type: 'replace-content', content });
+          }
+          options.onUnknownEvent?.(event);
+          return;
         case 'orchestration':
         case 'loader-hint':
         case 'keepalive':