2026-02-08 13:57:26 +09:00
|
|
|
const TARGET_TOOLS = ["task", "Task", "task_tool", "call_omo_agent"]
|
|
|
|
|
|
|
|
|
|
const SESSION_ID_PATTERNS = [
|
|
|
|
|
/Session ID: (ses_[a-zA-Z0-9_-]+)/,
|
|
|
|
|
/session_id: (ses_[a-zA-Z0-9_-]+)/,
|
|
|
|
|
/<task_metadata>\s*session_id: (ses_[a-zA-Z0-9_-]+)/,
|
|
|
|
|
/sessionId: (ses_[a-zA-Z0-9_-]+)/,
|
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
function extractSessionId(output: string): string | null {
|
|
|
|
|
for (const pattern of SESSION_ID_PATTERNS) {
|
|
|
|
|
const match = output.match(pattern)
|
|
|
|
|
if (match) return match[1] ?? null
|
|
|
|
|
}
|
|
|
|
|
return null
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
export function createTaskResumeInfoHook() {
|
|
|
|
|
const toolExecuteAfter = async (
|
|
|
|
|
input: { tool: string; sessionID: string; callID: string },
|
|
|
|
|
output: { title: string; output: string; metadata: unknown }
|
|
|
|
|
) => {
|
|
|
|
|
if (!TARGET_TOOLS.includes(input.tool)) return
|
2026-02-11 15:49:56 +09:00
|
|
|
const outputText = output.output ?? ""
|
|
|
|
|
if (outputText.startsWith("Error:") || outputText.startsWith("Failed")) return
|
|
|
|
|
if (outputText.includes("\nto continue:")) return
|
2026-02-08 13:57:26 +09:00
|
|
|
|
2026-02-11 15:49:56 +09:00
|
|
|
const sessionId = extractSessionId(outputText)
|
2026-02-08 13:57:26 +09:00
|
|
|
if (!sessionId) return
|
|
|
|
|
|
|
|
|
|
output.output =
|
2026-02-11 15:49:56 +09:00
|
|
|
outputText.trimEnd() +
|
2026-03-31 17:02:56 -07:00
|
|
|
`\n\nto continue: task(session_id="${sessionId}", load_skills=[], run_in_background=false, prompt="...")`
|
2026-02-08 13:57:26 +09:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return {
|
|
|
|
|
"tool.execute.after": toolExecuteAfter,
|
|
|
|
|
}
|
|
|
|
|
}
|