100 lines
3.1 KiB
TypeScript
100 lines
3.1 KiB
TypeScript
import { Type } from "@sinclair/typebox";
|
|
|
|
import { loadConfig } from "../../config/config.js";
|
|
import {
|
|
DEFAULT_AGENT_ID,
|
|
normalizeAgentId,
|
|
parseAgentSessionKey,
|
|
} from "../../routing/session-key.js";
|
|
import { resolveAgentConfig } from "../agent-scope.js";
|
|
import type { AnyAgentTool } from "./common.js";
|
|
import { jsonResult } from "./common.js";
|
|
import {
|
|
resolveInternalSessionKey,
|
|
resolveMainSessionAlias,
|
|
} from "./sessions-helpers.js";
|
|
|
|
const AgentsListToolSchema = Type.Object({});
|
|
|
|
type AgentListEntry = {
|
|
id: string;
|
|
name?: string;
|
|
configured: boolean;
|
|
};
|
|
|
|
export function createAgentsListTool(opts?: {
|
|
agentSessionKey?: string;
|
|
}): AnyAgentTool {
|
|
return {
|
|
label: "Agents",
|
|
name: "agents_list",
|
|
description:
|
|
"List agent ids you can target with sessions_spawn (based on allowlists).",
|
|
parameters: AgentsListToolSchema,
|
|
execute: async () => {
|
|
const cfg = loadConfig();
|
|
const { mainKey, alias } = resolveMainSessionAlias(cfg);
|
|
const requesterInternalKey =
|
|
typeof opts?.agentSessionKey === "string" && opts.agentSessionKey.trim()
|
|
? resolveInternalSessionKey({
|
|
key: opts.agentSessionKey,
|
|
alias,
|
|
mainKey,
|
|
})
|
|
: alias;
|
|
const requesterAgentId = normalizeAgentId(
|
|
parseAgentSessionKey(requesterInternalKey)?.agentId ?? DEFAULT_AGENT_ID,
|
|
);
|
|
|
|
const allowAgents =
|
|
resolveAgentConfig(cfg, requesterAgentId)?.subagents?.allowAgents ?? [];
|
|
const allowAny = allowAgents.some((value) => value.trim() === "*");
|
|
const allowSet = new Set(
|
|
allowAgents
|
|
.filter((value) => value.trim() && value.trim() !== "*")
|
|
.map((value) => normalizeAgentId(value)),
|
|
);
|
|
|
|
const configuredAgents = cfg.routing?.agents ?? {};
|
|
const configuredIds = Object.keys(configuredAgents).map((key) =>
|
|
normalizeAgentId(key),
|
|
);
|
|
const configuredNameMap = new Map<string, string>();
|
|
for (const [key, value] of Object.entries(configuredAgents)) {
|
|
if (!value || typeof value !== "object") continue;
|
|
const name =
|
|
typeof (value as { name?: unknown }).name === "string"
|
|
? ((value as { name?: string }).name?.trim() ?? "")
|
|
: "";
|
|
if (!name) continue;
|
|
configuredNameMap.set(normalizeAgentId(key), name);
|
|
}
|
|
|
|
const allowed = new Set<string>();
|
|
allowed.add(requesterAgentId);
|
|
if (allowAny) {
|
|
for (const id of configuredIds) allowed.add(id);
|
|
} else {
|
|
for (const id of allowSet) allowed.add(id);
|
|
}
|
|
|
|
const all = Array.from(allowed);
|
|
const rest = all
|
|
.filter((id) => id !== requesterAgentId)
|
|
.sort((a, b) => a.localeCompare(b));
|
|
const ordered = [requesterAgentId, ...rest];
|
|
const agents: AgentListEntry[] = ordered.map((id) => ({
|
|
id,
|
|
name: configuredNameMap.get(id),
|
|
configured: configuredIds.includes(id),
|
|
}));
|
|
|
|
return jsonResult({
|
|
requester: requesterAgentId,
|
|
allowAny,
|
|
agents,
|
|
});
|
|
},
|
|
};
|
|
}
|