fix: stream completion

Author: maticzav

src/lib/stream.ts

@@ -8,8 +8,8 @@ export type BrowserState = Readonly<{
   sessionId: string;
 
   liveUrl: string | null;
-
   steps: ReadonlyArray<TaskStepView>;
+  doneOutput: string | null;
 }> | null;
 
 type BrowserAction = {
@@ -24,12 +24,14 @@ export function reducer(state: BrowserState, action: BrowserAction): [BrowserSta
 
       if (state == null) {
         const liveUrl = action.status.sessionLiveUrl ?? null;
+        const doneOutput = action.status.doneOutput ?? null;
 
         const state: BrowserState = {
           taskId: action.status.id,
           sessionId: action.status.sessionId,
           liveUrl: liveUrl,
           steps: action.status.steps,
+          doneOutput: doneOutput,
         };
 
         return [state, action.status];
@@ -38,6 +40,7 @@ export function reducer(state: BrowserState, action: BrowserAction): [BrowserSta
       // UPDATE
 
       const liveUrl = action.status.sessionLiveUrl ?? null;
+      const doneOutput = action.status.doneOutput ?? null;
       const steps: TaskStepView[] = [...state.steps];
 
       if (action.status.steps != null) {
@@ -48,11 +51,20 @@ export function reducer(state: BrowserState, action: BrowserAction): [BrowserSta
         }
       }
 
-      const newState: BrowserState = { ...state, liveUrl, steps };
+      const newState: BrowserState = {
+        ...state,
+        liveUrl,
+        steps,
+        doneOutput,
+      };
 
       // CHANGES
 
-      if ((state.liveUrl == null && liveUrl != null) || state.steps.length !== steps.length) {
+      if (
+        (state.liveUrl == null && liveUrl != null) ||
+        state.steps.length !== steps.length ||
+        state.doneOutput != doneOutput
+      ) {
         const update: ReducerEvent = {
           ...action.status,
           steps: steps,