fix(vscode): forward session.updated SSE events so title reactively updates in TaskHeader

packages/kilo-vscode/src/KiloProvider.ts

@@ -820,6 +820,17 @@ export class KiloProvider implements vscode.WebviewViewProvider {
           session: this.sessionToWebview(event.properties.info),
         })
         break
+
+      case "session.updated":
+        // Keep local state in sync (e.g. title generation)
+        if (this.currentSession?.id === event.properties.info.id) {
+          this.currentSession = event.properties.info
+        }
+        this.postMessage({
+          type: "sessionUpdated",
+          session: this.sessionToWebview(event.properties.info),
+        })
+        break
     }
   }
 

packages/kilo-vscode/webview-ui/src/context/session.tsx

@@ -239,6 +239,10 @@ export const SessionProvider: ParentComponent = (props) => {
         case "sessionsLoaded":
           handleSessionsLoaded(message.sessions)
           break
+
+        case "sessionUpdated":
+          setStore("sessions", message.session.id, message.session)
+          break
       }
     })
 

packages/kilo-vscode/webview-ui/src/types/messages.ts

@@ -233,6 +233,11 @@ export interface SessionCreatedMessage {
   session: SessionInfo
 }
 
+export interface SessionUpdatedMessage {
+  type: "sessionUpdated"
+  session: SessionInfo
+}
+
 export interface MessagesLoadedMessage {
   type: "messagesLoaded"
   sessionID: string
@@ -302,6 +307,7 @@ export type ExtensionMessage =
   | PermissionRequestMessage
   | TodoUpdatedMessage
   | SessionCreatedMessage
+  | SessionUpdatedMessage
   | MessagesLoadedMessage
   | MessageCreatedMessage
   | SessionsLoadedMessage