85 lines
2.8 KiB
TypeScript
85 lines
2.8 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 = Array.isArray(cfg.agents?.list) ? cfg.agents?.list : [];
|
|
const configuredIds = configuredAgents.map((entry) => normalizeAgentId(entry.id));
|
|
const configuredNameMap = new Map<string, string>();
|
|
for (const entry of configuredAgents) {
|
|
const name = entry?.name?.trim() ?? "";
|
|
if (!name) continue;
|
|
configuredNameMap.set(normalizeAgentId(entry.id), 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,
|
|
});
|
|
},
|
|
};
|
|
}
|