Part data model (#950)

packages/opencode/src/tool/task.ts

@@ -4,6 +4,7 @@ import { z } from "zod"
 import { Session } from "../session"
 import { Bus } from "../bus"
 import { MessageV2 } from "../session/message-v2"
+import { Identifier } from "../id/id"
 
 export const TaskTool = Tool.define({
   id: "task",
@@ -16,9 +17,10 @@ export const TaskTool = Tool.define({
     const session = await Session.create(ctx.sessionID)
     const msg = (await Session.getMessage(ctx.sessionID, ctx.messageID)) as MessageV2.Assistant
 
-    function summary(input: MessageV2.Info) {
+    const parts: Record<string, MessageV2.Part> = {}
+    function summary(input: MessageV2.Part[]) {
       const result = []
-      for (const part of input.parts) {
+      for (const part of input) {
         if (part.type === "tool" && part.state.status === "completed") {
           result.push(part)
         }
@@ -26,12 +28,13 @@ export const TaskTool = Tool.define({
       return result
     }
 
-    const unsub = Bus.subscribe(MessageV2.Event.Updated, async (evt) => {
-      if (evt.properties.info.sessionID !== session.id) return
+    const unsub = Bus.subscribe(MessageV2.Event.PartUpdated, async (evt) => {
+      if (evt.properties.part.sessionID !== session.id) return
+      parts[evt.properties.part.id] = evt.properties.part
       ctx.metadata({
         title: params.description,
         metadata: {
-          summary: summary(evt.properties.info),
+          summary: Object.values(parts).sort((a, b) => a.id?.localeCompare(b.id)),
         },
       })
     })
@@ -39,12 +42,17 @@ export const TaskTool = Tool.define({
     ctx.abort.addEventListener("abort", () => {
       Session.abort(session.id)
     })
+    const messageID = Identifier.ascending("message")
     const result = await Session.chat({
+      messageID,
       sessionID: session.id,
       modelID: msg.modelID,
       providerID: msg.providerID,
       parts: [
         {
+          id: Identifier.ascending("part"),
+          messageID,
+          sessionID: session.id,
           type: "text",
           text: params.prompt,
         },
@@ -54,7 +62,7 @@ export const TaskTool = Tool.define({
     return {
       title: params.description,
       metadata: {
-        summary: summary(result),
+        summary: summary(result.parts),
       },
       output: result.parts.findLast((x) => x.type === "text")!.text,
     }