@hiai-gg/hiai-opencode 0.1.1 → 0.1.2

package/src/tools/call-omo-agent/background-agent-executor.ts

@@ -1,91 +1,91 @@
-import type { BackgroundManager } from "../../features/background-agent"
-import type { PluginInput } from "@opencode-ai/plugin"
-import { resolveMessageContext } from "../../features/hook-message-injector"
-import { getSessionAgent } from "../../features/claude-code-session-state"
-import { log } from "../../shared"
-import type { CallOmoAgentArgs } from "./types"
-import type { ToolContextWithMetadata } from "./tool-context-with-metadata"
-import { getMessageDir } from "./message-storage-directory"
-import { getSessionTools } from "../../shared/session-tools-store"
-
-export async function executeBackgroundAgent(
-	args: CallOmoAgentArgs,
-	toolContext: ToolContextWithMetadata,
-	manager: BackgroundManager,
-	client: PluginInput["client"],
-): Promise<string> {
-	try {
-		const messageDir = getMessageDir(toolContext.sessionID)
-		const { prevMessage, firstMessageAgent } = await resolveMessageContext(
-			toolContext.sessionID,
-			client,
-			messageDir
-		)
-
-		const sessionAgent = getSessionAgent(toolContext.sessionID)
-		const parentAgent =
-			toolContext.agent ?? sessionAgent ?? firstMessageAgent ?? prevMessage?.agent
-
-		log("[call_omo_agent] parentAgent resolution", {
-			sessionID: toolContext.sessionID,
-			messageDir,
-			ctxAgent: toolContext.agent,
-			sessionAgent,
-			firstMessageAgent,
-			prevMessageAgent: prevMessage?.agent,
-			resolvedParentAgent: parentAgent,
-		})
-
-		const task = await manager.launch({
-			description: args.description,
-			prompt: args.prompt,
-			agent: args.subagent_type,
-			parentSessionID: toolContext.sessionID,
-			parentMessageID: toolContext.messageID,
-			parentAgent,
-			parentTools: getSessionTools(toolContext.sessionID),
-		})
-
-		const waitStart = Date.now()
-		const waitTimeoutMs = 30_000
-		const waitIntervalMs = 50
-
-		let sessionId = task.sessionID
-		while (!sessionId && Date.now() - waitStart < waitTimeoutMs) {
-			const updated = manager.getTask(task.id)
-			if (updated?.status === "error" || updated?.status === "cancelled" || updated?.status === "interrupt") {
-				return `Task failed to start (status: ${updated.status}).\n\nTask ID: ${task.id}`
-			}
-			sessionId = updated?.sessionID
-			if (sessionId) {
-				break
-			}
-			if (toolContext.abort?.aborted) {
-				break
-			}
-			await new Promise<void>((resolve) => {
-				setTimeout(resolve, waitIntervalMs)
-			})
-		}
-
-		await toolContext.metadata?.({
-			title: args.description,
-			metadata: { sessionId: sessionId ?? "pending" },
-		})
-
-		return `Background agent task launched successfully.
-
-Task ID: ${task.id}
-Session ID: ${sessionId ?? "pending"}
-Description: ${task.description}
-Agent: ${task.agent} (subagent)
-Status: ${task.status}
-
-System notifies on completion. Use \`background_output\` with task_id="${task.id}" to check.
-
-Do NOT call background_output now. Wait for <system-reminder> notification first.`
-	} catch (error) {
-		const message = error instanceof Error ? error.message : String(error)
-		return `Failed to launch background agent task: ${message}`
-	}
-}
+import type { BackgroundManager } from "../../features/background-agent"
+import type { PluginInput } from "@opencode-ai/plugin"
+import { resolveMessageContext } from "../../features/hook-message-injector"
+import { getSessionAgent } from "../../features/claude-code-session-state"
+import { log } from "../../shared"
+import type { CallOmoAgentArgs } from "./types"
+import type { ToolContextWithMetadata } from "./tool-context-with-metadata"
+import { getMessageDir } from "./message-storage-directory"
+import { getSessionTools } from "../../shared/session-tools-store"
+
+export async function executeBackgroundAgent(
+	args: CallOmoAgentArgs,
+	toolContext: ToolContextWithMetadata,
+	manager: BackgroundManager,
+	client: PluginInput["client"],
+): Promise<string> {
+	try {
+		const messageDir = getMessageDir(toolContext.sessionID)
+		const { prevMessage, firstMessageAgent } = await resolveMessageContext(
+			toolContext.sessionID,
+			client,
+			messageDir
+		)
+
+		const sessionAgent = getSessionAgent(toolContext.sessionID)
+		const parentAgent =
+			toolContext.agent ?? sessionAgent ?? firstMessageAgent ?? prevMessage?.agent
+
+		log("[call_omo_agent] parentAgent resolution", {
+			sessionID: toolContext.sessionID,
+			messageDir,
+			ctxAgent: toolContext.agent,
+			sessionAgent,
+			firstMessageAgent,
+			prevMessageAgent: prevMessage?.agent,
+			resolvedParentAgent: parentAgent,
+		})
+
+		const task = await manager.launch({
+			description: args.description,
+			prompt: args.prompt,
+			agent: args.subagent_type,
+			parentSessionID: toolContext.sessionID,
+			parentMessageID: toolContext.messageID,
+			parentAgent,
+			parentTools: getSessionTools(toolContext.sessionID),
+		})
+
+		const waitStart = Date.now()
+		const waitTimeoutMs = 30_000
+		const waitIntervalMs = 50
+
+		let sessionId = task.sessionID
+		while (!sessionId && Date.now() - waitStart < waitTimeoutMs) {
+			const updated = manager.getTask(task.id)
+			if (updated?.status === "error" || updated?.status === "cancelled" || updated?.status === "interrupt") {
+				return `Task failed to start (status: ${updated.status}).\n\nTask ID: ${task.id}`
+			}
+			sessionId = updated?.sessionID
+			if (sessionId) {
+				break
+			}
+			if (toolContext.abort?.aborted) {
+				break
+			}
+			await new Promise<void>((resolve) => {
+				setTimeout(resolve, waitIntervalMs)
+			})
+		}
+
+		await toolContext.metadata?.({
+			title: args.description,
+			metadata: { sessionId: sessionId ?? "pending" },
+		})
+
+		return `Background agent task launched successfully.
+
+Task ID: ${task.id}
+Session ID: ${sessionId ?? "pending"}
+Description: ${task.description}
+Agent: ${task.agent} (subagent)
+Status: ${task.status}
+
+System notifies on completion. Use \`background_output\` with task_id="${task.id}" to check.
+
+Do NOT call background_output now. Wait for <system-reminder> notification first.`
+	} catch (error) {
+		const message = error instanceof Error ? error.message : String(error)
+		return `Failed to launch background agent task: ${message}`
+	}
+}

package/src/tools/call-omo-agent/background-executor.ts

@@ -1,98 +1,98 @@
-import type { CallOmoAgentArgs } from "./types"
-import type { BackgroundManager } from "../../features/background-agent"
-import type { PluginInput } from "@opencode-ai/plugin"
-import { log } from "../../shared"
-import type { DelegatedModelConfig } from "../../shared/model-resolution-types"
-import type { FallbackEntry } from "../../shared/model-requirements"
-import { resolveMessageContext } from "../../features/hook-message-injector"
-import { getSessionAgent } from "../../features/claude-code-session-state"
-import { getMessageDir } from "./message-dir"
-import { getSessionTools } from "../../shared/session-tools-store"
-
-export async function executeBackground(
-  args: CallOmoAgentArgs,
-  toolContext: {
-    sessionID: string
-    messageID: string
-    agent: string
-    abort: AbortSignal
-    metadata?: (input: { title?: string; metadata?: Record<string, unknown> }) => void
-  },
-  manager: BackgroundManager,
-  client: PluginInput["client"],
-  fallbackChain?: FallbackEntry[],
-  model?: DelegatedModelConfig,
-): Promise<string> {
-  try {
-    const messageDir = getMessageDir(toolContext.sessionID)
-    const { prevMessage, firstMessageAgent } = await resolveMessageContext(
-      toolContext.sessionID,
-      client,
-      messageDir
-    )
-
-    const sessionAgent = getSessionAgent(toolContext.sessionID)
-    const parentAgent = toolContext.agent ?? sessionAgent ?? firstMessageAgent ?? prevMessage?.agent
-    
-    log("[call_omo_agent] parentAgent resolution", {
-      sessionID: toolContext.sessionID,
-      messageDir,
-      ctxAgent: toolContext.agent,
-      sessionAgent,
-      firstMessageAgent,
-      prevMessageAgent: prevMessage?.agent,
-      resolvedParentAgent: parentAgent,
-    })
-
-    const task = await manager.launch({
-      description: args.description,
-      prompt: args.prompt,
-      agent: args.subagent_type,
-      parentSessionID: toolContext.sessionID,
-      parentMessageID: toolContext.messageID,
-      parentAgent,
-      parentTools: getSessionTools(toolContext.sessionID),
-      model,
-      fallbackChain,
-    })
-
-    const WAIT_FOR_SESSION_INTERVAL_MS = 50
-    const WAIT_FOR_SESSION_TIMEOUT_MS = 30000
-    const waitStart = Date.now()
-    let sessionId = task.sessionID
-    while (!sessionId && Date.now() - waitStart < WAIT_FOR_SESSION_TIMEOUT_MS) {
-      const updated = manager.getTask(task.id)
-      if (updated?.status === "error" || updated?.status === "cancelled" || updated?.status === "interrupt") {
-        return `Task failed to start (status: ${updated.status}).\n\nTask ID: ${task.id}`
-      }
-      sessionId = updated?.sessionID
-      if (sessionId) {
-        break
-      }
-      if (toolContext.abort?.aborted) {
-        break
-      }
-      await new Promise(resolve => setTimeout(resolve, WAIT_FOR_SESSION_INTERVAL_MS))
-    }
-
-    await toolContext.metadata?.({
-      title: args.description,
-      metadata: { sessionId: sessionId ?? "pending" },
-    })
-
-    return `Background agent task launched successfully.
-
-Task ID: ${task.id}
-Session ID: ${sessionId ?? "pending"}
-Description: ${task.description}
-Agent: ${task.agent} (subagent)
-Status: ${task.status}
-
-System notifies on completion. Use \`background_output\` with task_id="${task.id}" to check.
-
-Do NOT call background_output now. Wait for <system-reminder> notification first.`
-  } catch (error) {
-    const message = error instanceof Error ? error.message : String(error)
-    return `Failed to launch background agent task: ${message}`
-  }
-}
+import type { CallOmoAgentArgs } from "./types"
+import type { BackgroundManager } from "../../features/background-agent"
+import type { PluginInput } from "@opencode-ai/plugin"
+import { log } from "../../shared"
+import type { DelegatedModelConfig } from "../../shared/model-resolution-types"
+import type { FallbackEntry } from "../../shared/model-requirements"
+import { resolveMessageContext } from "../../features/hook-message-injector"
+import { getSessionAgent } from "../../features/claude-code-session-state"
+import { getMessageDir } from "./message-dir"
+import { getSessionTools } from "../../shared/session-tools-store"
+
+export async function executeBackground(
+  args: CallOmoAgentArgs,
+  toolContext: {
+    sessionID: string
+    messageID: string
+    agent: string
+    abort: AbortSignal
+    metadata?: (input: { title?: string; metadata?: Record<string, unknown> }) => void
+  },
+  manager: BackgroundManager,
+  client: PluginInput["client"],
+  fallbackChain?: FallbackEntry[],
+  model?: DelegatedModelConfig,
+): Promise<string> {
+  try {
+    const messageDir = getMessageDir(toolContext.sessionID)
+    const { prevMessage, firstMessageAgent } = await resolveMessageContext(
+      toolContext.sessionID,
+      client,
+      messageDir
+    )
+
+    const sessionAgent = getSessionAgent(toolContext.sessionID)
+    const parentAgent = toolContext.agent ?? sessionAgent ?? firstMessageAgent ?? prevMessage?.agent
+    
+    log("[call_omo_agent] parentAgent resolution", {
+      sessionID: toolContext.sessionID,
+      messageDir,
+      ctxAgent: toolContext.agent,
+      sessionAgent,
+      firstMessageAgent,
+      prevMessageAgent: prevMessage?.agent,
+      resolvedParentAgent: parentAgent,
+    })
+
+    const task = await manager.launch({
+      description: args.description,
+      prompt: args.prompt,
+      agent: args.subagent_type,
+      parentSessionID: toolContext.sessionID,
+      parentMessageID: toolContext.messageID,
+      parentAgent,
+      parentTools: getSessionTools(toolContext.sessionID),
+      model,
+      fallbackChain,
+    })
+
+    const WAIT_FOR_SESSION_INTERVAL_MS = 50
+    const WAIT_FOR_SESSION_TIMEOUT_MS = 30000
+    const waitStart = Date.now()
+    let sessionId = task.sessionID
+    while (!sessionId && Date.now() - waitStart < WAIT_FOR_SESSION_TIMEOUT_MS) {
+      const updated = manager.getTask(task.id)
+      if (updated?.status === "error" || updated?.status === "cancelled" || updated?.status === "interrupt") {
+        return `Task failed to start (status: ${updated.status}).\n\nTask ID: ${task.id}`
+      }
+      sessionId = updated?.sessionID
+      if (sessionId) {
+        break
+      }
+      if (toolContext.abort?.aborted) {
+        break
+      }
+      await new Promise(resolve => setTimeout(resolve, WAIT_FOR_SESSION_INTERVAL_MS))
+    }
+
+    await toolContext.metadata?.({
+      title: args.description,
+      metadata: { sessionId: sessionId ?? "pending" },
+    })
+
+    return `Background agent task launched successfully.
+
+Task ID: ${task.id}
+Session ID: ${sessionId ?? "pending"}
+Description: ${task.description}
+Agent: ${task.agent} (subagent)
+Status: ${task.status}
+
+System notifies on completion. Use \`background_output\` with task_id="${task.id}" to check.
+
+Do NOT call background_output now. Wait for <system-reminder> notification first.`
+  } catch (error) {
+    const message = error instanceof Error ? error.message : String(error)
+    return `Failed to launch background agent task: ${message}`
+  }
+}

package/src/tools/call-omo-agent/completion-poller.ts

@@ -1,65 +1,65 @@
-import type { PluginInput } from "@opencode-ai/plugin"
-import { log } from "../../shared"
-import { normalizeSDKResponse } from "../../shared"
-
-export async function waitForCompletion(
-  sessionID: string,
-  toolContext: {
-    sessionID: string
-    messageID: string
-    agent: string
-    abort: AbortSignal
-    metadata?: (input: { title?: string; metadata?: Record<string, unknown> }) => void
-  },
-  ctx: PluginInput
-): Promise<void> {
-  log(`[call_omo_agent] Polling for completion...`)
-
-  const POLL_INTERVAL_MS = 500
-  const MAX_POLL_TIME_MS = 5 * 60 * 1000 // 5 minutes max
-  const pollStart = Date.now()
-  let lastMsgCount = 0
-  let stablePolls = 0
-  const STABILITY_REQUIRED = 3
-
-  while (Date.now() - pollStart < MAX_POLL_TIME_MS) {
-    if (toolContext.abort?.aborted) {
-      log(`[call_omo_agent] Aborted by user`)
-      throw new Error("Task aborted.")
-    }
-
-    await new Promise(resolve => setTimeout(resolve, POLL_INTERVAL_MS))
-
-    const statusResult = await ctx.client.session.status()
-    const allStatuses = normalizeSDKResponse(statusResult, {} as Record<string, { type: string }>)
-    const sessionStatus = allStatuses[sessionID]
-
-    if (sessionStatus && sessionStatus.type !== "idle") {
-      stablePolls = 0
-      lastMsgCount = 0
-      continue
-    }
-
-    const messagesCheck = await ctx.client.session.messages({ path: { id: sessionID } })
-    const msgs = normalizeSDKResponse(messagesCheck, [] as Array<unknown>, {
-      preferResponseOnMissingData: true,
-    })
-    const currentMsgCount = msgs.length
-
-    if (currentMsgCount > 0 && currentMsgCount === lastMsgCount) {
-      stablePolls++
-      if (stablePolls >= STABILITY_REQUIRED) {
-        log(`[call_omo_agent] Session complete, ${currentMsgCount} messages`)
-        break
-      }
-    } else {
-      stablePolls = 0
-      lastMsgCount = currentMsgCount
-    }
-  }
-
-  if (Date.now() - pollStart >= MAX_POLL_TIME_MS) {
-    log(`[call_omo_agent] Timeout reached`)
-    throw new Error("Agent task timed out after 5 minutes.")
-  }
-}
+import type { PluginInput } from "@opencode-ai/plugin"
+import { log } from "../../shared"
+import { normalizeSDKResponse } from "../../shared"
+
+export async function waitForCompletion(
+  sessionID: string,
+  toolContext: {
+    sessionID: string
+    messageID: string
+    agent: string
+    abort: AbortSignal
+    metadata?: (input: { title?: string; metadata?: Record<string, unknown> }) => void
+  },
+  ctx: PluginInput
+): Promise<void> {
+  log(`[call_omo_agent] Polling for completion...`)
+
+  const POLL_INTERVAL_MS = 500
+  const MAX_POLL_TIME_MS = 5 * 60 * 1000 // 5 minutes max
+  const pollStart = Date.now()
+  let lastMsgCount = 0
+  let stablePolls = 0
+  const STABILITY_REQUIRED = 3
+
+  while (Date.now() - pollStart < MAX_POLL_TIME_MS) {
+    if (toolContext.abort?.aborted) {
+      log(`[call_omo_agent] Aborted by user`)
+      throw new Error("Task aborted.")
+    }
+
+    await new Promise(resolve => setTimeout(resolve, POLL_INTERVAL_MS))
+
+    const statusResult = await ctx.client.session.status()
+    const allStatuses = normalizeSDKResponse(statusResult, {} as Record<string, { type: string }>)
+    const sessionStatus = allStatuses[sessionID]
+
+    if (sessionStatus && sessionStatus.type !== "idle") {
+      stablePolls = 0
+      lastMsgCount = 0
+      continue
+    }
+
+    const messagesCheck = await ctx.client.session.messages({ path: { id: sessionID } })
+    const msgs = normalizeSDKResponse(messagesCheck, [] as Array<unknown>, {
+      preferResponseOnMissingData: true,
+    })
+    const currentMsgCount = msgs.length
+
+    if (currentMsgCount > 0 && currentMsgCount === lastMsgCount) {
+      stablePolls++
+      if (stablePolls >= STABILITY_REQUIRED) {
+        log(`[call_omo_agent] Session complete, ${currentMsgCount} messages`)
+        break
+      }
+    } else {
+      stablePolls = 0
+      lastMsgCount = currentMsgCount
+    }
+  }
+
+  if (Date.now() - pollStart >= MAX_POLL_TIME_MS) {
+    log(`[call_omo_agent] Timeout reached`)
+    throw new Error("Agent task timed out after 5 minutes.")
+  }
+}

package/src/tools/call-omo-agent/constants.ts

@@ -1,23 +1,23 @@
-export const PRIMARY_ALLOWED_AGENTS = [
-  "researcher",
-  "strategist",
-  "coder",
-  "critic",
-  "designer",
-  "brainstormer",
-  "platform-manager",
-  "multimodal",
-] as const
-
-export const ALLOWED_AGENTS = [
-  ...PRIMARY_ALLOWED_AGENTS,
-] as const
-
-export const CALL_OMO_AGENT_DESCRIPTION = `Spawn canonical agents or custom agents. run_in_background REQUIRED (true=async with task_id, false=sync).
-
-Canonical built-in agents:
-{primary_agents}
-
-Custom agents registered via user or project agent directories are also supported.
-
-Pass \`session_id=<id>\` to continue previous agent with full context. Nested subagent depth is tracked automatically and blocked past the configured limit. Prompts MUST be in English. Use \`background_output\` for async results.`
+export const PRIMARY_ALLOWED_AGENTS = [
+  "researcher",
+  "strategist",
+  "coder",
+  "critic",
+  "designer",
+  "brainstormer",
+  "platform-manager",
+  "multimodal",
+] as const
+
+export const ALLOWED_AGENTS = [
+  ...PRIMARY_ALLOWED_AGENTS,
+] as const
+
+export const CALL_OMO_AGENT_DESCRIPTION = `Spawn canonical agents or custom agents. run_in_background REQUIRED (true=async with task_id, false=sync).
+
+Canonical built-in agents:
+{primary_agents}
+
+Custom agents registered via user or project agent directories are also supported.
+
+Pass \`session_id=<id>\` to continue previous agent with full context. Nested subagent depth is tracked automatically and blocked past the configured limit. Prompts MUST be in English. Use \`background_output\` for async results.`

package/src/tools/call-omo-agent/index.ts

@@ -1,3 +1,3 @@
-export * from "./types"
-export * from "./constants"
-export { createCallOmoAgent } from "./tools"
+export * from "./types"
+export * from "./constants"
+export { createCallOmoAgent } from "./tools"

package/src/tools/call-omo-agent/message-dir.ts

@@ -1 +1 @@
-export { getMessageDir } from "../../shared/opencode-message-dir"
+export { getMessageDir } from "../../shared/opencode-message-dir"