ToothFairyAI_Public/tf_code
1
0
Fork 0
mirror of https://gitea.toothfairyai.com/ToothFairyAI/tf_code.git synced 2026-04-02 07:03:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

plugin(hook): add task tool execution hooks and command context tracking (#5642)

Browse Source
This commit is contained in:
Spoon
2025-12-17 02:28:09 +01:00
committed by GitHub
parent a6a8f41fd3
commit b8a8fb0de6
4 changed files with 45 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
packages/opencode/src/session/message-v2.ts
View File

@@ -160,6 +160,7 @@ export namespace MessageV2 {
prompt: z.string(),
description: z.string(),
agent: z.string(),
command: z.string().optional(),
})
export type SubtaskPart = z.infer<typeof SubtaskPart>

63
packages/opencode/src/session/prompt.ts
View File

@@ -325,42 +325,60 @@ export namespace SessionPrompt {
prompt: task.prompt,
description: task.description,
subagent_type: task.agent,
command: task.command,
},
time: {
start: Date.now(),
},
},
})) as MessageV2.ToolPart
const taskArgs = {
prompt: task.prompt,
description: task.description,
subagent_type: task.agent,
command: task.command,
}
await Plugin.trigger(
"tool.execute.before",
{
tool: "task",
sessionID,
callID: part.id,
},
{ args: taskArgs },
)
let executionError: Error | undefined
const result = await taskTool
.execute(
{
prompt: task.prompt,
description: task.description,
subagent_type: task.agent,
.execute(taskArgs, {
agent: task.agent,
messageID: assistantMessage.id,
sessionID: sessionID,
abort,
async metadata(input) {
await Session.updatePart({
...part,
type: "tool",
state: {
...part.state,
...input,
},
} satisfies MessageV2.ToolPart)
},
{
agent: task.agent,
messageID: assistantMessage.id,
sessionID: sessionID,
abort,
async metadata(input) {
await Session.updatePart({
...part,
type: "tool",
state: {
...part.state,
...input,
},
} satisfies MessageV2.ToolPart)
},
},
)
})
.catch((error) => {
executionError = error
log.error("subtask execution failed", { error, agent: task.agent, description: task.description })
return undefined
})
await Plugin.trigger(
"tool.execute.after",
{
tool: "task",
sessionID,
callID: part.id,
},
result,
)
assistantMessage.finish = "tool-calls"
assistantMessage.time.completed = Date.now()
await Session.updateMessage(assistantMessage)
@@ -1323,6 +1341,7 @@ export namespace SessionPrompt {
type: "subtask" as const,
agent: agent.name,
description: command.description ?? "",
command: input.command,
// TODO: how can we make task tool accept a more complex input?
prompt: await resolvePromptParts(template).then((x) => x.find((y) => y.type === "text")?.text ?? ""),
},