|
|
@@ -2,6 +2,8 @@ import { Tool } from "./tool"
|
|
|
import DESCRIPTION from "./task.txt"
|
|
|
import { z } from "zod"
|
|
|
import { Session } from "../session"
|
|
|
+import { Bus } from "../bus"
|
|
|
+import { Message } from "../session/message"
|
|
|
|
|
|
export const TaskTool = Tool.define({
|
|
|
id: "opencode.task",
|
|
|
@@ -17,6 +19,28 @@ export const TaskTool = Tool.define({
|
|
|
const msg = await Session.getMessage(ctx.sessionID, ctx.messageID)
|
|
|
const metadata = msg.metadata.assistant!
|
|
|
|
|
|
+ function summary(input: Message.Info) {
|
|
|
+ const result = []
|
|
|
+
|
|
|
+ for (const part of input.parts) {
|
|
|
+ if (part.type === "tool-invocation") {
|
|
|
+ result.push({
|
|
|
+ toolInvocation: part.toolInvocation,
|
|
|
+ metadata: input.metadata.tool[part.toolInvocation.toolCallId],
|
|
|
+ })
|
|
|
+ }
|
|
|
+ }
|
|
|
+ return result
|
|
|
+ }
|
|
|
+
|
|
|
+ const unsub = Bus.subscribe(Message.Event.Updated, async (evt) => {
|
|
|
+ if (evt.properties.info.metadata.sessionID !== ctx.sessionID) return
|
|
|
+ ctx.metadata({
|
|
|
+ title: params.description,
|
|
|
+ summary: summary(evt.properties.info),
|
|
|
+ })
|
|
|
+ })
|
|
|
+
|
|
|
const result = await Session.chat({
|
|
|
sessionID: session.id,
|
|
|
modelID: metadata.modelID,
|
|
|
@@ -28,10 +52,11 @@ export const TaskTool = Tool.define({
|
|
|
},
|
|
|
],
|
|
|
})
|
|
|
-
|
|
|
+ unsub()
|
|
|
return {
|
|
|
metadata: {
|
|
|
title: params.description,
|
|
|
+ summary: summary(result),
|
|
|
},
|
|
|
output: result.parts.findLast((x) => x.type === "text")!.text,
|
|
|
}
|