oh-my-opencode/src/tools/call-omo-agent/agent-resolver.ts

import type { PluginInput } from "@opencode-ai/plugin";
import { ALLOWED_AGENTS } from "./constants";
import { normalizeSDKResponse } from "../../shared";
import { log } from "../../shared/logger";

type AgentInfo = {
  name: string;
  mode?: "subagent" | "primary" | "all";
};

/**
 * Resolves the set of callable agent names at execute-time by merging the
 * hardcoded `ALLOWED_AGENTS` with any additional agents discovered dynamically
 * via `client.app.agents()`. Custom agents loaded from registered agent
 * directories appear here alongside built-ins.
 *
 * Falls back to `ALLOWED_AGENTS` alone if the dynamic lookup fails.
 *
 * @param client - The plugin client with access to the agent registry
 * @returns Array of lowercase callable agent names (excludes primary-mode agents)
 */
export async function resolveCallableAgents(
  client: PluginInput["client"],
): Promise<string[]> {
  try {
    const agentsResult = await client.app.agents();
    const agents = normalizeSDKResponse(agentsResult, [] as AgentInfo[], {
      preferResponseOnMissingData: true,
    });

    const dynamicAgents = agents
      .filter((a) => a && typeof a.name === "string" && a.name.trim().length > 0 && a.mode !== "primary")
      .map((a) => a.name.trim().toLowerCase());

    const merged = new Set([...ALLOWED_AGENTS, ...dynamicAgents]);
    return [...merged];
  } catch (error) {
    const message = error instanceof Error ? error.message : String(error);
    log(
      "[call_omo_agent] Failed to resolve dynamic agents, falling back to built-in list",
      { error: message },
    );
    return [...ALLOWED_AGENTS];
  }
}