From 11a96f542d5c397739e8c698fbc18374c4e74aeb Mon Sep 17 00:00:00 2001 From: kirillk Date: Mon, 6 Apr 2026 12:20:18 -0400 Subject: [PATCH] fix(vscode): backfill agent selections when agentsLoaded arrives after messagesLoaded --- .../kilo-vscode/webview-ui/src/context/session.tsx | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/packages/kilo-vscode/webview-ui/src/context/session.tsx b/packages/kilo-vscode/webview-ui/src/context/session.tsx index d5ff06c459b..e7259f5e1cd 100644 --- a/packages/kilo-vscode/webview-ui/src/context/session.tsx +++ b/packages/kilo-vscode/webview-ui/src/context/session.tsx @@ -483,6 +483,17 @@ export const SessionProvider: ParentComponent = (props) => { } }), ) + + // Rescan already-loaded message history so sessions whose messagesLoaded + // arrived before agentsLoaded (and therefore got no agent selection) are + // backfilled now that we know the valid agent names. + batch(() => { + for (const [sid, msgs] of Object.entries(store.messages)) { + if (store.agentSelections[sid]) continue + const agent = resolveSessionAgent(msgs, names) + if (agent) setStore("agentSelections", sid, agent) + } + }) }) // Request agents in case the initial push was missed.