From baaff81a060c83b84b95cc537222dac7042b573f Mon Sep 17 00:00:00 2001
From: Dax Raad <d@ironbay.co>
Date: Fri, 19 Sep 2025 01:09:52 -0400
Subject: [PATCH] fix task tool ui disappearing once done

---
 packages/opencode/src/tool/task.ts | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/packages/opencode/src/tool/task.ts b/packages/opencode/src/tool/task.ts
index 163b5a2f..5875722f 100644
--- a/packages/opencode/src/tool/task.ts
+++ b/packages/opencode/src/tool/task.ts
@@ -75,10 +75,14 @@ export const TaskTool = Tool.define("task", async () => {
         ],
       })
       unsub()
+      let all
+      all = await Session.messages(session.id)
+      all = all.filter((x) => x.info.role === "assistant")
+      all = all.flatMap((msg) => msg.parts.filter((x: any) => x.type === "tool") as MessageV2.ToolPart[])
       return {
         title: params.description,
         metadata: {
-          summary: result.parts.filter((x: any) => x.type === "tool"),
+          summary: all,
         },
         output: (result.parts.findLast((x: any) => x.type === "text") as any)?.text ?? "",
       }