Add app-wide ChatSessionManager to keep streams alive across navigation

Previously ChatViewModel was recreated on every chat switch, cancelling
the active stream and requiring a reconnect on return. ChatSessionManager
is an @observable cache keyed by chat UUID, injected app-wide, so VMs
(and their streams) survive switching chats or navigating between sprites.

switchToChat now looks up the cached VM instead of creating a new one.
The scenePhase handler resumes all cached VMs on foreground, not just the
active one. clearAllChats detaches all VMs before wiping the chat list.

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

Author: mcintyre94

Wisp.xcodeproj/project.pbxproj

@@ -14,6 +14,7 @@
 		3855BDFB5819B6D35F57ED9E /* Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = 099181E9B8BFE6F3475C5D36 /* Assets.xcassets */; };
 		3A36571818CB2EBA1ED78500 /* ToolInputDetailView.swift in Sources */ = {isa = PBXBuildFile; fileRef = AB04F3FB6C647DE1F6DC0FD0 /* ToolInputDetailView.swift */; };
 		4FE7C138A35A53B4C38BEF98 /* ClaudeStreamParser.swift in Sources */ = {isa = PBXBuildFile; fileRef = C8F1378232DC7FA71A8B5444 /* ClaudeStreamParser.swift */; };
+		AA33BB44CC55DD66EE77FF02 /* ChatSessionManager.swift in Sources */ = {isa = PBXBuildFile; fileRef = AA33BB44CC55DD66EE77FF01 /* ChatSessionManager.swift */; };
 		5A1F582DDCBB994013E4BA2E /* DashboardView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 0B99AA33329D50DEDB7343B7 /* DashboardView.swift */; };
 		5ABE241326BF34FF2E3C3E51 /* CreateSpriteSheet.swift in Sources */ = {isa = PBXBuildFile; fileRef = EA1634863D8543868BA11B02 /* CreateSpriteSheet.swift */; };
 		61CBE460AC96CBE6BF8AF0B0 /* SpriteSession.swift in Sources */ = {isa = PBXBuildFile; fileRef = D6BE6E0D68BBB35DE56600E1 /* SpriteSession.swift */; };
@@ -210,6 +211,7 @@
 		C4D5E6F7A8B90A41637E8FA1 /* CheckpointMarkerView.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = CheckpointMarkerView.swift; sourceTree = "<group>"; };
 		C8C4C3A1F2B2D57313431E50 /* Extensions.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = Extensions.swift; sourceTree = "<group>"; };
 		C8F1378232DC7FA71A8B5444 /* ClaudeStreamParser.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = ClaudeStreamParser.swift; sourceTree = "<group>"; };
+		AA33BB44CC55DD66EE77FF01 /* ChatSessionManager.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = ChatSessionManager.swift; sourceTree = "<group>"; };
 		CC11DD22EE33FF4400110002 /* ChatAttachmentButton.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = ChatAttachmentButton.swift; sourceTree = "<group>"; };
 		CC11DD22EE33FF4400110004 /* SpriteFileBrowserView.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = SpriteFileBrowserView.swift; sourceTree = "<group>"; };
 		CC11DD22EE33FF4400110005 /* FileEntryTests.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = FileEntryTests.swift; sourceTree = "<group>"; };
@@ -307,6 +309,7 @@
 			isa = PBXGroup;
 			children = (
 				AA22BB33CC44DD55EE66FF02 /* ClaudeQuestionTool.swift */,
+				AA33BB44CC55DD66EE77FF01 /* ChatSessionManager.swift */,
 				C8F1378232DC7FA71A8B5444 /* ClaudeStreamParser.swift */,
 				2554BB4DE8EC315CA48767BF /* ExecSession.swift */,
 				BB11CC22DD33EE44FF660A01 /* GitHubAPIClient.swift */,
@@ -728,6 +731,7 @@
 				D2B1EF5C6A7F8B9304C5E6F7 /* ClaudeModel.swift in Sources */,
 				2A18611EA392DA2BFFE74250 /* ClaudeStreamEvent.swift in Sources */,
 				4FE7C138A35A53B4C38BEF98 /* ClaudeStreamParser.swift in Sources */,
+				AA33BB44CC55DD66EE77FF02 /* ChatSessionManager.swift in Sources */,
 				99BA5812AF56D1C7501D1F0C /* CreateCheckpointSheet.swift in Sources */,
 				5ABE241326BF34FF2E3C3E51 /* CreateSpriteSheet.swift in Sources */,
 				5A1F582DDCBB994013E4BA2E /* DashboardView.swift in Sources */,

Wisp/App/WispApp.swift

@@ -5,6 +5,7 @@ import SwiftUI
 struct WispApp: App {
     @State private var apiClient = SpritesAPIClient()
     @State private var browserCoordinator = InAppBrowserCoordinator()
+    @State private var chatSessionManager = ChatSessionManager()
     @AppStorage("theme") private var theme: String = "system"
 
     init() {
@@ -27,6 +28,7 @@ struct WispApp: App {
             RootView()
                 .environment(apiClient)
                 .environment(browserCoordinator)
+                .environment(chatSessionManager)
                 .preferredColorScheme(preferredColorScheme)
                 .onChange(of: apiClient.isAuthenticated, initial: true) {
                     browserCoordinator.authToken = apiClient.spritesToken

Wisp/Services/ChatSessionManager.swift

@@ -0,0 +1,58 @@
+import Foundation
+import SwiftData
+
+/// App-wide cache of ChatViewModels, keyed by chat UUID.
+/// Keeps streams alive across chat switches and sprite navigation so connections
+/// are only torn down on explicit actions (interrupt, chat deletion, app termination).
+@Observable
+@MainActor
+final class ChatSessionManager {
+    private var cache: [UUID: ChatViewModel] = [:]
+
+    /// Returns the cached VM for a chat, or creates and loads a new one.
+    func viewModel(
+        for chat: SpriteChat,
+        spriteName: String,
+        apiClient: SpritesAPIClient,
+        modelContext: ModelContext
+    ) -> ChatViewModel {
+        if let existing = cache[chat.id] {
+            return existing
+        }
+        let vm = ChatViewModel(
+            spriteName: spriteName,
+            chatId: chat.id,
+            currentServiceName: chat.currentServiceName,
+            workingDirectory: chat.workingDirectory
+        )
+        vm.loadSession(apiClient: apiClient, modelContext: modelContext)
+        cache[chat.id] = vm
+        return vm
+    }
+
+    func isStreaming(chatId: UUID) -> Bool {
+        cache[chatId]?.isStreaming ?? false
+    }
+
+    /// Called when the app returns to the foreground — reconnects any VM that had a live stream.
+    func resumeAllAfterBackground(apiClient: SpritesAPIClient, modelContext: ModelContext) {
+        for vm in cache.values {
+            vm.resumeAfterBackground(apiClient: apiClient, modelContext: modelContext)
+            vm.reconnectIfNeeded(apiClient: apiClient, modelContext: modelContext)
+        }
+    }
+
+    /// Removes and detaches a VM when its chat is deleted.
+    func remove(chatId: UUID, modelContext: ModelContext) {
+        cache[chatId]?.detach(modelContext: modelContext)
+        cache.removeValue(forKey: chatId)
+    }
+
+    /// Detaches and removes all cached VMs. Call when clearing all chats.
+    func detachAll(modelContext: ModelContext) {
+        for vm in cache.values {
+            vm.detach(modelContext: modelContext)
+        }
+        cache.removeAll()
+    }
+}

Wisp/Views/SpriteDetail/SpriteDetailView.swift

@@ -9,12 +9,12 @@ struct SpriteDetailView: View {
     @State private var checkpointsViewModel: CheckpointsViewModel
     @State private var showChatSwitcher = false
     @State private var showStaleChatsAlert = false
-    @State private var knownStreamingChatIds: Set<UUID> = []
     @State private var showCopiedFeedback = false
     @State private var pendingFork: (checkpointId: String, messageId: UUID)? = nil
     @State private var isForking = false
     @State private var spriteQuickActionsViewModel: QuickActionsViewModel?
     @Environment(SpritesAPIClient.self) private var apiClient
+    @Environment(ChatSessionManager.self) private var chatSessionManager
     @Environment(\.modelContext) private var modelContext
     @Environment(\.scenePhase) private var scenePhase
     @Environment(\.horizontalSizeClass) private var sizeClass
@@ -226,8 +226,7 @@ struct SpriteDetailView: View {
         }
         .onChange(of: scenePhase) { _, newPhase in
             if newPhase == .active {
-                chatViewModel?.resumeAfterBackground(apiClient: apiClient, modelContext: modelContext)
-                chatViewModel?.reconnectIfNeeded(apiClient: apiClient, modelContext: modelContext)
+                chatSessionManager.resumeAllAfterBackground(apiClient: apiClient, modelContext: modelContext)
             }
         }
         .sheet(isPresented: $showChatSwitcher) {
@@ -241,6 +240,8 @@ struct SpriteDetailView: View {
         }
         .alert("Sprite Recreated", isPresented: $showStaleChatsAlert) {
             Button("Start Fresh", role: .destructive) {
+                chatSessionManager.detachAll(modelContext: modelContext)
+                chatViewModel = nil
                 chatListViewModel.clearAllChats(apiClient: apiClient, modelContext: modelContext)
                 let chat = chatListViewModel.createChat(modelContext: modelContext)
                 switchToChat(chat)
@@ -277,29 +278,18 @@ struct SpriteDetailView: View {
     private func switchToChat(_ chat: SpriteChat) {
         guard chatViewModel?.chatId != chat.id else { return }
 
-        // Detach old VM (cancel stream but keep service running)
-        if let oldVM = chatViewModel {
-            let wasStreaming = oldVM.detach(modelContext: modelContext)
-            if wasStreaming {
-                knownStreamingChatIds.insert(oldVM.chatId)
-            }
-        }
-
-        let vm = ChatViewModel(
+        // Look up or create a VM from the app-wide cache — old VM keeps streaming in background
+        let vm = chatSessionManager.viewModel(
+            for: chat,
             spriteName: sprite.name,
-            chatId: chat.id,
-            currentServiceName: chat.currentServiceName,
-            workingDirectory: chat.workingDirectory
+            apiClient: apiClient,
+            modelContext: modelContext
         )
-        vm.loadSession(apiClient: apiClient, modelContext: modelContext)
         chatViewModel = vm
         chatListViewModel.activeChatId = chat.id
 
-        // Always try reconnect — checks service status first, so it's
-        // cheap for old chats where the service has already stopped.
-        // This handles both switching between chats and navigating
-        // back to the sprite after the view was destroyed.
-        knownStreamingChatIds.remove(chat.id)
+        // Reconnect if idle with a service that may have new events.
+        // Guards on !isStreaming internally, so safe to call on an already-streaming VM.
         vm.reconnectIfNeeded(apiClient: apiClient, modelContext: modelContext)
     }