8514906c3d
Pass parentTools from session-tools-store through the background task lifecycle (launch → task → notify) so that when notifyParentSession sends promptAsync, the original tool restrictions (e.g., question: false) are preserved. This prevents the Question tool from re-enabling after call_omo_agent background tasks complete.
90 lines
3.1 KiB
TypeScript
90 lines
3.1 KiB
TypeScript
import type { DelegateTaskArgs, ToolContextWithMetadata } from "./types"
|
|
import type { ExecutorContext, ParentContext } from "./executor-types"
|
|
import { getTimingConfig } from "./timing"
|
|
import { storeToolMetadata } from "../../features/tool-metadata-store"
|
|
import { formatDetailedError } from "./error-formatting"
|
|
import { getSessionTools } from "../../shared/session-tools-store"
|
|
|
|
export async function executeBackgroundTask(
|
|
args: DelegateTaskArgs,
|
|
ctx: ToolContextWithMetadata,
|
|
executorCtx: ExecutorContext,
|
|
parentContext: ParentContext,
|
|
agentToUse: string,
|
|
categoryModel: { providerID: string; modelID: string; variant?: string } | undefined,
|
|
systemContent: string | undefined
|
|
): Promise<string> {
|
|
const { manager } = executorCtx
|
|
|
|
try {
|
|
const task = await manager.launch({
|
|
description: args.description,
|
|
prompt: args.prompt,
|
|
agent: agentToUse,
|
|
parentSessionID: parentContext.sessionID,
|
|
parentMessageID: parentContext.messageID,
|
|
parentModel: parentContext.model,
|
|
parentAgent: parentContext.agent,
|
|
parentTools: getSessionTools(parentContext.sessionID),
|
|
model: categoryModel,
|
|
skills: args.load_skills.length > 0 ? args.load_skills : undefined,
|
|
skillContent: systemContent,
|
|
category: args.category,
|
|
})
|
|
|
|
// OpenCode TUI's `Task` tool UI calculates toolcalls by looking up
|
|
// `props.metadata.sessionId` and then counting tool parts in that session.
|
|
// BackgroundManager.launch() returns immediately (pending) before the session exists,
|
|
// so we must wait briefly for the session to be created to set metadata correctly.
|
|
const timing = getTimingConfig()
|
|
const waitStart = Date.now()
|
|
let sessionId = task.sessionID
|
|
while (!sessionId && Date.now() - waitStart < timing.WAIT_FOR_SESSION_TIMEOUT_MS) {
|
|
if (ctx.abort?.aborted) {
|
|
return `Task aborted while waiting for session to start.\n\nTask ID: ${task.id}`
|
|
}
|
|
await new Promise(resolve => setTimeout(resolve, timing.WAIT_FOR_SESSION_INTERVAL_MS))
|
|
const updated = manager.getTask(task.id)
|
|
sessionId = updated?.sessionID
|
|
}
|
|
|
|
const unstableMeta = {
|
|
title: args.description,
|
|
metadata: {
|
|
prompt: args.prompt,
|
|
agent: task.agent,
|
|
category: args.category,
|
|
load_skills: args.load_skills,
|
|
description: args.description,
|
|
run_in_background: args.run_in_background,
|
|
sessionId: sessionId ?? "pending",
|
|
command: args.command,
|
|
},
|
|
}
|
|
await ctx.metadata?.(unstableMeta)
|
|
if (ctx.callID) {
|
|
storeToolMetadata(ctx.sessionID, ctx.callID, unstableMeta)
|
|
}
|
|
|
|
return `Background task launched.
|
|
|
|
Task ID: ${task.id}
|
|
Description: ${task.description}
|
|
Agent: ${task.agent}${args.category ? ` (category: ${args.category})` : ""}
|
|
Status: ${task.status}
|
|
|
|
System notifies on completion. Use \`background_output\` with task_id="${task.id}" to check.
|
|
|
|
<task_metadata>
|
|
session_id: ${sessionId}
|
|
</task_metadata>`
|
|
} catch (error) {
|
|
return formatDetailedError(error, {
|
|
operation: "Launch background task",
|
|
args,
|
|
agent: agentToUse,
|
|
category: args.category,
|
|
})
|
|
}
|
|
}
|