Browse Source

fix(opencode): ensure unsub(PartUpdated) is always called in TaskTool (#9992)

Goni Zahavy 1 month ago
parent
commit
63f5669eb5
1 changed files with 4 additions and 1 deletions
  1. 4 1
      packages/opencode/src/tool/task.ts

+ 4 - 1
packages/opencode/src/tool/task.ts

@@ -160,7 +160,10 @@ export const TaskTool = Tool.define("task", async (ctx) => {
         },
         },
         parts: promptParts,
         parts: promptParts,
       })
       })
-      unsub()
+      .finally(() => {
+        unsub()
+      })
+
       const messages = await Session.messages({ sessionID: session.id })
       const messages = await Session.messages({ sessionID: session.id })
       const summary = messages
       const summary = messages
         .filter((x) => x.info.role === "assistant")
         .filter((x) => x.info.role === "assistant")