feat(tui): refresh agents after update events

packages/tui/src/context/data.tsx

@@ -158,6 +158,9 @@ export const { use: useData, provider: DataProvider } = createSimpleContext({
             result.location.provider.refresh(event.location),
           ])
           break
+        case "agent.updated":
+          void result.location.agent.refresh(event.location)
+          break
         case "session.next.agent.switched":
           if (store.session.info[event.data.sessionID])
             setStore("session", "info", event.data.sessionID, "agent", event.data.agent)

packages/tui/src/routes/session/index.tsx

@@ -170,9 +170,22 @@ export function Session() {
   })
   onCleanup(() => setEpilogue())
   const messages = sessionMessages
+  const descendantSessionIDs = createMemo(() => {
+    if (session()?.parentID) return []
+    const sessions = data.session.list()
+    const childrenByParent = sessions.reduce((acc, item) => {
+      if (!item.parentID) return acc
+      acc.set(item.parentID, [...(acc.get(item.parentID) ?? []), item.id])
+      return acc
+    }, new Map<string, string[]>())
+    function collect(sessionID: string): string[] {
+      return (childrenByParent.get(sessionID) ?? []).flatMap((id) => [id, ...collect(id)])
+    }
+    return collect(route.sessionID)
+  })
   const permissions = createMemo(() => {
     if (session()?.parentID) return []
-    return data.session.permission.list(route.sessionID) ?? []
+    return [route.sessionID, ...descendantSessionIDs()].flatMap((sessionID) => data.session.permission.list(sessionID) ?? [])
   })
   const questions = createMemo(() => {
     if (session()?.parentID) return []
@@ -227,6 +240,12 @@ export function Session() {
   const editor = useEditorContext()
   const rows = createSessionRows(() => route.sessionID)
 
+  createEffect(
+    on(descendantSessionIDs, (sessionIDs) => {
+      void Promise.all(sessionIDs.map((sessionID) => data.session.permission.refresh(sessionID)))
+    }),
+  )
+
   createEffect(() => {
     const sessionID = route.sessionID
     void (async () => {

packages/tui/test/cli/tui/data.test.tsx

@@ -356,6 +356,53 @@ test("refreshes effective catalog data after catalog updates", async () => {
   }
 })
 
+test("refreshes agents after agent updates", async () => {
+  const events = createEventStream()
+  let requests = 0
+  const calls = createFetch((url) => {
+    if (url.pathname !== "/api/agent") return
+    requests++
+    return json({
+      location: { directory, project: { id: "proj_test", directory } },
+      data: [
+        {
+          id: requests === 1 ? "build" : "reviewer",
+          request: { headers: {}, body: {} },
+          mode: "primary",
+          hidden: false,
+          permissions: [],
+        },
+      ],
+    })
+  }, events)
+  let data!: ReturnType<typeof useData>
+
+  function Probe() {
+    data = useData()
+    return <box />
+  }
+
+  const app = await testRender(() => (
+    <TestTuiContexts>
+      <SDKProvider client={createClient(calls.fetch)} api={createApi(calls.fetch)}>
+        <ProjectProvider>
+          <DataProvider>
+            <Probe />
+          </DataProvider>
+        </ProjectProvider>
+      </SDKProvider>
+    </TestTuiContexts>
+  ))
+
+  try {
+    await wait(() => data.location.agent.list()?.[0]?.id === "build")
+    emitEvent(events, { id: "evt_agent", type: "agent.updated", data: {} })
+    await wait(() => data.location.agent.list()?.[0]?.id === "reviewer")
+  } finally {
+    app.renderer.destroy()
+  }
+})
+
 test("refreshes references after updates", async () => {
   const events = createEventStream()
   let requests = 0