Improve workflow reconnection handling

Separate workflow completion from abortion to preserve reconnection ability.
When a workflow is aborted or errors, keep the in-progress flag so users can
reconnect on page reload. Only clear the flag when workflow completes successfully.

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <[email protected]>

Author: MH4GF

frontend/apps/app/components/SessionDetailPage/hooks/useStream/useStream.ts

@@ -81,15 +81,20 @@ export const useStream = ({ designSessionId, initialMessages }: Props) => {
   const abortRef = useRef<AbortController | null>(null)
   const retryCountRef = useRef(0)
 
-  const finalizeStream = useCallback((sessionId: string) => {
+  const completeWorkflow = useCallback((sessionId: string) => {
     setIsStreaming(false)
     abortRef.current = null
     retryCountRef.current = 0
-
-    // Clear workflow in progress flag
     clearWorkflowInProgress(sessionId)
   }, [])
 
+  const abortWorkflow = useCallback(() => {
+    setIsStreaming(false)
+    abortRef.current = null
+    retryCountRef.current = 0
+    // Do NOT clear workflow flag - allow reconnection
+  }, [])
+
   const stop = useCallback(() => {
     abortRef.current?.abort()
     abortRef.current = null
@@ -180,7 +185,7 @@ export const useStream = ({ designSessionId, initialMessages }: Props) => {
         })
 
         if (!res.body) {
-          finalizeStream(params.designSessionId)
+          abortWorkflow()
           return err({
             type: 'network',
             message: ERROR_MESSAGES.FETCH_FAILED,
@@ -192,7 +197,7 @@ export const useStream = ({ designSessionId, initialMessages }: Props) => {
 
         if (!endEventReceived) {
           if (controller.signal.aborted) {
-            finalizeStream(params.designSessionId)
+            abortWorkflow()
             return err({
               type: 'abort',
               message: 'Request was aborted',
@@ -204,10 +209,10 @@ export const useStream = ({ designSessionId, initialMessages }: Props) => {
           return ok('shouldRetry')
         }
 
-        finalizeStream(params.designSessionId)
+        completeWorkflow(params.designSessionId)
         return ok('complete')
       } catch (unknownError) {
-        finalizeStream(params.designSessionId)
+        abortWorkflow()
 
         if (
           unknownError instanceof Error &&
@@ -225,7 +230,7 @@ export const useStream = ({ designSessionId, initialMessages }: Props) => {
         })
       }
     },
-    [finalizeStream, processStreamEvents],
+    [completeWorkflow, abortWorkflow, processStreamEvents],
   )
 
   const replay = useCallback(
@@ -245,14 +250,14 @@ export const useStream = ({ designSessionId, initialMessages }: Props) => {
       }
 
       const timeoutMessage = ERROR_MESSAGES.CONNECTION_TIMEOUT
-      finalizeStream(params.designSessionId)
+      abortWorkflow()
       setError(timeoutMessage)
       return err({
         type: 'timeout',
         message: timeoutMessage,
       })
     },
-    [finalizeStream, runStreamAttempt],
+    [abortWorkflow, runStreamAttempt],
   )
 
   const start = useCallback(

frontend/apps/app/components/SessionDetailPage/utils/workflowStorage.ts

@@ -6,7 +6,8 @@ const WORKFLOW_KEY_PREFIX = 'liam:workflow:'
 export const getWorkflowInProgress = (designSessionId: string): boolean => {
   if (typeof window === 'undefined') return false
   const key = `${WORKFLOW_KEY_PREFIX}${designSessionId}`
-  return sessionStorage.getItem(key) === 'in_progress'
+  const value = sessionStorage.getItem(key)
+  return value === 'in_progress'
 }
 
 /**