6e0f6d53a7
Extract agent call pipeline: - agent-type-normalizer.ts, tool-context-with-metadata.ts - subagent-session-creator.ts, subagent-session-prompter.ts - sync-agent-executor.ts, background-agent-executor.ts - session-completion-poller.ts, session-message-output-extractor.ts - message-storage-directory.ts
90 lines
3.2 KiB
TypeScript
90 lines
3.2 KiB
TypeScript
import type { PluginInput } from "@opencode-ai/plugin"
|
|
import { log } from "../../shared"
|
|
import { extractNewSessionOutput, type SessionMessage } from "./session-message-output-extractor"
|
|
import { waitForSessionCompletion } from "./session-completion-poller"
|
|
import { resolveOrCreateSessionId } from "./subagent-session-creator"
|
|
import { promptSubagentSession } from "./subagent-session-prompter"
|
|
import type { CallOmoAgentArgs } from "./types"
|
|
import type { ToolContextWithMetadata } from "./tool-context-with-metadata"
|
|
|
|
function buildTaskMetadata(sessionID: string): string {
|
|
return ["<task_metadata>", `session_id: ${sessionID}`, "</task_metadata>"].join(
|
|
"\n",
|
|
)
|
|
}
|
|
|
|
function getMessagesArray(result: unknown): SessionMessage[] {
|
|
if (Array.isArray(result)) return result as SessionMessage[]
|
|
if (typeof result !== "object" || result === null) return []
|
|
if (!("data" in result)) return []
|
|
const data = (result as { data?: unknown }).data
|
|
return Array.isArray(data) ? (data as SessionMessage[]) : []
|
|
}
|
|
|
|
export async function executeSyncAgent(
|
|
args: CallOmoAgentArgs,
|
|
toolContext: ToolContextWithMetadata,
|
|
ctx: PluginInput,
|
|
): Promise<string> {
|
|
const sessionResult = await resolveOrCreateSessionId(ctx, args, toolContext)
|
|
if (!sessionResult.ok) {
|
|
return sessionResult.error
|
|
}
|
|
const sessionID = sessionResult.sessionID
|
|
|
|
await toolContext.metadata?.({
|
|
title: args.description,
|
|
metadata: { sessionId: sessionID },
|
|
})
|
|
|
|
log(`[call_omo_agent] Sending prompt to session ${sessionID}`)
|
|
log("[call_omo_agent] Prompt preview", { preview: args.prompt.substring(0, 100) })
|
|
|
|
const promptResult = await promptSubagentSession(ctx, {
|
|
sessionID,
|
|
agent: args.subagent_type,
|
|
prompt: args.prompt,
|
|
})
|
|
if (!promptResult.ok) {
|
|
const errorMessage = promptResult.error
|
|
if (errorMessage.includes("agent.name") || errorMessage.includes("undefined")) {
|
|
return `Error: Agent "${args.subagent_type}" not found. Make sure the agent is registered in your opencode.json or provided by a plugin.\n\n${buildTaskMetadata(sessionID)}`
|
|
}
|
|
return `Error: Failed to send prompt: ${errorMessage}\n\n${buildTaskMetadata(sessionID)}`
|
|
}
|
|
|
|
log("[call_omo_agent] Prompt sent, polling for completion...")
|
|
const completion = await waitForSessionCompletion(ctx, {
|
|
sessionID,
|
|
abortSignal: toolContext.abort,
|
|
maxPollTimeMs: 5 * 60 * 1000,
|
|
pollIntervalMs: 500,
|
|
stabilityRequired: 3,
|
|
})
|
|
if (!completion.ok) {
|
|
if (completion.reason === "aborted") {
|
|
return `Task aborted.\n\n${buildTaskMetadata(sessionID)}`
|
|
}
|
|
return `Error: Agent task timed out after 5 minutes.\n\n${buildTaskMetadata(sessionID)}`
|
|
}
|
|
|
|
const messagesResult = await ctx.client.session.messages({
|
|
path: { id: sessionID },
|
|
})
|
|
if (messagesResult.error) {
|
|
log("[call_omo_agent] Messages error", { error: messagesResult.error })
|
|
return `Error: Failed to get messages: ${messagesResult.error}`
|
|
}
|
|
|
|
const messages = getMessagesArray(messagesResult)
|
|
log("[call_omo_agent] Got messages", { count: messages.length })
|
|
|
|
const extracted = extractNewSessionOutput(sessionID, messages)
|
|
if (!extracted.hasNewOutput) {
|
|
return `No new output since last check.\n\n${buildTaskMetadata(sessionID)}`
|
|
}
|
|
|
|
log("[call_omo_agent] Got response", { length: extracted.output.length })
|
|
return `${extracted.output}\n\n${buildTaskMetadata(sessionID)}`
|
|
}
|