Skip polling loop in reconnectToServiceLogs when service already stopped

reconnectIfNeeded now captures the service status it fetches rather than
discarding it. If the service is already stopped, we pass serviceAlreadyStopped
to reconnectToServiceLogs so it exits after a single log fetch instead of
making a redundant status API call and looping.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

Author: mcintyre94

Wisp/ViewModels/ChatViewModel.swift

@@ -594,7 +594,7 @@ final class ChatViewModel {
         // Cancel the stale stream and reconnect via service logs
         streamTask?.cancel()
         streamTask = Task {
-            await reconnectToServiceLogs(apiClient: apiClient, modelContext: modelContext, serviceIsRunning: true)
+            await reconnectToServiceLogs(apiClient: apiClient, modelContext: modelContext, serviceAlreadyStopped: false)
         }
     }
 
@@ -650,8 +650,8 @@ final class ChatViewModel {
             guard let serviceInfo = try? await apiClient.getServiceStatus(spriteName: spriteName, serviceName: serviceName)
             else { return }
 
-            let isRunning = serviceInfo.state.status == "running"
-            await reconnectToServiceLogs(apiClient: apiClient, modelContext: modelContext, serviceIsRunning: isRunning)
+            let alreadyStopped = serviceInfo.state.status != "running"
+            await reconnectToServiceLogs(apiClient: apiClient, modelContext: modelContext, serviceAlreadyStopped: alreadyStopped)
         }
     }
 
@@ -796,7 +796,7 @@ final class ChatViewModel {
         // Attempt reconnection on disconnect
         if case .disconnected = streamResult {
             logger.info("[Chat] Disconnected mid-stream, attempting reconnect via service logs")
-            await reconnectToServiceLogs(apiClient: apiClient, modelContext: modelContext, serviceIsRunning: true)
+            await reconnectToServiceLogs(apiClient: apiClient, modelContext: modelContext, serviceAlreadyStopped: false)
             return
         }
 
@@ -1041,10 +1041,10 @@ final class ChatViewModel {
     func runReconnectLoop(
         apiClient: some ServiceLogsProvider,
         modelContext: ModelContext,
-        serviceIsRunning: Bool = false
+        serviceAlreadyStopped: Bool = false
     ) async {
         isReplaying = true
-        isReplayingLiveService = serviceIsRunning
+        isReplayingLiveService = !serviceAlreadyStopped
         defer {
             applyReplayBuffer()
             isReplaying = false
@@ -1111,10 +1111,12 @@ final class ChatViewModel {
             // If we got a result event, Claude is done
             if receivedResultEvent { break }
 
-            // Check if service is still running
-            let isRunning = (try? await apiClient.getServiceStatus(spriteName: spriteName, serviceName: serviceName))?.state.status == "running"
+            // If we already knew the service was stopped before entering, no need to re-check
+            if serviceAlreadyStopped { break }
 
-            if isRunning {
+            // Check if service is still running before retrying
+            if let serviceInfo = try? await apiClient.getServiceStatus(spriteName: spriteName, serviceName: serviceName),
+               serviceInfo.state.status == "running" {
                 logger.info("[Chat] Service still running, will re-poll after delay")
                 try? await Task.sleep(for: .seconds(2))
                 continue
@@ -1150,9 +1152,9 @@ final class ChatViewModel {
     private func reconnectToServiceLogs(
         apiClient: SpritesAPIClient,
         modelContext: ModelContext,
-        serviceIsRunning: Bool = false
+        serviceAlreadyStopped: Bool = false
     ) async {
-        await runReconnectLoop(apiClient: apiClient, modelContext: modelContext, serviceIsRunning: serviceIsRunning)
+        await runReconnectLoop(apiClient: apiClient, modelContext: modelContext, serviceAlreadyStopped: serviceAlreadyStopped)
 
         if let queued = queuedPrompt, !Task.isCancelled {
             let prompt = buildPrompt(text: queued, attachments: queuedAttachments)

WispTests/ChatViewModelTests.swift

@@ -1046,7 +1046,7 @@ struct ChatViewModelTests {
     // MARK: - Live service reconnect
 
     @Test func runReconnectLoop_liveService_appliesContentCorrectly() async throws {
-        // When serviceIsRunning is true, isReplaying is cleared on the first new event
+        // When serviceAlreadyStopped is false (service is live), isReplaying is cleared on the first new event
         // so events render incrementally rather than being batched. The final content
         // should be identical to the batched path.
         let ctx = try makeModelContext()
@@ -1068,7 +1068,7 @@ struct ChatViewModelTests {
         }
         let mock = MockServiceLogsProvider(streams: [stream], statuses: ["stopped"])
 
-        await vm.runReconnectLoop(apiClient: mock, modelContext: ctx, serviceIsRunning: true)
+        await vm.runReconnectLoop(apiClient: mock, modelContext: ctx, serviceAlreadyStopped: false)
 
         #expect(assistantMsg.content.count == 1)
         if case .text(let text) = assistantMsg.content.first {
@@ -1106,7 +1106,7 @@ struct ChatViewModelTests {
         // The text and result events are new and should be processed via the live path.
         vm.processedEventUUIDs = ["s1-system"]
 
-        await vm.runReconnectLoop(apiClient: mock, modelContext: ctx, serviceIsRunning: true)
+        await vm.runReconnectLoop(apiClient: mock, modelContext: ctx, serviceAlreadyStopped: false)
 
         // New content lands even though a historical event was skipped first
         #expect(assistantMsg.content.count == 1)